Python進階:都說好用的 Python 命令行庫click


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

它旨在讓編寫命令行工具的過程既快速又有趣,還能防止由於無法實現預期的 CLI API 所產生挫敗感。它有如下三個特點:

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

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 的作用相似,點擊進入,免費領取python學習資料,但功能更加強大,能處理好 Unicode 和 二進制數據的情況。

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):

  

在上面的示例中:

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

不論是裝飾器的方式、還是各種默認行為,click 都是像它的介紹所說的那樣,讓人盡可能少地編寫代碼,讓整個過程變得快速而有趣。

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.
 

  

小結

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

click 除了以 Pythonic 的方式讓命令行程序的實現變得更加優雅和好用外,還提供了比 argparse 和 docopt 都要強大的功能。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM