Python3 Click模塊的使用方法詳解
Click 是 Flask 的團隊 pallets 開發(fā)的優(yōu)秀開源項目，它為命令行工具的開發(fā)封裝了大量方法，使開發(fā)者只需要專注于功能實現(xiàn)。恰好我最近在開發(fā)的一個小工具需要在命令行環(huán)境下操作，就寫個學習筆記。
國際慣例，先來一段 “Hello World” 程序（假定已經(jīng)安裝了 Click 包）。
# hello.py
import click
@click.command()
@click.option('--count', default=1, help='Number of greetings.')
@click.option('--name', prompt='Your name',
help='The person to greet.')
def hello(count, name):
"""Simple program that greets NAME for a total of COUNT times."""
for x in range(count):
click.echo('Hello %s!' % name)
if __name__ == '__main__':
hello()
執(zhí)行 python hello.py --count=3，不難猜到控制臺的輸出結(jié)果。除此之外，Click 還悄悄地做了其他的工作，比如幫助選項：
$ python hello.py --help Usage: hello.py [OPTIONS] Simple program that greets NAME for a total of COUNT times. Options: --count INTEGER Number of greetings. --name TEXT The person to greet. --help Show this message and exit.
函數(shù)秒變 CLI
從上面的 “Hello World” 演示中可以看出，Click 是通過裝飾器來把一個函數(shù)方法裝飾成命令行接口的，這個裝飾器方法就是 `@click.command()`。
import click
@click.command()
def hello():
click.echo('Hello World!')
`@click.command()裝飾器把hello()方法變成了Command對象，當它被調(diào)用時，就會執(zhí)行該實例內(nèi)的行為。而–help參數(shù)就是Command` 對象內(nèi)置的參數(shù)。
不同的 Command 實例可以關聯(lián)到 group 中。group 下綁定的命令就成為了它的子命令，參考下面的代碼：
@click.group()
def cli():
pass
@click.command()
def initdb():
click.echo('Initialized the database')
@click.command()
def dropdb():
click.echo('Dropped the database')
cli.add_command(initdb)
cli.add_command(dropdb)
`@click.group裝飾器把方法裝飾為可以擁有多個子命令的Group對象。由Group.add_command()方法把Command對象關聯(lián)到Group對象。 也可以直接用@Group.command裝飾方法，會自動把方法關聯(lián)到該Group` 對象下。
@click.group()
def cli():
pass
@cli.command()
def initdb():
click.echo('Initialized the database')
@cli.command()
def dropdb():
click.echo('Dropped the database')
命令行的參數(shù)是不可或缺的，Click 支持對 command 方法添加自定義的參數(shù)，由 option() 和 argument() 裝飾器實現(xiàn)。
@click.command()
@click.option('--count', default=1, help='number of greetings')
@click.argument('name')
def hello(count, name):
for x in range(count):
click.echo('Hello %s!' % name)
打包跨平臺可執(zhí)行程序
通過 Click 編寫了簡單的命令行方法后，還需要把 .py 文件轉(zhuǎn)換成可以在控制臺里運行的命令行程序。最簡單的辦法就是在文件末尾加上如下代碼：
if __name__ == '__main__': command()
Click 支持使用 setuptools 來更好的實現(xiàn)命令行程序打包，把源碼文件打包成系統(tǒng)中的可執(zhí)行程序，并且不限平臺。一般我們會在源碼根目錄下創(chuàng)建 setup.py 腳本，先看一段簡單的打包代碼：
from setuptools import setup
setup(
name='hello',
version='0.1',
py_modules=['hello'],
install_requires=[
'Click',
],
entry_points='''
[console_scripts]
hello=hello:cli
''',
)
留意 entry_points 字段，在 console_scripts 下，每一行都是一個控制臺腳本，等號左邊的的是腳本的名稱，右邊的是 Click 命令的導入路徑。
詳解命令行參數(shù)
上面提到了自定義命令行參數(shù)的兩個裝飾器：`@click.option()和@click.argument()`，兩者有些許區(qū)別，使用場景也有所不同。
總體而言，argument() 裝飾器比 option() 功能簡單些，后者支持下面的特性：
- 自動提示缺失的輸入；
- option 參數(shù)可以從環(huán)境變量中獲取，argument 參數(shù)則不行；
- option 參數(shù)在 help 輸出中有完整的文檔，argument 則沒有；
而 argument 參數(shù)可以接受可變個數(shù)的參數(shù)值，而 option 參數(shù)只能接收固定個數(shù)的參數(shù)值（默認是 1 個）。
Click 可以設置不同的參數(shù)類型，簡單類型如 click.STRING，click.INT，click.FLOAT，click.BOOL。
命令行的參數(shù)名由 “-short_name” 和 “–long_name” 聲明，如果參數(shù)名既沒有以 “-“ 開頭，也沒有以 “–” 開頭，那么這邊變量名會成為被裝飾方法的內(nèi)部變量，而非方法參數(shù)。
Option 參數(shù)
option 最基礎的用法就是簡單值變量，option 接收一個變量值，下面是一段示例代碼：
@click.command()
@click.option('--n', default=1)
def dots(n):
click.echo('.' * n)
如果在命令行后面跟隨參數(shù) --n=2 就會輸出兩個點，如果傳參數(shù)，默認輸出一個點。上面的代碼中，參數(shù)類型沒有顯示給出，但解釋器會認為是 INT 型，因為默認值 1 是 int 值。
有些時候需要傳入多個值，可以理解為一個 list，option 只支持固定長度的參數(shù)值，即設置后必須傳入，個數(shù)由 nargs 確定。
@click.command()
@click.option('--pos', nargs=2, type=float)
def findme(pos):
click.echo('%s / %s' % pos)
findme --pos 2.0 3.0 輸出結(jié)果就是 2.0 / 3.0
既然可以傳入 list，那么 tuple 呢？Click 也是支持的：
@click.command()
@click.option('--item', type=(unicode, int))
def putitem(item):
click.echo('name=%s id=%d' % item)
這樣就傳入了一個 tuple 變量，putitem --item peter 1338 得到的輸出就是 name=peter id=1338
上面沒有設置 nargs，因為 nargs 會自動取 tuple 的長度值。因此上面的代碼實際上等同于：
@click.command()
@click.option('--item', nargs=2, type=click.Tuple([unicode, int]))
def putitem(item):
click.echo('name=%s id=%d' % item)
option 還支持同一個參數(shù)多次使用，類似 git commit -m aa -m bb 中 -m 參數(shù)就傳入了 2 次。option 通過 multiple 標識位來支持這一特性：
@click.command()
@click.option('--message', '-m', multiple=True)
def commit(message):
click.echo('\n'.join(message))
有時候，命令行參數(shù)是固定的幾個值，這時就可以用到 Click.choice 類型來限定傳參的潛在值：
# choice
@click.command()
@click.option('--hash-type', type=click.Choice(['md5', 'sha1']))
def digest(hash_type):
click.echo(hash_type)
當上面的命令行程序參數(shù) --hash-type 不是 md5 或 sha1，就會輸出錯誤提示，并且在 --help 提示中也會對 choice 選項有顯示。
如果希望命令行程序能在我們錯誤輸入或漏掉輸入的情況下，友好的提示用戶，就需要用到 Click 的 prompt 功能，看代碼：
# prompt
@click.command()
@click.option('--name', prompt=True)
def hello(name):
click.echo('Hello %s!' % name)
如果在執(zhí)行 hello 時沒有提供 –name 參數(shù)，控制臺會提示用戶輸入該參數(shù)。也可以自定義控制臺的提示輸出，把 prompt 改為自定義內(nèi)容即可。
對于類似賬戶密碼等參數(shù)的輸入，就要進行隱藏顯示。option 的 hide_input 和 confirmation_promt 標識就是用來控制密碼參數(shù)的輸入：
# password
@click.command()
@click.option('--password', prompt=True, hide_input=True,
confirmation_prompt=True)
def encrypt(password):
click.echo('Encrypting password to %s' % password.encode('rot13'))
Click 把上面的操作進一步封裝成裝飾器 click.password_option()，因此上面的代碼也可以簡化成：
# password
@click.command()
@click.password_option()
def encrypt(password):
click.echo('Encrypting password to %s' % password.encode('rot13'))
有的參數(shù)會改變命令行程序的執(zhí)行，比如 node 是進入 Node 控制臺，而 node --verion 是輸出 node 的版本號。Click 提供 eager 標識對參數(shù)名進行標記，攔截既定的命令行執(zhí)行流程，而是調(diào)用一個回調(diào)方法，執(zhí)行后直接退出。下面模擬 click.version_option() 的功能，實現(xiàn) --version 參數(shù)名輸出版本號：
# eager
def print_version(ctx, param, value):
if not value or ctx.resilient_parsing:
return
click.echo('Version 1.0')
ctx.exit()
@click.command()
@click.option('--version', is_flag=True, callback=print_version,
expose_value=False, is_eager=True)
def hello():
click.echo('Hello World!')
對于類似刪除數(shù)據(jù)庫表這樣的危險操作，Click 支持彈出確認提示，--yes 標識位置為 True 時會讓用戶再次確認：
# yes parameters
def abort_if_false(ctx, param, value):
if not value:
ctx.abort()
@click.command()
@click.option('--yes', is_flag=True, callback=abort_if_false,
expose_value=False,
prompt='Are you sure you want to drop the db?')
def dropdb():
click.echo('Dropped all tables!')
測試運行下：
$ dropdb Are you sure you want to drop the db? [y/N]: n Aborted! $ dropdb --yes Dropped all tables!
同樣的，Click 對次進行了封裝，click.confirmation_option() 裝飾器實現(xiàn)了上述功能：
@click.command()
@click.confirmation_option(prompt='Are you sure you want to drop the db?')
def dropdb():
click.echo('Dropped all tables!')
前面只講了默認的參數(shù)前綴 -- 和 -，Click 允許開發(fā)者自定義參數(shù)前綴（雖然嚴重不推薦）。
# other prefix
@click.command()
@click.option('+w/-w')
def chmod(w):
click.echo('writable=%s' % w)
if __name__ == '__main__':
chmod()
如果想要用 / 作為前綴，而且要像上面一樣采用布爾標識，會產(chǎn)生沖突，因為布爾標識也是用 /，這種情況下可以用 ; 代替布爾標識的 /：
@click.command()
@click.option('/debug;/no-debug')
def log(debug):
click.echo('debug=%s' % debug)
if __name__ == '__main__':
log()
既然支持 Choice，不難聯(lián)想到 Range，先看代碼：
# range
@click.command()
@click.option('--count', type=click.IntRange(0, 20, clamp=True))
@click.option('--digit', type=click.IntRange(0, 10))
def repeat(count, digit):
click.echo(str(digit) * count)
if __name__ == '__main__':
repeat()
Argument 參數(shù)
Argument 的作用類似 Option，但沒有 Option 那么全面的功能。
和 Option 一樣，Argument 最基礎的應用就是傳遞一個簡單變量值：
@click.command()
@click.argument('filename')
def touch(filename):
click.echo(filename)
命令行后跟的參數(shù)值被賦值給參數(shù)名 filename。
另一個用的比較廣泛的是可變參數(shù)，也是由 nargs 來確定參數(shù)個數(shù)，變量值會以 tuple 的形式傳入函數(shù)：
@click.command()
@click.argument('src', nargs=-1)
@click.argument('dst', nargs=1)
def copy(src, dst):
for fn in src:
click.echo('move %s to folder %s' % (fn, dst))
運行程序：
$ copy foo.txt bar.txt my_folder move foo.txt to folder my_folder move bar.txt to folder my_folder
Click 支持通過文件名參數(shù)對文件進行操作，click.File() 裝飾器就是處理這種操作的，尤其是在類 Unix 系統(tǒng)下，它支持以 - 符號作為標準輸入/輸出。
# File
@click.command()
@click.argument('input', type=click.File('rb'))
@click.argument('output', type=click.File('wb'))
def inout(input, output):
while True:
chunk = input.read(1024)
if not chunk:
break
output.write(chunk)
運行程序，先將文本寫進文件，再讀取
$ inout - hello.txt hello ^D $ inout hello.txt - hello
如果參數(shù)值只是想做為文件名而已呢，很簡單，將 type 指定為 click.Path()：
@click.command()
@click.argument('f', type=click.Path(exists=True))
def touch(f):
click.echo(click.format_filename(f))
$ touch hello.txt
hello.txt
$ touch missing.txt
Usage: touch [OPTIONS] F
Error: Invalid value for "f": Path "missing.txt" does not exist.
更多關于Python3 Click模塊的使用方法請查看下面的相關鏈接
相關文章
今天給大伙分享一下 Python 爬蟲的教程，這次主要涉及到的是關于某 APP 的逆向分析并抓取數(shù)據(jù)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-05-05
在Python中使用循環(huán)進行迭代的方法小結(jié)
Python中的循環(huán)結(jié)構(gòu)是編程中的重要組成部分,本文詳細介紹這兩種循環(huán)的使用方法、它們之間的差異以及如何選擇合適的循環(huán)類型,此外,我還將介紹一些高級循環(huán)控制技巧,如列表推導式和生成器表達式,感興趣的朋友一起看看吧2024-01-01
這篇文章主要介紹了Python字符串中如何去除數(shù)字之間的逗號，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2022-05-05
mitmproxy 是一款工具，也可以說是 python 的一個包，在命令行操作的工具。這篇文章主要介紹了詳解安裝mitmproxy以及遇到的坑和簡單用法，感興趣的小伙伴們可以參考一下2019-01-01
在本篇文章里小編給各位整理的是關于用vue.js組件模擬v-model指令實例方法和相關代碼，需要的朋友們可以跟著學習下。2019-07-07
這篇文章主要介紹了詳解Python 小數(shù)據(jù)池和代碼塊緩存機制的相關資料，幫助大家更好的理解和學習使用python，感興趣的朋友可以了解下2021-04-04
Python計算三角函數(shù)之a(chǎn)sin()方法的使用
這篇文章主要介紹了Python計算三角函數(shù)之a(chǎn)sin()方法的使用,是Python入門的基礎知識,需要的朋友可以參考下2015-05-05
大家都知道python中的注釋有多種，有單行注釋，多行注釋，批量注釋，中文注釋也是常用的。python注釋也有自己的規(guī)范，這篇文章文章中會給大家詳細介紹Pyhton中單行和多行注釋的使用方法及規(guī)范，有需要朋友們可以參考借鑒。2016-10-10
當我們用python進行數(shù)據(jù)處理時會遇到很多缺失值，對缺失值我們需要進行刪除或者填補，本文主要介紹了pandas中NaN缺失值的處理方法，感興趣的可以了解一下2021-05-05
python不相等的兩個字符串的 if 條件判斷為True詳解
這篇文章主要介紹了python不相等的兩個字符串的 if 條件判斷為True詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-03-03