Python 命令行参数解析之 Click

在Python笔记：命令行参数解析中介绍了sys.argv、getopt 和 argparse三种命令行参数解析方法，今天来分享另外一个python命令行工具click，它是用来支持 Web开发框架 Flask的，都由Pallets 项目组织开发。

click简介
选项设置
总结

click简介

click相比于其他命令行工具的一个优势就是支持多个命令的嵌套和组合，主要包含以下特性：（摘自：为什么用 Click?）

没有限制可以简单组合
完全遵循 Unix 命令行约定
支持从环境变量中加载值
支持特定值的提示
充分地可嵌套可组合
兼容python 2 和 3
支持外部文件处理
与一些常用的工具帮手结合 (获得 terminal 大小, ANSI 字符颜色, 直接输入按键, 屏幕清理, 获取 config 路径, 启动 apps 和 editors, 等。)

选项设置

下面使用click来实现文章Python笔记：命令行参数解析中的命令行，可以对比一下它们的差异。

import click

@click.command()
@click.option('--field', '-f', help='字段', multiple=True)
# @click.option('--display-filter', '-Y', prompt='display-filter', nargs=2, help='条件')
@click.option('--display-filter', '-Y', prompt='display-filter', type=(str, int, float, bool), help='条件')
@click.option('--count', '-c', default=2, prompt='count', help='条件')
def cli(field, display_filter,count):
    """Simple program that greets NAME for a total of COUNT times."""

    click.echo(f'{field} {display_filter} {count}')

if __name__ == '__main__':
    cli()

可以看到，click使用装饰器@click.command()将cli()方法包装成 click 对象，然后使用@click.option来添加选项参数，下面来介绍这些参数的含义。

基本选项

默认第一个参数为长选项，help参数用于设置选项的描述信息：

$ python3 test_click.py --help
Usage: test_click.py [OPTIONS]

  Simple program that greets NAME for a total of COUNT times.

Options:
  -f, --field TEXT                字段
  -Y, --display-filter <TEXT INTEGER FLOAT BOOLEAN>...
                                  条件
  -c, --count                     计数
  --help                          Show this message and exit.

也可以使用help_option方法修改help选项显示信息：

@click.help_option('--help', '-h', help='帮助信息')

default参数：设置选项默认值
type：设置参数类型
prompt：设置提示信息，设置prompt=true或者prompt=提示信息后，如果没有设置参数，会出现提示，默认关闭。

多个选项和选项多个值

1、选项可以设置多个值，可以通过两种方式：

通过 nargs 参数来配置。多个值将被放入一个元组（tuple）中，只支持固定数量的参数。
使用type参数设置不同类型的值：type=(str, int, float, bool)

2、multiple=True允许多个选项赋值：

一个参数传递多次，并且记录每次的值

$ python3 test_click.py -f tcp -f udp -Y ip.version==4 -c 1 
('tcp', 'udp') ip.version==4 1

密码提示

Click 支持隐藏输入信息和确认，比如输入密码时隐藏内容，也可以直接用 password_option()装饰器：

import click

@click.command()
@click.option('--password', '-p', prompt=True, hide_input=True, confirmation_prompt=True)
# @click.password_option('--password', '-p', prompt=True, hide_input=True, confirmation_prompt=True)
def cli(password):
    click.echo(f'the password is {password}')    

if __name__ == '__main__':
    cli()

执行：

$ python test_click.py
Password:
Repeat for confirmation:
the password is 123456

$ python test_click.py -p 123456
the password is 123456

总结

Click相比argparse使用起来更加简洁，可以实现快速构建命令行程序，当然在扩展性上就没有argparse库好。本文只介绍了click的部分功能，更详细的用法可参考：

click库官方文档：https://click.palletsprojects.com/en/8.0.x/
click库中文文档：https://click-docs-zh-cn.readthedocs.io/zh/latest/

--THE END--