Python实现解析命令行参数的常见方法总结

# -*- coding: utf-8 -*-
# time: 2022/9/1 0:10
# file: args_demo.py
# 公众号: 玩转测试开发
import sys


def f1():
    print(f"正在运行的脚本名称: {sys.argv[0]}")
    print(f"脚本的参数数量: {len(sys.argv)}")
    print(f"脚本的参数: {sys.argv}")


if __name__ == '__main__':
    f1()

# -*- coding: utf-8 -*-
# time: 2022/9/1 0:10
# file: args_demo.py
# 公众号: 玩转测试开发
import sys
import getopt


def f1():
    print(f"正在运行的脚本名称: {sys.argv[0]}")
    print(f"脚本的参数数量: {len(sys.argv)}")
    print(f"脚本的参数: {sys.argv}")


def f2(argv):
    input_file = ""
    output_file = ""
    opts, args = getopt.getopt(argv[1:], "hi:o:", ["help", "input_file=", "output_file="])

    # "hi:o:": 短格式分析串, h 后面没有冒号, 表示后面不带参数; i 和 o 后面带有冒号, 表示后面带参数
    # ["help", "input_file=", "output_file="]: 长格式分析串列表, help后面没有等号, 表示后面不带参数; input_file和output_file后面带冒号, 表示后面带参数
    # 返回值包括 `opts` 和 `args`, opts 是以元组为元素的列表, 每个元组的形式为: (选项, 附加参数)，如: ('-i', 'test.png');
    # args是个列表，其中的元素是那些不含'-'或'--'的参数

    for opt, arg in opts:
        if opt in ("-h", "--help"):
            # 即：处理 -h参数的案例说明，帮助文档等
            print('args_demo.py -i <input_file> -o <output_file>')
            print('or: args_demo.py --input_file=<input_file> --output_file=<output_file>')
            sys.exit()
        elif opt in ("-i", "--input_file"):
            input_file = arg
        elif opt in ("-o", "--output_file"):
            output_file = arg
    print(f'输入文件为：{input_file}')
    print(f'输出文件为：{output_file}')

    # 打印不含'-'或'--'的参数
    for i in range(0, len(args)):
        print(f'不含 - 或 -- 的参数 {i + 1} 为：{args[i]}')


if __name__ == '__main__':
    # f1()
    f2(sys.argv)

python args_demo.py -h
python args_demo.py -i a.txt
python args_demo.py -o b.txt
python args_demo.py -i a.txt  -o b.txt c.txt

import argparse
parser = argparse.ArgumentParser()
parser.parse_args()

python argparse_demo.py -q
usage: argparse_demo.py [-h]
argparse_demo.py: error: unrecognized arguments: -q

# -*- coding: utf-8 -*-
# time: 2022/8/31 16:20
# file: argparse_demo1.py
# author: tom
import argparse

parser = argparse.ArgumentParser()
parser.add_argument("name", help="添加的第1个参数name，姓名。")
args = parser.parse_args()
print(f"args.name：{args.name}")
print(f"type：{type(args.name)}")

python argparse_demo1.py tom
python argparse_demo1.py -h
python argparse_demo1.py

# -*- coding: utf-8 -*-
# time: 2022/8/31 16:20
# file: argparse_demo1.py
# author: tom
import argparse

parser = argparse.ArgumentParser()
parser.add_argument("--name", help="添加的第1个参数name，姓名。")
parser.add_argument("--age", help="添加的第2个参数age，年龄。", type=int)
args = parser.parse_args()
print(f"args.name：{args.name}")
print(f"type：{type(args.name)}")

print("*" * 50)

print(f"args.age：{args.age}")
print(f"type：{type(args.age)}")

python argparse_demo.py --name=tom --age=30
python argparse_demo.py --name tom --age 30