快速上手 Python 命令行模块 Click

简介:

快速上手 Python 命令行模块 Click
关于Click?
说下 Click 模块是干啥的,简单说,它就是把我们的 Python 脚本的一些函数,通过
添加带有 Click 关键字的装饰器进行装饰进而将函数调用的形式转化为命令行传参的形式然后执行。听不懂也没关系,我们会一步一步来,基本上按照我的实际应用情况来写的。
本文不会涉及太多复杂的语法和理论,将会用通俗的语言和大家进行分享。

安装
python3 -m pip install click
一个简单的例子
首先我们创建一个demo.py

import click #(1)
@click.command() #(2)
def main():

click.echo("hello click") #(3)

if name == '__main__':

main()

装饰器@click.command()会将我们的函数包装成 click 对象,然后我们可以在函数中调用 click 的一些方法,常用的是click.echo,它的作用类似我们的 print,输出用的。
通过命令行我们可以这样运行这行代码

python3 demo.py
好了,我们的第一个例子完事了,我们发现此时的代码使用不使用 click好像没什么区别。
接下来我们就要对它进行拓展了,如果我们需要传入一个数字,然后打印这个时候,我们的代码就可以这样写了。

传入我们的第一个参数
import click

@click.command()
@click.option("-n", "--num", help="input a num")
def main(num):

click.echo(f"{num =}")

if name == '__main__':

main()

这里我们前面代码的基础上给函数 main 增加了 @click.option 装饰器。

接下来说下这几个参数的含义
-n:表示我们在命令行指定参数名的时候使用它即可,注意是一个短'-'
--num:是第一个参数的完整名称,我们在程序中接收值的时候使用它。注意是二个短'-'。
help:在命令行输入 "python3 demo.py --help" 的时候,它可以提示我们这个程序有哪些命令可以用。和我们使用命令行一个道理的。
然后我们的函数main的参数名就是,我们要接收的参数的完整名称,同时通过click.echo打印出来。,f"{num=}" 是 Python 3.8 以后的语法糖,如果 num=3 那么它等价于 num = 3。
最后记得在 __main__ 里执行我们的 main 方法。
好了,介绍完了代码,我们可以运行了,运行示例:
首先假设我们不知道它有几个参数。

python3 demo.py --help
通过help我们可以得到如下信息

Usage: demo.py [OPTIONS]

Options:
-n, --num TEXT input a num #这是定义该字段help的提示内容
--help Show this message and exit.
Usage: 对应我们当前文件名
Options: 是一行一个参数,一个参数分为-开头的缩略参数,和--开头的完整参数名。
然后我们后面可以看到它的类型是 TEXT。紧接着就是该参数的提示信息,通过 help 我们可以设置。

python3 demo.py -n 3
#输出
num ='3'
或者

python3 demo.py --num 3
#输出
num ='3'
上面两者输入方法是等价的使用哪个都行。

现在思考一个问题,如果我们需要 num 的值为数字类型的 3 怎么弄呢?

声明参数类型
这里提供两种方法(当然不仅两种)
方法一:使用 type 关键字,type 就是 python 里的类型

@click.option("-n", "--num",type=int,help="input a num")
再次执行代码

python3 demo.py -n 4
#输出
num =4
同时查看 help 信息的时候 TEXT 变为了 INTEGER。
方法二:使用 default 关键字,指定默认值为 1

@click.option("-n", "--num",default=1,help="input a num")
将 default 的值设置为数字,我们的命令行就知道了我们的参数类型为 int,
这里处理指定了参数类型,还设定了默认值。设定为默认值的参数,可以不指定其值,这时候会使用默认值。
如果我们使用 help 查看信息会发现和上面的方法一没什么区别的。这时候我们可以通过指定另外一个关键字,让它在 help 信息里显示默认值

@click.option("-n", "--num",default=1,help="input a num",show_default=True)
通过加入 show_default 我们可以让 default 的值在 help 信息中显示了,内容格式如下:

Usage: demo.py [OPTIONS]

Options:
-n, --num INTEGER input a num [default: 1]
--help Show this message and exit.
再加一个参数
在上面代码的基础上对代码进行部分修改,主要是新添加一个参数 id。

import click

@click.command()
@click.option("-i", "--id", required=True, help="input an id")
@click.option("-n", "--num", type=int, help="input a number", show_default=True)
def main(id, num):

click.echo(f"your {id=} {num=}")

if name == '__main__':

main()

给之前的函数再添加 @click.option装饰器即可。
这里我添加了的参数为 id ,因为一般情况下 id 是不能为空的,所以我们就可以通过required = True对它进行限制,表示该参数为必传参数。如果不传就出现错误

python3 demo.py -n 1234

没给id传参,出现错误,提示缺少参数。

Usage: demo.py [OPTIONS]
Try "demo.py --help" for help.

Error: Missing option "-i" / "--id".
正确的使用方法应该是

python3 demo.py -i 1 -n 1234
到目前为止一个简答的命令行工具就生成了。接下来说下我用它做过什么事情。

处理实际问题
现在我们有个需求,根据用户名去 mongo 数据库中查找对应的用户登录信息,最终的生成信息格式如下:

不好意思人太多了,让您久等了,您的信息来了!


用户名:lisa
密码: 1234qwer
登录网站: http://www.xxxx.com

目前密码唯一的不要修改哦!
该条消息不用回复了,谢谢。
一开始我是通过在 python 脚本中加个配置文件,然后通过配置文件的形式进行用户名的修改,但是这种方式不灵活,每次都需要重新运行 Python 代码。或者我们还可以使用 fastapi搭建一个RESTful api的服务,但是我的懒得搭这个服务。最终我选择使用命令行的形式去运行。使用的模块就是今天说的这个 click 模块。
接下来写一段需要代码:

@click.command()
@click.option('-u', '--user_name', type=str, help='search user_name')
def main(user_name):

click.echo(f'search user:{user_name}')
result = m.get_user_info(user_name) #数据库查询
try:
    info = f"不好意思人太多了,让您久等了,您的信息来了!\n{'*' * 50}\n用户名: {result.get('user_name')}\n" \
        f"密码: {result.get('user_pwd')}\n登录网站: {result.get('url')}\n{'*' * 50}️\n目前密码唯一的不要修改哦!\n该条消息不用回复了,谢谢。"
except Exception as e:
    info = "Not Found"
click.echo(info)

if name == '__main__':

main()

通过上面的码我们就可以通过命令行的形式进行查询了。

python3 demo.py -u 1234
非常的方便。
如果这个时候,我需要一个临时添加用户的功能,就需要重新写一个函数了,
然后我们在命令行中如何控制两个函数的运行呢?这就是接下来要说的组。

创建组的形式
所谓的创建组,就是通过一个主入口函数,去关联其他的函数,然后其他的函数名可以作为命令直接使用。
好了首先创建一个主入口函数

@click.group()
def main():
pass
这个时候我们发现 main 上面的装饰器变为了@click.group()。
我们通过它准备创建一个命令行组。接下来我们开始创建组成员,所谓的组成员就是一个函数。

@main.command()
@click.option('-u', '--user_name', type=str, help='add user_name')
def get_user(user_name):

 click.echo(f'search user:{user_name}')

这个组成员的作用和它的函数名是一样的就是查询用户信息。
这里需要注意的是组成员的装饰器由原来的
@click.command变为了@main.command。
main 就是上面 main 方法名。然后同样下面的 option 是声明一些参数。
接下来我们创建第二个组成员,用来添加用户信息。

@main.command()
@click.option('-u', '--user_name', required=True, type=str, help="要添加的用户名")
@click.option('-p', '--password', required=True, type=str, help="要添加的密码")
@click.option('-t', '--id_type', required=True, default="phone", type=str, help="添加的账户类型",show_default=True)
def add_user(user_name, password, id_type):

 #do something.....
 click.echo(f"{user_name=}  {password=} {id_type=}")

首先通过@main.command()将它加入到组。然后就是 option 一系列添加参数的操作。这个具体的参数信息上面都说了这里就不提了。好了我们就创建这两个成员,
如果需要其他的功能,比如说删除用户,可以继续添加一个 delete_user 函数,以此类推。

下面我就说下如何执行上面的两个成员函数。

首先,先看下它的 help 命令,都有什么内容,一般不知道一个命令行应用有什么命令参数的我时候我们可以使用它。

python3 demo.py --help
输出以下内容

Usage: demo.py [OPTIONS] COMMAND [ARGS]...

Options:
--help Show this message and exit.

Commands:
add-user
get-user
其中 Commands 就是我们的成员函数的调用命令,需要注意一下它将函数原来的"_"变为了“-”。
然后我们就可以调用查询方法了

python3 demo.py get-user -u 123
然后我们就可以得到结果

search user:123
同样的调用添加用户信息的方法。

python3 demo.py add-user -u 123 -p "1234qwer"
因为-t不是必传参数所以我们可以忽略,使用默认值"phone"。
好了,这就是今天要说的内容,基本上够日常操作了。

更多内容,感兴趣的朋友可以参考官方文档。

原文地址https://www.cnblogs.com/c-x-a/p/12461646.html

相关文章
|
2月前
|
开发者 Python
如何在Python中管理模块和包的依赖关系?
在实际开发中,通常会结合多种方法来管理模块和包的依赖关系,以确保项目的顺利进行和可维护性。同时,要及时更新和解决依赖冲突等问题,以保证代码的稳定性和可靠性
59 4
|
23天前
|
Python
Python Internet 模块
Python Internet 模块。
121 74
|
2月前
|
算法 数据安全/隐私保护 开发者
马特赛特旋转算法:Python的随机模块背后的力量
马特赛特旋转算法是Python `random`模块的核心,由松本真和西村拓士于1997年提出。它基于线性反馈移位寄存器,具有超长周期和高维均匀性,适用于模拟、密码学等领域。Python中通过设置种子值初始化状态数组,经状态更新和输出提取生成随机数,代码简单高效。
123 63
|
2月前
|
测试技术 Python
手动解决Python模块和包依赖冲突的具体步骤是什么?
需要注意的是,手动解决依赖冲突可能需要一定的时间和经验,并且需要谨慎操作,避免引入新的问题。在实际操作中,还可以结合使用其他方法,如虚拟环境等,来更好地管理和解决依赖冲突😉。
|
2天前
|
Python
[oeasy]python057_如何删除print函数_dunder_builtins_系统内建模块
本文介绍了如何删除Python中的`print`函数,并探讨了系统内建模块`__builtins__`的作用。主要内容包括: 1. **回忆上次内容**:上次提到使用下划线避免命名冲突。 2. **双下划线变量**:解释了双下划线(如`__name__`、`__doc__`、`__builtins__`)是系统定义的标识符,具有特殊含义。
15 3
|
2月前
|
持续交付 Python
如何在Python中自动解决模块和包的依赖冲突?
完全自动解决所有依赖冲突可能并不总是可行,特别是在复杂的项目中。有时候仍然需要人工干预和判断。自动解决的方法主要是提供辅助和便捷,但不能完全替代人工的分析和决策😉。
|
2月前
|
Python
Python的模块和包
总之,模块和包是 Python 编程中非常重要的概念,掌握它们可以帮助我们更好地组织和管理代码,提高开发效率和代码质量
47 5
|
2月前
|
数据可视化 Python
如何在Python中解决模块和包的依赖冲突?
解决模块和包的依赖冲突需要综合运用多种方法,并且需要团队成员的共同努力和协作。通过合理的管理和解决冲突,可以提高项目的稳定性和可扩展性
|
2月前
|
JavaScript 前端开发 Python
python中的OS模块的基本使用
欢迎来到瑞雨溪的博客,一名热爱JavaScript与Vue的大一学生。博客分享前端技术及全栈开发经验,持续更新中,期待您的关注和支持!🎉🎉🎉
41 0
|
2月前
|
JavaScript 前端开发 Python
python中的platform模块的基本使用
欢迎来到瑞雨溪的博客,一名热爱JavaScript与Vue的大一学生。博客分享前端技术,助你成长。关注我,持续更新中!🎉🎉🎉
30 0