Python 命令行 – click 库

一. 认识 Click 库

连载关于命令行系列文章,我们提到了 Python 标准库模块 argparse 可用于解析命令行参数~ 但由于 argparse 使用复杂, add_argument 方法参数众多。为此,第三方库模块 click 应运而生,极大地改善了 argparse 的易用性。

注: Click 第三方库由 Flask 的作者 Armin Ronacher 开发。 Click 相较于 argparse 较好比 requests 相较于 urllib

使用 click 之前,需要先进行安装:

pip install click

Click 模块构建 Python 命令行,分以下 2 步走:

  • 使用 @click.command() 装饰器,使被装饰的函数成为命令行接口;
  • 使用 @click.option() 等装饰器,添加命令行选项。

下面是一个简单的示例:

import click

@click.command()
@click.option('--count', default=1, help='Number of grettings.')
@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()

函数 hello 接受两个参数 countname ,分别从命令行中读取值。程序中用到了 click 模块的 command option 以及 echo ,它们的作用如下:

  • command 使得函数 hello 作为命令行接口;
  • option 添加命令行选项;
  • click.echo 主要是从兼容性考虑:Python2 的 print 是语句,而 Python3 中则为函数。

运行结果:

Python 命令行 - click 库

由于我们刚刚在添加 name 参数时,指定了 prompt='Your name' ,因此当执行命令行时,如果没有指定 --nameClick 将提示用户在交互模式下输入。 --count 参数在定义时,我们指定了默认值 default=1

此外,和 argparse 一样, click 也会自动为我们生成好帮助信息:

Python 命令行 - click 库

二. 添加参数的详细用法

Click 库提供的装饰器 @click.option 通过指定命令行选项的名称,从命令行读取参数值,再将其传递给函数。其中, option 常用的设置参数如下:

  • default 设置命令行参数的默认值;
  • help 参数说明;
  • type 指定参数类型,如 string int float
  • prompt 当命令行未指定相应参数时,会根据 prompt 提示用户输入;
  • nargs 指定命令行参数接受的值的个数。

示例 1 标准库的 argparse 可以通过 nargs 配置参数的个数,通过 type 设置参数的数据类型。在 click 中也可以实现:

import click

@click.command()
@click.option('--pos', nargs=2, type=float)
def example(pos):
    click.echo('pos={}'.format(pos))

if __name__ == '__main__':
    example()

运行结果:

Python 命令行 - click 库

上述的例子中,程序会自动将参数转换为浮点数,并且以元组的形式赋值给变量 pos

示例 2 对于枚举类型的命令行参数,可以指定 typeclick.Choice 实例。 Choice 的参数是一个列表,改列表列出所有可能的值。

import click

@click.command()
@click.option('--subject', type=click.Choice(['math','chinese','english']))
def score(subject):
    click.echo('{} score: 100'.format(subject))

if __name__ == '__main__':
    score()

运行结果:

Python 命令行 - click 库

示例 3 交互式地输入密码

argparse 模块中,输入密码只能像普通参数一样设置,这将带来一定的安全隐患:使用 history 就可以轻易获取到我们的密码。

click 中,上述问题得到了非常优美的解决:只需要设置 prompt=True ,就能够交互式地输入密码;设置 hide_input=True 即可隐藏我们的命令行输入;设置 confirmation_prompt=True 就可以进行密码的两次验证。

import click
from werkzeug.security import generate_password_hash

@click.command()
@click.option('--password', prompt=True, hide_input=True, confirmation_prompt=True)
def encrypt(password):
    click.echo('Encrypting password to %s' % generate_password_hash(password))

if __name__ == '__main__':
    encrypt()

运行结果:

Python 命令行 - click 库

示例 5 在编辑器中编辑输入的数据

熟悉 Linux 系统的用户,很可能使用过命令 fc,在编辑长命令时非常便捷。输入 fc 命令并回车会打开一个编辑器,改编辑器已经保存了上一条命令的内容。我们只需要在编辑器中修复错误内容,然后保存并退出,刚才编辑的命令将会自动执行。

Python 命令行 - click 库

使用 Click 库,在 Python 中也能实现类似的功能:

from __future__ import print_function
import click
import os

message = click.edit()
print(message)
print(os.system(message))

运行代码,程序将自动进入默认编辑器。进入编辑器之后,就可以在编辑器中编辑输入的数据,输入的内容将被赋值给程序中的变量 message

Python 命令行 - click 库

注:Click 是一个非常优秀的开源项目,更多详细用法,可以参见 Click 官网:https://click.palletsprojects.com/en/8.0.x/

Original: https://blog.csdn.net/qq_15256443/article/details/122420933
Author: ITROOKIEIS
Title: Python 命令行 – click 库

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/745744/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

  • NLP工具——doccano标注系统自动标注功能使用

    NLP工具——doccano标注系统自动标注功能使用 0. 简介 1. doccano安装 2. 创建标注任务 3. 写一个命名实体识别的接口 4. 添加自动标注功能 5. 增加标…

    Python 2023年10月26日
    052
  • pytest篇9-失败重试机制

    01 — 引言 上一篇总结了pytest的skip和skipif的使用,skip是无条件的跳过用例,skip是当条件为真时,会跳过测试用例,可以灵活运用起来。 当部分用例因为一些偶…

    Python 2023年9月11日
    048
  • Python学习日记-第二十二天- 飞机大战项目准备(理解精灵 和精灵组)

    系列文章目录 项目准备 使用pygame 创建图形窗口 理解图像并实现图像绘制 理解游戏循环 和 游戏时钟 理解精灵 和精灵组 前言 今天是更新飞机大战项目准备,最有一个环节 理解…

    Python 2023年9月24日
    047
  • yolov5目标检测神经网络——损失函数计算原理

    前面已经写了4篇关于yolov5的文章,链接如下: 1、基于libtorch的yolov5目标检测网络实现——COCO数据集json标签文件解析 2、基于libtorch的yolo…

    Python 2023年9月30日
    058
  • Django ORM的get与filter的区别对比

    get和filter都是条件查询语句,类似于sql语句的where。查询xx字段为xx。查询条件不同在于get的查询符合条件的返回模型类的对象 符合条件的对象只能为一个,如果符合筛…

    Python 2023年8月6日
    072
  • einsum方法详解(爱因斯坦求和)

    einsum方法详解(爱因斯坦求和) einsum是pytorch、numpy中一个十分优雅的方法,如果利用得当,可完全代替所有其他的矩阵计算方法,不过这需要一定的学习成本。本文旨…

    Python 2023年8月29日
    097
  • 【疑难杂症】两DataFrame的时间索引反向/求差集,查找缺失时间数据

    两DataFrame的时间索引反向/求差集,查找缺失时间数据 1 构造两个时间DataFrame数据 2 目的 3 实现方式 * 3.1 方式一 3.2 方式二 3.3 方式三 手…

    Python 2023年8月18日
    055
  • pandas duplicated() 重复行标记与drop_duplicates()删除

    pandas.DataFrame.duplicated DataFrame.duplicated(subset=None, keep=’first’) 返回…

    Python 2023年8月18日
    069
  • Nginx一键配置方法, 再也不用担心我的Nginx配置了

    Nginx做为开发者的神器, 据2021年8月web服务器排行榜, nginx占据了30%的流量稳居第一, 相信大部分程序员都不会陌生. 相信大部分的开发人员都经历过配置Nginx…

    Python 2023年8月5日
    084
  • Django【执行查询】(一)

    官方Django3.2 文档:https://docs.djangoproject.com/en/3.2/topics/db/queries/本文大部分内容参考官方3.2版本文档撰…

    Python 2023年11月2日
    052
  • Halcon深度学习总结

    一、Halcon17之后就已经推出了深度学习模块,之前一直没太关注过,最近一个项目需要用到目标检测,于是研究了Halcon深度学习三大模块,图像分类,目标检测,语义分割。并总结出来…

    Python 2023年10月8日
    048
  • 04. Pandas查询数据

    今天没有漂亮的PPT,只有干货!! Pandas查询数据的几种方法 .loc既能查询,又能覆盖写入,强烈推荐! Pandas使用df.loc查询数据的方法 以上查询方法,既适用于行…

    Python 2023年8月22日
    054
  • drf — 分页组件Pagination

    REST framework提供了分页的支持,提供了两个分页类 使用方法 1.在配置文件settings.py中配置使用 REST_FRAMEWORK = { ‘DEFAULT_P…

    Python 2023年6月10日
    063
  • Python课程设计之俄罗斯方块

    Python课程设计之俄罗斯方块 演示效果 下载地址 运行效果 基础页面 界面动起来 生成、移动、固定、变形 消除与得分 演示效果 点击查看 下载地址 点击下载Python课程设计…

    Python 2023年8月1日
    0114
  • [排序算法] 快速排序 (C++) (含三种写法)

    快速排序解释 快速排序 Quick Sort 与归并排序一样,也是典型的分治法的应用。 (如果有对 归并排序还不了解的童鞋,可以看看这里哟~ 归并排序)❤❤❤ (本文作者: Amα…

    Python 2023年10月14日
    056
  • opencv图像旋转

    图像的旋转 图像旋转是指图像按照某个位置转动一定的角度的过程,旋转中图像仍保持着原始尺寸。图像旋转后图像水平对称轴、垂直对称轴及中心坐标原点都可能会发生变换,因此需要对图像旋转中的…

    Python 2023年9月17日
    047
亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球