如何在 argparse 中使用条件语句？

Question

所以我创建了我的argparse ，它有两个不同的标志。 一个是-a ，另一个是-b 。 当我使用特定标志运行我的脚本damage.py时，我希望它能够根据传递的标志执行一个函数。 例如，如果我通过damage.py -t ，它将运行函数tester() ，如我的导入所示并打印 hello，如果我通过-d它将运行另一个函数。 到目前为止，我的代码如下：

import argparse


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("-a", "--export-date", action="store_true", required=True)
    parser.add_argument("-b", "--execute-test", action="store_true", required=False)


if __name__ == '__main__':
    main()

Answer 1

您可以像这样直接访问它们，而不是先将这些值保存到变量中：

if args.export_date:
    # Do something with date

if args.execute_test:
    tester()

这意味着当你像python damage.py -dt这样运行你的程序时，它将运行日期块中的代码和测试块中的代码。