如何使用argparse向python函数传递参数？

Question

我有一个食品使用argparse获取main（）函数的命令行参数。在

""" foo.py """
import argparse

def main():
    parser = argparse.ArgumentParser(description='test')
    parser.add_argument('--all', '-a', action='store_true', help='all')
    args = parser.parse_args()

    if args.all:
        pass

if __name__ == '__main__':
    main()

我必须在另一个Python脚本上测试这个main（）函数棒.py. 我的问题是如何传递参数棒.py. 我目前的解决方案是改变系统argv. 寻找更好的解决方案。在

^{pr2}$

Answer 1

您可以修改main函数以接收参数列表。在

""" foo.py """
import argparse

def main(passed_args=None):
    parser = argparse.ArgumentParser(description='test')
    parser.add_argument(' all', '-a', action='store_true', help='all')
    args = parser.parse_args(passed_args)

    if args.all:
        pass

if __name__ == '__main__':
    main()

""" bar.py """
import sys
import foo

if __name__ == '__main__':
    foo.main(["-a"])