首頁 > 軟體

Python利用裝飾器click處理解析命令列引數

2022-10-23 18:00:03

簡介

在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解析命令列引數內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com