作者：HellogitHub-Prodesire

涉及的示例代碼，已同步更新到 HelloGitHub-Team 倉(cāng)庫(kù)[1]

一、前言

在上一篇文章中，我們初步掌握了 click 的簡(jiǎn)單用法，并了解到它與 argparse 和 docopt 的不同。接下來(lái)，將深入介紹 click 的各類(lèi)用法，以讓你能輕松打造復(fù)雜的命令行程序。

在概念上， click 把命令行分為 3 個(gè)組成：參數(shù)、選項(xiàng)和命令。

• 參數(shù) 就是跟在命令后的除選項(xiàng)外的內(nèi)容，比如 git add a.txt 中的 a.txt 就是表示文件路徑的參數(shù)
• 選項(xiàng) 就是以 – 或 — 開(kāi)頭的參數(shù)，比如 -f、–File
• 命令 就是命令行的初衷了，比如 git 就是命令，而 git add 中的 add 則是 git 的子命令

本系列文章默認(rèn)使用 Python 3 作為解釋器進(jìn)行講解。若你仍在使用 Python 2，請(qǐng)注意兩者之間語(yǔ)法和庫(kù)的使用差異哦~

二、參數(shù)

2.1 基本參數(shù)

基本參數(shù) 就是通過(guò)位置里指定參數(shù)值。

比如，我們可以指定兩個(gè)位置參數(shù) x 和 y ，先添加的 x 位于第一個(gè)位置，后加入的 y 位于第二個(gè)位置。那么在命令行中輸入 1 2的時(shí)候，分別對(duì)應(yīng)到的就是 x 和 y：

@click.command()@click.argument(‘x’)@click.argument(‘y’)def hello(x, y): print(x, y)

2.2 參數(shù)類(lèi)型

參數(shù)類(lèi)型 就是將參數(shù)值作為什么類(lèi)型去解析，默認(rèn)情況下是字符串類(lèi)型。我們可以通過(guò) type 入?yún)?lái)指定參數(shù)類(lèi)型。

click 支持的參數(shù)類(lèi)型多種多樣：

• str / click.STRING 表示字符串類(lèi)型，這也是默認(rèn)類(lèi)型
• int / click.INT 表示整型
• float / click.FLOAT 表示浮點(diǎn)型
• BOOL / click.BOOL 表示布爾型。很棒之處在于，它會(huì)識(shí)別表示真/假的字符。對(duì)于 1、yes、y 和 true 會(huì)轉(zhuǎn)化為 True；0、no、n 和 false 會(huì)轉(zhuǎn)化為 False
• click.UUID 表示 UUID，會(huì)自動(dòng)將參數(shù)轉(zhuǎn)換為 uuid.UUID 對(duì)象
• click.FILE 表示文件，會(huì)自動(dòng)將參數(shù)轉(zhuǎn)換為文件對(duì)象，并在命令行結(jié)束時(shí)自動(dòng)關(guān)閉文件
• click.PATH 表示路徑
• click.Choice 表示選擇選項(xiàng)
• click.IntRange 表示范圍選項(xiàng)

同 argparse 一樣，click 也支持自定義類(lèi)型，需要編寫(xiě) click.ParamType 的子類(lèi)，并重載 convert 方法。

官網(wǎng)提供了一個(gè)例子，實(shí)現(xiàn)了一個(gè)整數(shù)類(lèi)型，除了普通整數(shù)之外，還接受十六進(jìn)制和八進(jìn)制數(shù)字， 并將它們轉(zhuǎn)換為常規(guī)整數(shù)：

class BasedIntParamType(click.ParamType): name = “integer” def convert(self, value, param, ctx): try: if value[:2].lower() == “0x”: return int(value[2:], 16) elif value[:1] == “0”: return int(value, 8) return int(value, 10) except TypeError: self.fail( “expected string for int() conversion, got ” f”{value!r} of type {type(value).__name__}”, param, ctx, ) except ValueError: self.fail(f”{value!r} is not a valid integer”, param, ctx)BASED_INT = BasedIntParamType()

2.3 文件參數(shù)

在基本參數(shù)的基礎(chǔ)上，通過(guò)指定參數(shù)類(lèi)型，我們就能構(gòu)建出各類(lèi)參數(shù)。

文件參數(shù) 是非常常用的一類(lèi)參數(shù)，通過(guò) type=click.File 指定，它能正確處理所有 Python 版本的 unicode 和 字節(jié)，使得處理文件十分方便。

@click.command()@click.argument(‘input’, type=click.File(‘rb’)) # 指定文件為二進(jìn)制讀@click.argument(‘output’, type=click.File(‘wb’)) # 指定文件為二進(jìn)制寫(xiě)def inout(input, output): while True: chunk = input.read(1024) # 此時(shí) input 為文件對(duì)象，每次讀入 1024 字節(jié) if not chunk: break output.write(chunk) # 此時(shí) output 為文件對(duì)象，寫(xiě)入上步讀入的內(nèi)容

2.4 文件路徑參數(shù)

文件路徑參數(shù) 用來(lái)處理文件路徑，可以對(duì)路徑做是否存在等檢查，通過(guò) type=click.Path 指定。不論文件名是 unicode 還是字節(jié)類(lèi)型，獲取到的參數(shù)類(lèi)型都是 unicode 類(lèi)型。

@click.command()@click.argument(‘filename’, type=click.Path(exists=True)) # 要求給定路徑存在，否則報(bào)錯(cuò)def hello(filename): click.echo(click.format_filename(filename))

如果文件名是以 – 開(kāi)頭，會(huì)被誤認(rèn)為是命令行選項(xiàng)，這個(gè)時(shí)候需要在參數(shù)前加上 — 和空格，比如

$ python hello.py — -foo.txt-foo.txt

2.5 選擇項(xiàng)參數(shù)

選擇項(xiàng)參數(shù) 用來(lái)限定參數(shù)內(nèi)容，通過(guò) type=click.Choice 指定。

比如，指定文件讀取方式限制為 read-only 和 read-write：

@click.command()@click.argument(‘mode’, type=click.Choice([‘read-only’, ‘read-write’]))def hello(mode): click.echo(mode)

2.6 可變參數(shù)

可變參數(shù) 用來(lái)定義一個(gè)參數(shù)可以有多個(gè)值，且能通過(guò) nargs 來(lái)定義值的個(gè)數(shù)，取得的參數(shù)的變量類(lèi)型為元組。

若 nargs=N，N為一個(gè)數(shù)字，則要求該參數(shù)提供 N 個(gè)值。若 N 為 -1 則接受提供無(wú)數(shù)量限制的參數(shù)，如:

@click.command()@click.argument(‘foo’, nargs=-1)@click.argument(‘bar’, nargs=1)def hello(foo, bar): pass

如果要實(shí)現(xiàn) argparse 中要求參數(shù)數(shù)量為 1 個(gè)或多個(gè)的功能，則指定 nargs=-1 且 required=True 即可：

@click.command()@click.argument(‘foo’, nargs=-1, required=True)def hello(foo, bar): pass

2.7 從環(huán)境變量讀取參數(shù)

通過(guò)在 click.argument 中指定 envvar，則可讀取指定名稱(chēng)的環(huán)境變量作為參數(shù)值，比如：

@click.command()@click.argument(‘filename’, envvar=’FILENAME’)def hello(filename): print(filename)

執(zhí)行如下命令查看效果：

$ FILENAME=hello.txt python3 hello.pyhello.txt

而在 argparse 中，則需要自己從環(huán)境變量中讀取。

三、小節(jié)

本文講解了 click 中基本參數(shù)的用法，在此基礎(chǔ)上介紹了各種類(lèi)型的參數(shù)，最后說(shuō)明了從環(huán)境變量中獲取參數(shù)值的寫(xiě)法。

在下一篇文章中，我們來(lái)繼續(xù)深入了解 click 的功能，看看它都支持什么樣的“選項(xiàng)”。

參考資料

[1]HelloGitHub-Team 倉(cāng)庫(kù): https://github.com/HelloGitHub-Team/Article

『講解開(kāi)源項(xiàng)目系列』——讓對(duì)開(kāi)源項(xiàng)目感興趣的人不再畏懼、讓開(kāi)源項(xiàng)目的發(fā)起者不再孤單。跟著我們的文章，你會(huì)發(fā)現(xiàn)編程的樂(lè)趣、使用和發(fā)現(xiàn)參與開(kāi)源項(xiàng)目如此簡(jiǎn)單。歡迎留言聯(lián)系我們、加入我們，讓更多人愛(ài)上開(kāi)源、貢獻(xiàn)開(kāi)源～