python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python装饰器click解析命令行参数

Python利用装饰器click处理解析命令行参数

作者:玩转测试开发

这篇文章主要为大家详细介绍了Python如何利用装饰器click实现处理解析命令行参数功能,文中的示例代码简洁易懂,需要的小伙伴快跟随小编一起了解一下

简介

在Python开发和测试过程中主要有两种模式可以选择:脚本模式、命令行模式。在代码的开发和调试过程中使用脚本模式很方便,目前比较主流的命令行工具主要有sys.argv、getopt、argparse,也可以使用装饰器的方式来解析命令行参数,即使用click模块。

基本形式

python main.py -a v1 -b v2 ...

历史攻略

Python:解析命令行参数

之前的案例

argparse 将提供的选项默认视为字符串,新增int类型的案例

# -*- 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)}")

改造成click装饰

层次更清晰、明了。

# -*- coding: utf-8 -*-
# time: 2022/9/4 16:09
# file: cmd_demo.py
# 公众号: 玩转测试开发
import click


@click.command()
@click.option('--name', prompt='姓名:', help='姓名')
@click.option('--age', prompt='年龄:', default=25, help='年龄,默认值25')
def args_data(name, age):
    print(f"name:{name}")
    print(f"type:{type(name)}")
    print("*" * 80)
    print(f"age:{age}")
    print(f"type:{type(age)}")


if __name__ == '__main__':
    args_data()

依次执行:

python cmd_demo.py --help
python cmd_demo.py --name=tom --age=30
python cmd_demo.py --name ken --age 20

执行结果:

(venv) D:\>python cmd_demo.py --help
Usage: cmd_demo.py [OPTIONS]

Options:
  --name TEXT    姓名
  --age INTEGER  年龄,默认值25
  --help         Show this message and exit.

(venv) D:\>python cmd_demo.py --name=tom --age=30
name:tom
type:<class 'str'>
*************************************************
age:30
type:<class 'int'>

(venv) D:\>python cmd_demo.py --name ken --age 20
name:ken
type:<class 'str'>
*************************************************
age:20
type:<class 'int'>

到此这篇关于Python利用装饰器click处理解析命令行参数的文章就介绍到这了,更多相关Python装饰器click解析命令行参数内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:
阅读全文