Question

需要帮助理解Python中的argparse用例3.尝试编写一个使用参数调用各种函数的简单程序。例如：

[程序] [行动] [可选参数]

nachos.py delete 20394739
- or -
nachos.py -d 20394739

nachos.py search 30459389
- or -
nachos.py -s 30459389

nachos.py list
- or -
nachos.py -l

试图弄清楚如何使用Python 3实现这一点。请帮忙。谢谢！

import argparse
parser = argparse.ArgumentParser(description='Example list of options')
parser.add_argument('-d', '--delete', action='delete', help='Delete ID')
parser.add_argument('-s', '--search', action='search', help='Search ID')
parser.add_argument('-l', '--list', action='list', help='List all ID')
args = parser.parse_args()

def (some function to handle each case)

*更新*

这似乎到目前为止工作，但检查可能很长的if语句列表似乎缓慢而笨重。有没有办法用一个看似更具体和更重要的功能来做到这一点？

parser = argparse.ArgumentParser(description='Example list of options', add_help=True)
parser.add_argument('-d', '--delete', dest='command', action='store_const', const='delete', help='Delete ID')
parser.add_argument('-s', '--search', dest='command', action='store_const', const='search', help='Search ID')
parser.add_argument('-l', '--list', dest='command', action='store_const', const='list', help='List all ID')
args = parser.parse_args()

if args.command == 'delete':
        print('Run delete')
elif args.command == 'search':
        print('Run search')
else:
        print('Run list')

Answer 1

您需要查看action文档中argparse参数的含义。

parser.add_argument('-d', '--delete', action='delete', help='Delete ID')
parser.add_argument('-s', '--search', action='search', help='Search ID')
parser.add_argument('-l', '--list', action='list', help='List all ID')

这些不是有效值。有效的字符串是'store'（默认），'append'，'store_true'。它们指的是定义的argparse动作，而不是您的功能。

如果我将解析器更改为：

parser = argparse.ArgumentParser(description='Example list of options')
parser.add_argument('-d', '--delete', help='Delete ID')
parser.add_argument('-s', '--search', help='Search ID')
parser.add_argument('-l', '--list', action='store_true',help='List all ID')
args = parser.parse_args()

这些输入将产生一个显示如下的参数：

nachos.py --delete 20394739
nachos.py -d 20394739
args.delete # '20394739'

nachos.py --search 30459389
nachos.py -s 30459389
args.search # '30459389'

nachos.py --list
nachos.py -l
args.list   # True

您需要在解析后检查这些值以确定要采取的操作。 argparse计算出用户想要的内容。采取行动是你的责任。

在学习和调试时，这是一个好主意

print(args)

In [194]: print(parser.parse_args('--delete 2039 --search 304 -l'.split()))
Namespace(delete='2039', list=True, search='304')

In [195]: print(parser.parse_args('-d 2039'.split()))
Namespace(delete='2039', list=False, search=None)

查看解析如何与输入匹配。

有更详细的方法可以执行此操作，例如subparsers和set_defaults，但我认为您需要首先了解基本的argparse工作方式。

subparsers部分下的argparse文档讨论了如何调度命令(subcommand setdefaults)。以下是使用store_const的变体。

In [2]: parser=argparse.ArgumentParser();
In [3]: parser.add_argument('-d')
In [4]: def foo(args):
   ...:     print(args)
In [5]: parser.add_argument('-f',action='store_const',const=foo)

使用store_const，您可以在args属性中放置一个函数，而不仅仅是它的字符串名称。我可以在default参数中添加不同的函数。

In [6]: args=parser.parse_args('-d 1234 -f'.split())
In [7]: print(args)
Namespace(d='1234', f=<function foo at 0xac24ca4c>)

请注意，args.f现在是一个函数，而不是字符串。

In [8]: if args.f is not None:
   ...:     args.f(args)
   ...:     
Namespace(d='1234', f=<function foo at 0xac24ca4c>)

Answer 2

调用程序中指定的参数时，

...
parser.add_argument('-d', '--delete', action='delete', help='DeleteID')
...

使用该选项调用程序的选项语法是

nachos.py -d 20394739
# OR
nachos.py --delete 20394739

Answer 3

add_argument()的action参数不带任意字符串值。相反，它为您提供了预定义操作的选择。在您的情况下，您可能想要执行类似

的操作

parser = argparse.ArgumentParser(description='Example list of options')
parser.add_argument('-d', '--delete', dest='command', action='store_const', const='delete', help='Delete ID')
parser.add_argument('-s', '--search', dest='command', action='store_const', const='search', help='Search ID')
parser.add_argument('-l', '--list', dest='command', action='store_const', const='list', help='List all ID')
args = parser.parse_args()

然后，您可以访问args.command以检查用户选择的操作。

if args.command == 'delete':
     ...
elif args.command == 'search':
     ...

如果有很多功能，还可以：

def handle_delete(args):
    pass
# ... as above
args = parser.parse_args()
eval('handle_' + args.command)(args)

eval在当前模块中查找名称handle_<command>，然后调用返回的对象。因此，对于delete命令，我们调用handle_delete(args)。请注意，您必须注意command不是任意Python代码。

另一种方法是覆盖提供argparse.Action的子类并覆盖每个操作的__call__()方法：

class DeleteAction(argparse.Action):
    def __call__(self, parser, namespace, values, option_string=None):
        for value in values:
            print('Deleting {}'.format(value))

parser = argparse.ArgumentParser()
parser.add_argument('-d', nargs=1, action=DeleteAction)

此处values将包含-d的参数。但请注意，在解析过程中将调用__call__()。

Python 3 argparse调用函数

3 个答案: