python 中的argparse模块——自定义程序参数

官方文档

argparse模块可以在运行程序时传参到程序中，一般的步骤如下：

#导入模块
import argparse

#创建解析器
parser = argparse.ArgumentParser(description = 'parameter')
#通过add_argument()函数添加参数，默认type为str型
parser.add_argument('--batch', type = int, default = 100, help = 'the batch size')
#解析参数
args = parser.parse_args()

使用示例：

#导入模块
import argparse

def parameter():
    #创建解析器
    parser = argparse.ArgumentParser(description = 'parameter')
    #添加参数
    parser.add_argument('--batch', type = int, default = 100, help = 'the batch size')
    parser.add_argument('--expire_time', type = int,  default = 60, help = 'the expire_time')
    parser.add_argument('--table_name', type = str, help = 'the table_name')
    parser.add_argument('--date', type = int, default = date_default, help = 'the date')
    parser.add_argument('--name', type = str, default = 'driver_recall_condition', help = 'the name')
    parser.add_argument('--key', type = str, help = 'the key')
    parser.add_argument('--val', type = str, help = 'the val')

    args = parser.parse_args()
    return args
 
 if __name__=='__main__':
    
    arg = parameter()
    batch = arg.batch
    expire_time = arg.expire_time
    print(expire_time )