Click: 命令列工具神器

wangsys發表於2021-09-09

Click是一個Python用來快速實現命令列應用程式的包,主要優勢表現在以下三點:

  • 任意巢狀命令

  • 自動生成幫助頁

  • 自動執行時lazy載入子命令

示例程式:

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

執行結果:

$ python hello.py --count=3Your name: John
Hello John!
Hello John!
Hello John!

它還會自動生成格式化好的幫助資訊:

$ 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.

快速入門

安裝

pip install Click

建立一個命令

透過裝時器函式click.command()來註冊命令

import click@click.command()def hello():
    click.echo("Hello World!")if __name__ == '__main__':
    hello()

輸出結果:

$ python hello.pyHello World!

相應的幫助頁:

$ python hello.py --help
Usage: hello.py [OPTIONS]

Options:
  --help  Show this message and exit.

輸出

為什麼不用print,還要加入一個echo呢。Click嘗試用一個相容Python 2和Python 3相同方式來處理。也防止了一些終端編碼不一致出現UnicodeError異常。

Click 2.0還加入了ANSI colors支援,如果輸出結果到檔案中還會自動去處ANSI codes。

要使用ANSI colors我們需要colorama包配合操作:

pipenv install colorama

示例:
click.echo(click.style("Hello World!", fg='green')) # click.secho('Hello World!', fg='green')

巢狀命令

import click@click.group()def cli():
    pass@cli.command()def initdb():
    click.echo('Initialized the database')@cli.command()def dropdb():
    click.echo('Dropped the database')if __name__ == '__main__':
    cli()

新增引數

使用option()和argument()裝飾器增加引數

@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)
What it looks like:

$ python hello.py --helpUsage: hello.py [OPTIONS] NAMEOptions:
  --count INTEGER  number of greetings
  --help           Show this message and exit.

繼承setuptools

原文出處:https://www.cnblogs.com/erhuabushuo/p/10070136.html  

來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/3244/viewspace-2819281/,如需轉載,請註明出處,否則將追究法律責任。

相關文章