python使用argparse模块实现在终端命令行传参

直接上代码

import argparse
# 定义终端要传送的参数
parser = argparse.ArgumentParser(description="A description of what the program does")
parser.add_argument('--num_send', '-n', default=100, type=int, help='Number of sends.')
parser.add_argument("--time_sleep", '-s', default=0.1, type=float, help="time selp.")
parser.add_argument("--json_file_index", '-j', default=1, type=int, help=" index of json file.")
parser.add_argument("--thread_num", '-t', default=1, type=int, help="index of thread.")
parser.add_argument("--message_len", '-m', default=1, type=int, help="len of message.")
parser.add_argument("--message", '-ms', default="a", type=str, help="the message.")
parser.add_argument("--device_num", '-d', default=500, type=int, help="num of device.")
args = parser.parse_args()

def message_genr():
    message = ""
    for i in range(args.message_len):
        message += args.message
    return message

在后续代码中使用传入的参数形式为：args.参数名，比如上述代码中的

args.message_len
args.message

假如我上面的代码文件名称是test.py，那么在终端执行test.py并向其传参数的写法为：

$ python3 test.py --num_send=50 --time_sleep=0.1

也可写为：

$ python3 test.py -n 50 -s 0.1

如果有的参数命令行没有写，就按默认值default来，比如：–thread_num的默认值为1

Done!!!