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`), telling 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

这应该让你开始。在最糟糕的情况下，有大量可用的文档（例如，this one）......

Answer 2

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

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

-----将args发送到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:])

然后，您可以使用-a或--algorithm=选项指定其他算法：

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

请参阅：getopt documentation

将参数从cmd传递给python脚本

3 个答案: