快捷導(dǎo)航

Python利用裝飾器click處理解析命令行參數(shù)

更新時(shí)間：2022年10月23日 10:53:01 作者：玩轉(zhuǎn)測試開發(fā)

這篇文章主要為大家詳細(xì)介紹了Python如何利用裝飾器click實(shí)現(xiàn)處理解析命令行參數(shù)功能，文中的示例代碼簡潔易懂，需要的小伙伴快跟隨小編一起了解一下

簡介

在Python開發(fā)和測試過程中主要有兩種模式可以選擇：腳本模式、命令行模式。在代碼的開發(fā)和調(diào)試過程中使用腳本模式很方便，目前比較主流的命令行工具主要有sys.argv、getopt、argparse，也可以使用裝飾器的方式來解析命令行參數(shù)，即使用click模塊。

基本形式

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

歷史攻略

Python：解析命令行參數(shù)

之前的案例

argparse 將提供的選項(xiàng)默認(rèn)視為字符串，新增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個(gè)參數(shù)name，姓名。")
parser.add_argument("--age", help="添加的第2個(gè)參數(shù)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
# 公眾號: 玩轉(zhuǎn)測試開發(fā)
import click


@click.command()
@click.option('--name', prompt='姓名：', help='姓名')
@click.option('--age', prompt='年齡：', default=25, help='年齡，默認(rèn)值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()

依次執(zhí)行：

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

執(zhí)行結(jié)果：

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

Options:
--name TEXT 姓名
--age INTEGER 年齡，默認(rèn)值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'>