你們都說好用的 Python 命令行庫：click

做者：HelloGitHub-Prodesire

HelloGitHub 的《講解開源項目》系列，項目地址：https://github.com/HelloGitHub-Team/Article

1、前言

在本系列前面幾篇文章中，咱們分別介紹了 argparse 和 docopt 的主要功能和用法。它們各具特點，都能出色地完成命令行任務。argparse 是面向過程的，須要先設置解析器，再定義參數，再解析命令行，最後實現業務邏輯。而 docopt 先用聲明式的語法定義出參數，再過程式地解析命令行和實現業務邏輯。在一些人看來，這些方式都不夠優雅。

而今天要介紹的 click 則是用一種你很熟知的方式來玩轉命令行。命令行程序本質上是定義參數和處理參數，而處理參數的邏輯必定是與所定義的參數有關聯的。那可不能夠用函數和裝飾器來實現處理參數邏輯與定義參數的關聯呢？而 click 正好就是以這種方式來使用的。

本系列文章默認使用 Python 3 做爲解釋器進行講解。
若你仍在使用 Python 2，請注意二者之間語法和庫的使用差別哦~

2、介紹

click 是一個以儘量少的代碼、以組合的方式建立優美的命令行程序的 Python 包。它有很高的可配置性，同時也能開箱即用。

它旨在讓編寫命令行工具的過程既快速又有趣，還能防止因爲沒法實現預期的 CLI API 所產生挫敗感。它有以下三個特色：

• 任意嵌套命令
• 自動生成幫助
• 支持運行時延遲加載子命令

3、快速開始

3.1 業務邏輯

首先定義業務邏輯，是否是感受到有些難以置信呢？

不管是 argparse 仍是 docopt，業務邏輯都是被放在最後一步，但 click 倒是放在第一步。細想一想 click 的這種方式才更符合人的思惟吧？不論用什麼命令行框架，咱們最終關心的就是實現業務邏輯，其它的能省則省。

咱們以官方示例爲例，來介紹 click 的用法和哲學。假設命令行程序的輸入是 name 和 count，功能是打印指定次數的名字。

那麼在 hello.py 中，很容易寫出以下代碼：

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)

這段代碼的邏輯很簡單，就是循環 count 次，使用 click.echo 打印 name。其中，click.echo 和 print 的做用類似，但功能更增強大，能處理好 Unicode 和 二進制數據的狀況。

3.2 定義參數

很顯然，咱們須要針對 count 和 name 來定義它們所對應的參數信息。

• count 對應爲命令行選項 --count，類型爲數字，咱們但願在不提供參數時，其默認值是 1
• name 對應爲命令行選項 --name，類型爲字符串，咱們但願在不提供參數時，能給人提示

使用 click，就能夠寫成下面這樣：

from click 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):
    ...

在上面的示例中：

• 使用裝飾器的方式，即定義了參數，又將之與處理邏輯綁定，這真是優雅。和 argparse、docopt 比起來，就少了一步綁定過程
• 使用 click.command 表示 hello 是對命令的處理
• 使用 click.option 來定義參數選項
  • 對於 --count 來講，使用 default 來指定默認值。而因爲默認值是數字，進而暗示 --count 選項的類型爲數字
  • 對於 --name 來講，使用 prompt 來指定未輸入該選項時的提示語
  • 使用 help 來指定幫助信息

不管是裝飾器的方式、仍是各類默認行爲，click 都是像它的介紹所說的那樣，讓人儘量少地編寫代碼，讓整個過程變得快速而有趣。

3.3 代碼梳理

使用 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()

若咱們指定次數和名字：

$ python3 hello.py --count 2 --name Eric
Hello Eric!
Hello Eric!

若咱們什麼都不指定，則會提示輸入名字，並默認輸出一次：

$ python3 hello.py
Your name: Eric
Hello Eric!

咱們還能夠經過 --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.

4、小結

click 的思路很是簡單，定義處理函數，經過它的裝飾器來定義參數。使用裝飾器的絕妙之處就在於把定義和綁定這兩個步驟合爲一個步驟，使得整個過程變得如絲般順滑。

click 除了以 Pythonic 的方式讓命令行程序的實現變得更加優雅和好用外，還提供了比 argparse 和 docopt 都要強大的功能。在接下來幾節中，咱們將會逐步揭開它的面紗。

---

『講解開源項目系列』——讓對開源項目感興趣的人再也不畏懼、讓開源項目的發起者再也不孤單。跟着咱們的文章，你會發現編程的樂趣、使用和發現參與開源項目如此簡單。歡迎留言聯繫咱們、加入咱們，讓更多人愛上開源、貢獻開源～