Python 命令行参数：调用函数

Question

所以我被困在一个我正在从事的项目中，该项目涉及 python 中的命令行。

所以基本上，这就是我想要完成的：

我在一个类中有一组函数，比如说，

def do_option1(self, param1, param2) :
    #some python code here

def do_option2(self, param1): 
    #some python code here

def do_option3(self, param1, param2, param3):
    #some python code here

基本上，当用户将filename.py option2 param1 放入命令行时，我希望它调用函数do_option2 并将参数param1 传递给它。

同样，当用户输入filename.py option3 param1 param2 param3时，我希望它执行带有给定参数的do_option3函数。

我知道 python 中有两个模块，称为 argparse 和 optparse，但我很难理解这两个模块，我不确定这两个模块中的任何一个是否能够完成我需要做的事情。

Answer 1

使用argparse 子命令解析器

p = argparse.ArgumentParser()
subparsers = p.add_subparsers()

option1_parser = subparsers.add_parser('option1')
# Add specific options for option1 here, but here's
# an example
option1_parser.add_argument('param1')
option1_parser.set_defaults(func=do_option1)

option2_parser = subparsers.add_parser('option2')
# Add specific options for option1 here
option2_parser.set_defaults(func=do_option2)

option3_parser = subparsers.add_parser('option3')
# Add specific options for option3 here
option3_parser.set_defaults(func=do_option3)

args = p.parse_args()
args.func(args)

然后，您的每个do_option 函数都需要稍微重写，以获取一个可以从中提取所需值的参数。例如：

def do_option1(args):
    param1 = args.param1
    # And continue

Answer 2

我使用这个模板可以帮助你。

import argparse

parser = argparse.ArgumentParser()

parser.add_argument('--option1', help='description for option1')
parser.add_argument('--option2', help='description for option2')
parser.add_argument('--option3', help='description for option3')

args = parser.parse_args()

if args.option1:
    ...do something

if args.option2:
    ...do something

if args.option3:
    ...do something

然后您可以运行您的脚本，传递如下参数：

python script.py --option1 my-option1 --option3 my-option3 --option2 my-option2

请注意，参数的位置并不重要，因为您在参数值之前指定了参数的名称。

Answer 3

options = {
    'option_1': my_class.option_1,
    'option_2': my_class.option_2,
    'option_3': my_class.option_3,
}

option, params = sys.argv[0], sys.argv[1:]
options[option](*params)

应该做的伎俩。您可能需要添加一些检查以确保用户至少将一些参数传递给您的脚本。

Answer 4

Execute function via arg

正如接受的答案中所述，您可以（ab？）使用add_argument 的type 参数来执行函数。

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('option2', type=interface.do_option2)

args = parser.parse_args()

Answer 5

如果你想直接传入参数而不使用标签，你可以使用 sys 模块：sys.argv 会给你['filename.py', 'option3', 'param1', 'param2', 'param3'] 用于filename.py option3 param1 param2 param3的命令行

http://www.tutorialspoint.com/python/python_command_line_arguments.htm