Question

我刚刚开始使用argparse，我的代码可以正常工作，但我试图用不同的值来解析用户如何输入参数。

目前这是我的代码：

def setup_args():
    """
    Set up args for the tool
    """
    parser = argparse.ArgumentParser(
        description=("Get all file versions of a status in a project"),
        formatter_class=argparse.RawDescriptionHelpFormatter)

    # Positional Arguments
    parser.add_argument('project',
                        type=str,
                        help='Name of the to look into')

    parser.add_argument('status',
                        type=str,
                        help='Define which status to look into')

    # Optional Arguments
    parser.add_argument('-o',
                        '--output',
                        action='store_true',
                        help='Write to output to text file if used')

if __name__ == "__main__":
    args = setup_args()

    # Check the status set
    status_list = ['Pending', 'Work in Progress', 'Approved', 'Rejected']
    if not args.status in site_list:
        raise ValueError("Please input one of the status : 'Pending', 'Work in Progress', 'Approved', 'Rejected'")

    output_query(args.project, status, args.client, args.output)

正如您在main中看到的那样..它只注册我定义的那些区分大小写的状态名称。有没有什么方法可以让我的代码在小型大写字母中输入？'pending', 'work in progress', 'approved', 'rejected'或简短形式 - 'p', 'wip', 'a', 'r'？

我可以实现的一种方法是使用if..elif..

if args.client == ('pending' or 'p'):
    args.client = 'Pending'
elif args.client == ('work in progress' or 'wip'):
    args.client = 'Work in Progress'
elif args.client == ('approved' or 'a'):
    args.client = 'Approved'
elif args.client == ('rejected' or 'r'):
    args.client = 'Rejected'

虽然它有效但看起来有点啰嗦＆＃39;对我来说。如果我有多个参数，这意味着我需要输入很多if...elif...这可能不实用，除非这是唯一的方法。有没有更好的解决方案来解决这个问题？

编辑：

这就是我运行命令的方式：python prog.py my_project Pending 但是我想的是可以用这种方式输入的方案：python prog.py my_project pending或python prog.py my_project p，注意大写字母P已成为小写字母..

Answer 1

您可以使用status推广.lower()支票，并限制您检查的字符数。

例如，如果我定义了＆＃39;选项的缩略列表，我可以测试看起来像大牌的任何内容：

In [239]: choices = ['pend', 'work', 'appr','reje']
In [240]: status_list = ['Pending', 'Work in Progress', 'Approved', 'Rejected']
In [241]: for wd in status_list:
     ...:     if wd.lower()[:4] in choices:
     ...:         print(wd)
     ...:         
Pending
Work in Progress
Approved
Rejected

您可能不希望您的用户输入完整的“正在进行的工作”＃39;串。这样做需要引用。否则shell会将其分成3个字符串。

此测试的变体使用startswith：

for wd in status_list:
     if any([wd.lower().startswith(n) for n in choices]):
         print(wd)

你也可以让解析器进行值检查

parser.add_argument('status',
                    # type=str,   # default, don't need to add it
                    choices = ['pending', 'work', 'approved', 'rejected'],
                    help='Define which status to look into')

如果字符串不匹配，则会生成一条很好的错误消息。它将选择纳入帮助。试一试，看看会发生什么。

缺点是它不允许缩写或大/小写。（自定义type函数可以绕过这些限制，但这是一种更高级的技术。）

==================

使用type的方法是定义一个小函数：

def abrev(astr):
    return astr.lower()[:4]

适用于上述测试：

for wd in status_list:
    if abrev(wd) in choices:
        print(wd)

在解析器中，它可以用作：

In [253]: p = argparse.ArgumentParser()
In [254]: p.add_argument('status', type=abrev, choices=choices);

In [255]: p.print_help()
usage: ipython3 [-h] {pend,work,appr,reje}

positional arguments:
  {pend,work,appr,reje}

optional arguments:
  -h, --help            show this help message and exit

示例电话：

In [256]: p.parse_args(['Work'])
Out[256]: Namespace(status='work')

In [257]: p.parse_args(['status'])
usage: ipython3 [-h] {pend,work,appr,reje}
ipython3: error: argument status: invalid choice: 'stat' (choose from 'pend', 'work', 'appr', 'reje')
...

In [258]: p.parse_args(['reject'])
Out[258]: Namespace(status='reje')

In [259]: p.parse_args(['Pending'])
Out[259]: Namespace(status='pend')

Answer 2

虽然我没有对它进行测试，但查看add_argument函数文档，您应该能够将它们作为第一个参数添加到它。

name或flags - 选项字符串的名称或列表，例如foo或-f， - foo。

https://docs.python.org/3/library/argparse.html#the-add-argument-method

定义不同形式的argparse参数输入

2 个答案: