使用 argparse 运行我的脚本中的 2 个函数之一

Question

目前我的 .py 脚本中有 2 个函数。

#1 连接到数据库并进行一些处理。

#2 对文件进行一些其他处理

目前，在运行脚本之前，我必须手动注释/取消注释我想要在 ma​​in if 语句块中运行的函数。

如何使用 argparse，以便它在运行脚本时询问我要运行哪个函数？

Answer 1

可以通过 action='store_const' 和 const=直接告诉 ArgumentParser 对象具有您所需行为的函数或对象; 在 add_argument() 调用中配对，或与 set_defaults() 调用配对（当您使用子解析器时，后者最有用）。如果这样做，您可以在从解析返回的 parsed_args 对象上查找您的函数，而不是在全局命名空间中查找它。

举个小例子：

import argparse

def foo(parsed_args):
    print "woop is {0!r}".format(getattr(parsed_args, 'woop'))

def bar(parsed_args):
    print "moop is {0!r}".format(getattr(parsed_args, 'moop'))

parser = argparse.ArgumentParser()

parser.add_argument('--foo', dest='action', action='store_const', const=foo)
parser.add_argument('--bar', dest='action', action='store_const', const=bar)
parser.add_argument('--woop')
parser.add_argument('--moop')

parsed_args = parser.parse_args()
if parsed_args.action is None:
    parser.parse_args(['-h'])
parsed_args.action(parsed_args)

然后你可以这样称呼它：

% python /tmp/junk.py --foo                                           
woop is None
% python /tmp/junk.py --foo --woop 8 --moop 17                        
woop is '8'
% python /tmp/junk.py --bar --woop 8 --moop 17                        
moop is '17'

Answer 2

如果它只是运行 A 或 B 的标志，那么一个简单的“store_true”参数应该没问题。

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--run_a_not_b', action='store_true')
_StoreTrueAction(option_strings=['--run_a_not_b'], dest='run_a_not_b', nargs=0, const=True, default=False, type=None, choices=None, help=None, metavar=None)
>>> parser.parse_args('--run_a_not_b')
>>> parsed_args = parser.parse_args('--run_a_not_b'.split())
>>> if parsed_args.run_a_not_b:
    print "run a"
else:
    print "run b"


run a
>>> parsed_args = parser.parse_args(''.split())
>>> if parsed_args.run_a_not_b:
    print "run a"
else:
    print "run b"


run b

或者，如果您想实际传递要调用的函数的名称，您可以这样做（有点黑客）：

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--func_to_run', type=str)
_StoreAction(option_strings=['--func_to_run'], dest='func_to_run', nargs=None, const=None, default=None, type=<type 'str'>, choices=None, help=None, metavar=None)
>>> parsed_args = parser.parse_args('--func_to_run my_other_func'.split())
>>> parsed_args.func_to_run
'my_other_func'
>>> f = globals()[parsed_args.func_to_run]
<function my_other_func at 0x011F6670>
>>> f()

编辑：要处理整数参数，您只需指定类型即可

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--run_a_not_b', action='store_true')
>>> parser.add_argument('--func_arg', type=int)
>>> parsed_args = parser.parse_args('--run_a_not_b --arg 42'.split())
>>> parsed_args = parser.parse_args('--run_a_not_b --func_arg 42'.split())
>>> parsed_args
Namespace(func_arg=42, run_a_not_b=True)

，如果您在此示例中选择，则可以简单地获取 parsed_args.func_arg 的值。

Answer 3

您可以考虑使用布料来实现此目的。