click模块 的使用
##官方文档##
click模块 可以 轻松将一个函数变成一个命令行工具
# 1. click是基于装饰器的，我们可以在方法上使用click.command()装饰器来将该方法变成一个命令行工具
import click @click.command() def hello(): click.echo('Hello World!') if __name__ == '__main__': hello()
1 click.option
option 最基本的用法就是通过指定命令行选项的名称，从命令行读取参数值，再将其传递给函数。我们除了设置命令行选项的名称，我们还会指定默认值，help 说明等，option 常用的设置参数如下：
- default: 设置命令行参数的默认值
- help: 参数说明
- type: 参数类型，可以是 str, int, float 等
- prompt: 当在命令行中没有输入相应的参数时，会根据 prompt 提示用户输入可以指定 True, 或者特定字符串来提示用户输入
- nargs: 指定命令行参数接收的值的个数, -1 表示可以接收多个参数,
# 1 可以指定默认值, help 信息, prompt 提示信息
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()
# 2 还可以 指定 type
import click @click.command() @click.option('--rate', '-rate',type=int, help='rate') # 指定 rate 是 int 类型 def show(rate): print(f'type(rate):{type(rate)}') click.echo('rate: %s' % rate) if __name__ == '__main__': show()# 我指定为int 类型 , 如果传入name 就会报错
##2-1 type 也可以设定可选值
import click @click.command() @click.option('--hash-type', type=click.Choice(['md5', 'sha1'])) def digest(hash_type): click.echo('hash_type:{}'.format(hash_type)) if __name__ == '__main__': digest()
## 4 有的时候可能需要输入密码 ,可以用 option
import click @click.command() @click.option('--password', prompt=True, hide_input=True, confirmation_prompt=True) def input_password(password): click.echo('password: %s' % password)如果 两次 输入相同,则通过. 如果输入不相同, 则提醒你 重新输入.
# 4-1 click.password_option() 有一个专门的装饰器
import click @click.command() @click.password_option() def input_password(password): click.echo('password: %s' % password) 
###5 可以通过--, 或 - 来传递参数
import click @click.command() @click.option('--model', '-model', type=click.Choice(['consume_loan', 'large', 'small']), default='large') def run(model): click.echo("model:{}".format(model)) if __name__ == '__main__': run()python test_click.py -model small
python test_click.py --model small
2. click.argument
@click.option可选参数@click.argument固定参数。它的使用和 option 类似，但支持的功能比 option 少。
可以参考 这个文档
## 1 为命令行添加 固定参数
import click @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) if __name__ == '__main__': hello()
##2 看看多个 argument 的例子：
import click @click.command() @click.argument('x') @click.argument('y') @click.argument('z') def show(x, y, z): click.echo('x: %s, y: %s, z:%s' % (x, y, z)) if __name__ == '__main__': show()
##3 nrgs 的用法 ,可以 吸收多余的参数
# 其中，nargs=-1 表明参数 src 接收不定量的参数值，参数值会以 tuple 的形式传入函数。
# 如果 nargs 大于等于 1，表示接收 nargs 个参数值，上面的例子中，dst 接收一个参数值。
# The value is then passed as a tuple. Note that only one argument can be set to nargs=-1, as it will eat up all arguments.
# 用来吸收多余的参数, 作为一个元祖 传值
import click @click.command() @click.argument('src', nargs=-1) @click.argument('dst', nargs=1) def move(src, dst): click.echo('move %s to %s' % (src, dst)) if __name__ == '__main__': move() 
## 4嵌套命令
# 可以通过group来实现命令行的嵌套，也就是让一个命令行工具具有多个命令
import click @click.group() def cli(): click.echo('Hello world') @cli.command() def initdb(): click.echo('Initialized the database') @cli.command() def dropdb(): click.echo('Dropped the database') if __name__ == '__main__': cli()
参考文档:
分享快乐,留住感动. 2018-06-01 19:59:27 --frank
文章来源: python 第三方模块click 的使用