python - 将参数从 cmd 传递到 python 脚本

Question

我在 python 中编写脚本并通过输入以下命令使用 cmd 运行它们：

C:\> python script.py

我的一些脚本包含基于标志调用的单独算法和方法。现在我想直接通过 cmd 传递标志，而不是在运行之前进入脚本并更改标志，我想要类似于：

C:\> python script.py -algorithm=2

我读过人们将 sys.argv 用于几乎类似的目的，但是阅读手册和论坛我无法理解它是如何工作的。

Answer 1

有一些专门用于解析命令行参数的模块getopt：optparse和argparse. optparse已弃用，并且getopt功能不如argparse，因此我建议您使用后者，从长远来看它会更有帮助。

这是一个简短的示例：

import argparse

# Define the parser
parser = argparse.ArgumentParser(description='Short sample app')

# Declare an argument (`--algo`), saying that the 
# corresponding value should be stored in the `algo` 
# field, and using a default value if the argument 
# isn't given
parser.add_argument('--algo', action="store", dest='algo', default=0)

# Now, parse the command line arguments and store the 
# values in the `args` variable
args = parser.parse_args()

# Individual arguments can be accessed as attributes...
print args.algo

那应该让你开始。在最坏的情况下，网上有很多可用的文档（例如，这个）......

Answer 2

它可能无法回答您的问题，但有些人可能会觉得它很有用（我在这里寻找这个）：

如何将 2 个 args (arg1 + arg2) 从 cmd 发送到 python 3：

----- 在 test.cmd 中发送参数：

python "C:\Users\test.pyw" "arg1" "arg2"

----- 检索 test.py 中的参数：

print ("This is the name of the script= ", sys.argv[0])
print("Number of arguments= ", len(sys.argv))
print("all args= ", str(sys.argv))
print("arg1= ", sys.argv[1])
print("arg2= ", sys.argv[2])

Answer 3

尝试使用该getopt模块。它可以处理短命令行选项和长命令行选项，并且在其他语言（C、shell 脚本等）中以类似的方式实现：

import sys, getopt


def main(argv):

    # default algorithm:
    algorithm = 1

    # parse command line options:
    try:
       opts, args = getopt.getopt(argv,"a:",["algorithm="])
    except getopt.GetoptError:
       <print usage>
       sys.exit(2)

    for opt, arg in opts:
       if opt in ("-a", "--algorithm"):
          # use alternative algorithm:
          algorithm = arg

    print "Using algorithm: ", algorithm

    # Positional command line arguments (i.e. non optional ones) are
    # still available via 'args':
    print "Positional args: ", args

if __name__ == "__main__":
   main(sys.argv[1:])

然后，您可以使用-aor--algorithm=选项指定不同的算法：

python <scriptname> -a2               # use algorithm 2
python <scriptname> --algorithm=2    # ditto

请参阅：getopt 文档