当我们需要在命令行或其他编程语言中运行 .py 文件并传入指定参数,编写程序接口的重要性就不言而喻了。传统的 Python 使用系统内置库 Argparse 实现接口定义。Google 于 2017 年底开源了从 TensorFlow 中衍变而来,对 Argparse 的简易封装,使用更为简便。以下便是相关代码实现:
import sys from absl import flags from absl import app flags.DEFINE_string('A','String of A.','Help of A.') flags.DEFINE_integer('B',7,'Help of B.') FLAGS = flags.FLAGS def main(*args): print('Param A = %s'%FLAGS.A) sys.stdout.write('Param B = %s'%FLAGS.B) if __name__=='__main__': #flags.mark_flag_as_required("A") app.run(main)编写一个新的 .py 文件,命名为“test.py”,在命令行将地址调到相应目录下,输入以下命令,看看回应:
>>> test.py Param A = String of A. Param B = 7 >>> test.py --A='New_String_of_A.' Param A = 'New_String_of_A.' Param B = 7 >>> test.py --A='New_String_of_A.' --B=4 Param A = 'New_String_of_A.' Param B = 4 >>> test.py -help -- A: Help of A. (default: 'String of A.') -- B: Help of B. (default: '7') (an integer)使用原始的 Argpase 实现同样的功能,代码如下:
import sys import argparse parser = argparse.ArgumentParser(description='') parse.add_argument('--A', type=str, default='String of A.', help='Help of A.') parse.add_argument('--B', type=int, default='String of A.', help='Help of B.') args = parser.parse_args() def main(): print('Param A = %s'%args.A) sys.stdout.write('Param B = %s'%args.B) if __name__=='__main__': main()