使用 argparse 处理 if-else 的最佳方法？

如何解决使用 argparse 处理 if-else 的最佳方法？

假设我们有很多参数解析选项

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Download files from Canvas.')

    parser.add_argument(
        '-user_id',type=str,default="",)

    parser.add_argument(
        '-course_id',)

    parser.add_argument(
        '-student_id',)
# ...
    parser.add_argument(
        '-book_id',)

我想运行由输入决定的函数

if user_id != "" and course_id == "" and student_id != "" and book_id == "":
    f()

elif user_id != "" and course_id == "" and student_id != "" and book_id != "":
    g()

# ...

elif user_id == "" and course_id == "" and student_id == "" and book_id == "":
    h()

等等。好吧，这些是很多 if-else 语句。我想消除必须做所有这些。在运行函数之前，有没有人有足够的方法来减少变量之间的比较量？还是取决于上下文？谢谢！！

解决方法

构建一个将选项组合映射到处理函数的字典。

当你写这个时：

if user_id != "" and course_id == "" and student_id != "" and book_id == "":

您有效地将这四个选项视为布尔值元组 值。你在问，“参数是否等于 (False,True,错，对）？”

我们可以利用这一点。首先，我们需要代码来翻译 选项转换为实际的元组。类似的东西：

args = parser.parse_args()
sig = tuple(getattr(args,x) == '' for x in ('user_id','course_id','student_id','book_id'))

如果你的程序没有收到命令行参数，这会设置 sig 等于 (True,True)。如果您通过 --student-id 1， 这变成了 (True,False,True)。

我们可以创建一个字典来将这些模式映射到一个处理程序 功能：

argmap = {
    (False,True): f,(False,False): g,(True,True): h,}

有了这两种机制，我们可以在处理程序中查找一个 给定一组这样的参数：

handler = argmap.get(sig)
if handler is None:
    print('no handler for',args)
else:
    handler()

综合起来，我们得到：

import argparse


def f():
    print('this is function f')


def g():
    print('this is function g')


def h():
    print('this is function h')


argmap = {
    (False,}


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Download files from Canvas.')

    parser.add_argument(
        '--user-id',default='',)

    parser.add_argument(
        '--course-id',)

    parser.add_argument(
        '--student-id',)

    parser.add_argument(
        '--book-id',)

    args = parser.parse_args()
    sig = tuple(getattr(args,'book_id'))
    handler = argmap.get(sig)
    if handler is None:
        print('no handler for',args)
    else:
        handler()

这产生作为输出：

$ python argtest.py
this is function h
$ python argtest.py --user-id 1 --student-id 1
this is function f
$ python argtest.py --user-id 1 --student-id 1 --book-id 1
this is function g
$ python argtest.py --user-id 1
no handler for Namespace(user_id='1',course_id='',student_id='',book_id='')