python click 函数

Click模块

click模块是Flask的作者开发的一个第三方模块,用于快速创建命令行。它的作用与Python标准库的argparse相同,但是,使用起来更简单。

click是一个第三方库,因此使用起来需要先行安装

安装click模块

使用pip命令即可完成模块的安装

1

pip install click

基本使用

Click对argparse的主要改在在于易用性,使用click模块主要分为两个步骤:

  1. 使用@click.command() 装饰一个函数,使之成为命令行接口
  2. 使用@click.option() 等装饰函数,为其添加命令行选项

下列为click官方提供的例子:

1

2

3

4

5

6

7

8

9

10

11

12

13

import click

 

@click.command()

@click.option('--count', default=1help='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."""  # 会当作help信息进行输出

    for in range(count):

        click.echo('Hello %s!' % name)

 

if __name__ == '__main__':

    hello()

在上面的例子中,函数hello接受两个参数,分别是count和name,他们的取值从命令行中获取,这里我们使用了click模块中的command、option、echo,他们的作用如下:

  • command:使函数hello成为命令行接口
  • option:增加命令行选项
  • echo:输出结果,使用echo进行输出是为了更好的兼容性,因为python 2中的print是个语句,python 3中的print 是一个函数

运行上面的脚本,可以通过命令指定--name,--count的值,由于我们在option中指定了prompt选项,那么如果我们执行脚本没有传递name这个参数时,Click会提示我们在交互模式下输入

PS:与argparse模块一样,click也会为我们自动生成提示信息

1

2

3

4

5

6

7

8

9

lidaxindeMacBook-Pro:hello DahlHin$ python3 click模块.py --help

Usage: click模块.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.

其他参数

option最基本的用法就是通过指定命令行选项的名称,从命令行读取参数值,再将其传递给函数。option常用的参数含义:

  • default: 设置命令行参数的默认值
  • help:参数说明
  • type:参数类型,可以是string、int、float等
  • prompt:当在命令行中没有输入相应的参数时,会更具prompt提示用户输入
  • nargs:指定命令行参数接受的值的个数
  • required:是否为必填参数

1

2

3

4

5

6

7

8

9

10

import click

 

@click.command()

@click.option('--pos',nargs=2,type=float)

 

def getfloat(pos):

    click.echo('%s / %s' % pos )

 

if __name__ == '__main__':

    getfloat()

注意:option中定义的参数名称,那么就需要用同名的变量进行接受。

更多参数请参考: http://click.pocoo.org/5/options/#choice-opts 

扩展用法

场景一:我们限定用户输入的值,那么就需要使用Click模块中的Choice函数,Choice的参数是一个列表,该列表中列出所有可能的值。

1

2

3

4

5

6

7

8

9

import click

 

@click.command()

@click.option('-c',required=True,type=click.Choice(['start','stop']))   # 限定-c的值为start,或者stop,required表示是否为必填参数

def getcommand(c):

    click.echo('command is %s' % c )

 

if __name__ == '__main__':

    getcommand()

场景二:应用程序从命令行读取密码。 

使用标准库中的argparse模块只能像输入普通参数一样输入密码。这种方式存在一定安全隐患,例如输入的密码会保存在history中,查看命令历史列表就能获取密码

在Click中,这个问题就能完美的解决,只需要是这prompt为True,那么我们就能交互式输入密码,设置hide_input为True,就能隐藏密码,设置confirmation_prompt为True,就可以进行密码的两次验证,使用起来非常便捷。

1

2

3

4

5

6

7

8

9

10

import click

 

@click.command()

@click.option('-p',prompt='Your Password',hide_input=True,confirmation_prompt=True)

 

def getpassword(p):

    click.echo('Your Password is : %s' % p)

 

if __name__ == '__main__':

    getpassword()

  

点击链接了解更多。

参考:https://www.cnblogs.com/dachenzi/p/7995539.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值