命令行神器 Click 快速上手

栏目: Python · 发布时间: 5年前

内容简介:写 Python 的经常要写一些命令行工具,虽然标准库提供有命令行解析工具百闻不如一试创建 click_demo.py ,写一个最简单的函数

Python 的经常要写一些命令行工具,虽然标准库提供有命令行解析工具 Argparse ,但是写起来非常麻烦,我很少会使用它。命令行 工具 中用起来最爽的就是 Click,它是 Flask 的团队 pallets 的开源项目。Click 只要很少的代码就可以优雅地创造一个命令行工具,它致力于将创建命令行工具的过程变的快速而有趣。

百闻不如一试

安装

pip install Click

使用

创建 click_demo.py ,写一个最简单的函数

import click

@click.command()
def hello():
    click.echo('Hello World!')

if __name__ == '__main__':
    hello()

运行:

python click_demo.py
Hello World!

装饰器 click.command() 使函数秒变命令行工具, echo 函数的作用等同于 print 函数。

参数

装饰器 click.option() 可以给命令行函数指定参数

import click

@click.command()
@click.option("--count", default=1, help="打印次数", type=int)
def hello(count):
    """
    这是一个简单示例
    """
    for i in range(count):
        click.echo('Hello World!')

if __name__ == '__main__':
    hello()
count

执行脚本的时候后面加上参数 --help 就可以查看说明文档。

$ python click_demo.py --help

Usage: click_demo.py [OPTIONS]

  这是一个简单示例

Options:
  --count INTEGER  打印次数
  --help        Show this message and exit.

指定参数:

>python click_demo.py --count 3

Hello World!
Hello World!
Hello World!

prompt

有些命令行工具在运行的时候要求用户输入信息,可以给 option 装饰器指定 prompt 参数

import click

@click.command()
@click.option("--count", default=1, help="打印次数", type=int)
@click.option("--name", prompt="请输入名字", help="姓名")
def hello(count, name):
    """
    这是一个简单示例
    """
    for i in range(count):
        click.echo(f'Hello {name}!')

if __name__ == '__main__':
    hello()
$ python click_demo.py

请输入名字: lzjun
Hello lzjun!

Group

Click 很重要的一个特性就是它的分组功能,当一个命令行工具的逻辑已经非常复杂的时候,为了解耦,我们需要将不同的逻辑放在不同的命令中,这样既可以避免单个命令行工具函数臃肿。来看个例子:

# db.py
import click

@click.group()
def db():
    pass

@click.command()
@click.option("--name", help="用户名")
def add(name):
    """
    添加用户
    :param name:
    :return:
    """
    click.echo(f'add user {name}')

@click.command()
@click.option("--id", help="用户名")
def delete(id):
    """
    删除用户
    :param id:
    :return:
    """
    click.echo(f'delete user {id}')

db.add_command(delete)
db.add_command(add)

if __name__ == '__main__':
    db()

这是一个操作数据库DB的命令行工具,提供了添加用户和删除用户的命令行等其它操作,如果所有的业务逻辑全部写在一个函数,维护变得异常困难。

@click.group 装饰器把函数装饰成为一个Group对象,通过 Group 可以添加很多子命令。

python db.py  --help
Usage: db.py [OPTIONS] COMMAND [ARGS]...

Options:
  --help  Show this message and exit.

Commands:
  add     添加用户 :param name: :return:
  delete  删除用户 :param id: :return:

从帮助文档中我们看到 adddelete 就是两个子命令。 Flask 框架的 flask 命令也是一个 Group 命令。

$ flask
Usage: flask [OPTIONS] COMMAND [ARGS]...

  A general utility script for Flask applications.

  Provides commands from Flask, extensions, and the application. Loads the
  application defined in the FLASK_APP environment variable, or from a
  wsgi.py file. Setting the FLASK_ENV environment variable to 'development'
  will enable debug mode.

    $ export FLASK_APP=hello.py
    $ export FLASK_ENV=development
    $ flask run

Options:
  --version  Show the flask version
  --help     Show this message and exit.

Commands:
  add_user           添加账号 :param name: :param username: :param password:...
  init_invite_code   添加邀请码
  recommend_account  更新推荐账号
  routes             Show the routes for the app.
  run                Runs a development server.
  shell              Runs a shell in the app context.

提供了几个预定义的子命令和自定义的命令。

--- EOF ---

任何问题可以扫描二维码给我留言

关注公众号「Python之禅」,回复「1024」免费获取Python资源

命令行神器 Click 快速上手

以上所述就是小编给大家介绍的《命令行神器 Click 快速上手》,希望对大家有所帮助,如果大家有任何疑问请给我留言,小编会及时回复大家的。在此也非常感谢大家对 码农网 的支持!

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

ACM国际大学生程序设计竞赛题解

ACM国际大学生程序设计竞赛题解

赵端阳//袁鹤 / 电子工业 / 2010-7 / 39.00元

随着各大专院校参加ACM/ICPC热情的高涨,迫切需要有关介绍ACM国际大学生程序设计竞赛题解的书籍。《ACM国际大学生程序设计竞赛题解(2)》根据浙江大学在线题库的部分题目,经过分类、筛选、汇编,并进行了解答(个别特别简单或者特别复杂的题目未选择),比较详细地分析和深入浅出地讲解了解题的方法和用到的算法。题目的类型包括基础编程、模拟、字符串处理、搜索、动态规划、回溯、图论、几何和数学题。 ......一起来看看 《ACM国际大学生程序设计竞赛题解》 这本书的介绍吧!

随机密码生成器
随机密码生成器

多种字符组合密码

XML、JSON 在线转换
XML、JSON 在线转换

在线XML、JSON转换工具

HEX CMYK 转换工具
HEX CMYK 转换工具

HEX CMYK 互转工具