Question

我有一個foo.py使用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腳本bar.py上測試這個main（）函數。我的問題是如何在bar.py中傳遞參數。我目前的解決方案是改變sys.argv。尋找更好的解決方案。

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

if __name__ == '__main__': 
    sys.argv.append('-a') 
    foo.main() 

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') 
    if passed_args: 
     args = parser.parse_args(passed_args) 
    else: 
     args = parser.parse_args() 

    if args.all: 
     pass 

if __name__ == '__main__': 
    main() 

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

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