深入浅析Python 命令行模块 Click
Click是用Python写的一个第三方模块,用于快速创建命令行。我们知道,Python内置了一个Argparse的标准库用于创建命令行,但使用起来有些繁琐,Click相比于Argparse,就好比requests相比于urllib。
关于Click?
说下Click模块是干啥的,简单说,它就是把我们的Python脚本的一些函数,通过添加带有Click关键字的装饰器进行装饰进而将函数调用的形式转化为命令行传参的形式然后执行。听不懂也没关系,我们会一步一步来,基本上按照我的实际应用情况来写的。
本文不会涉及太多复杂的语法和理论,将会用通俗的语言和大家进行分享。
安装
python3-mpipinstallclick
一个简单的例子
首先我们创建一个demo.py
importclick#(1) @click.command()#(2) defmain(): click.echo("helloclick")#(3) if__name__=='__main__': main()
装饰器@click.command()会将我们的函数包装成click对象,然后我们可以在函数中调用click的一些方法,常用的是click.echo,它的作用类似我们的print,输出用的。
通过命令行我们可以这样运行这行代码
python3demo.py
好了,我们的第一个例子完事了,我们发现此时的代码使用不使用click好像没什么区别。
接下来我们就要对它进行拓展了,如果我们需要传入一个数字,然后打印这个时候,我们的代码就可以这样写了。
传入我们的第一个参数
importclick @click.command() @click.option("-n","--num",help="inputanum") defmain(num): click.echo(f"{num=}") if__name__=='__main__': main()
这里我们前面代码的基础上给函数main增加了@click.option装饰器。
接下来说下这几个参数的含义
-n:表示我们在命令行指定参数名的时候使用它即可,注意是一个短'-'
--num:是第一个参数的完整名称,我们在程序中接收值的时候使用它。注意是二个短'-'。
help:在命令行输入"python3demo.py--help"的时候,它可以提示我们这个程序有哪些命令可以用。和我们使用命令行一个道理的。
然后我们的函数main的参数名就是,我们要接收的参数的完整名称,同时通过click.echo打印出来。,f"{num=}"是Python3.8以后的语法糖,如果num=3那么它等价于num=3。
最后记得在__main__里执行我们的main方法。
好了,介绍完了代码,我们可以运行了,运行示例:
首先假设我们不知道它有几个参数。
python3demo.py--help
通过help我们可以得到如下信息
Usage:demo.py[OPTIONS]
Options:
-n,--numTEXT inputanum #这是定义该字段help的提示内容
--help Showthismessageandexit.
Usage:对应我们当前文件名
Options:是一行一个参数,一个参数分为-开头的缩略参数,和--开头的完整参数名。
然后我们后面可以看到它的类型是TEXT。紧接着就是该参数的提示信息,通过help我们可以设置。
python3demo.py-n3 #输出 num='3' 或者 python3demo.py--num3 #输出 num='3'
上面两者输入方法是等价的使用哪个都行。
现在思考一个问题,如果我们需要num的值为数字类型的3怎么弄呢?
声明参数类型
这里提供两种方法(当然不仅两种)
方法一:使用type关键字,type就是python里的类型
@click.option("-n","--num",type=int,help="inputanum")
再次执行代码
python3demo.py-n4 #输出 num=4
同时查看help信息的时候TEXT变为了INTEGER。
方法二:使用default关键字,指定默认值为1
@click.option("-n","--num",default=1,help="inputanum")
将default的值设置为数字,我们的命令行就知道了我们的参数类型为int,
这里处理指定了参数类型,还设定了默认值。设定为默认值的参数,可以不指定其值,这时候会使用默认值。
如果我们使用help查看信息会发现和上面的方法一没什么区别的。这时候我们可以通过指定另外一个关键字,让它在help信息里显示默认值
@click.option("-n","--num",default=1,help="inputanum",show_default=True)
通过加入show_default我们可以让default的值在help信息中显示了,内容格式如下:
Usage:demo.py[OPTIONS]
Options:
-n,--numINTEGER inputanum [default:1]
--help Showthismessageandexit.
再加一个参数
在上面代码的基础上对代码进行部分修改,主要是新添加一个参数id。
importclick @click.command() @click.option("-i","--id",required=True,help="inputanid") @click.option("-n","--num",type=int,help="inputanumber",show_default=True) defmain(id,num): click.echo(f"your{id=}{num=}") if__name__=='__main__': main()
给之前的函数再添加@click.option装饰器即可。
这里我添加了的参数为id,因为一般情况下id是不能为空的,所以我们就可以通过required=True对它进行限制,表示该参数为必传参数。如果不传就出现错误
python3demo.py-n1234
#没给id传参,出现错误,提示缺少参数。
Usage:demo.py[OPTIONS]
Try"demo.py--help"forhelp.Error:Missingoption"-i"/"--id".
正确的使用方法应该是
python3demo.py-i1-n1234
到目前为止一个简答的命令行工具就生成了。接下来说下我用它做过什么事情。
处理实际问题
现在我们有个需求,根据用户名去mongo数据库中查找对应的用户登录信息,最终的生成信息格式如下:
不好意思人太多了,让您久等了,您的信息来了!
**************************************************
用户名:lisa
密码:1234qwer
登录网站:http://www.xxxx.com
**************************************************️
目前密码唯一的不要修改哦!
该条消息不用回复了,谢谢。
一开始我是通过在python脚本中加个配置文件,然后通过配置文件的形式进行用户名的修改,但是这种方式不灵活,每次都需要重新运行Python代码。或者我们还可以使用fastapi搭建一个RESTfulapi的服务,但是我的懒得搭这个服务。最终我选择使用命令行的形式去运行。使用的模块就是今天说的这个click模块。
接下来写一段需要代码:
@click.command() @click.option('-u','--user_name',type=str,help='searchuser_name') defmain(user_name): click.echo(f'searchuser:{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该条消息不用回复了,谢谢。" exceptExceptionase: info="NotFound" click.echo(info) if__name__=='__main__': main()
通过上面的码我们就可以通过命令行的形式进行查询了。
python3demo.py-u1234
非常的方便。
如果这个时候,我需要一个临时添加用户的功能,就需要重新写一个函数了,
然后我们在命令行中如何控制两个函数的运行呢?这就是接下来要说的组。
创建组的形式
所谓的创建组,就是通过一个主入口函数,去关联其他的函数,然后其他的函数名可以作为命令直接使用。
好了首先创建一个主入口函数
@click.group() defmain(): pass
这个时候我们发现main上面的装饰器变为了@click.group()。
我们通过它准备创建一个命令行组。接下来我们开始创建组成员,所谓的组成员就是一个函数。
@main.command() @click.option('-u','--user_name',type=str,help='adduser_name') defget_user(user_name): click.echo(f'searchuser:{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) defadd_user(user_name,password,id_type): #dosomething..... click.echo(f"{user_name=}{password=}{id_type=}")
首先通过@main.command()将它加入到组。然后就是option一系列添加参数的操作。这个具体的参数信息上面都说了这里就不提了。好了我们就创建这两个成员,
如果需要其他的功能,比如说删除用户,可以继续添加一个delete_user函数,以此类推。
下面我就说下如何执行上面的两个成员函数。
首先,先看下它的help命令,都有什么内容,一般不知道一个命令行应用有什么命令参数的我时候我们可以使用它。
python3demo.py--help
输出以下内容
Usage:demo.py[OPTIONS]COMMAND[ARGS]...
Options:
--help Showthismessageandexit.Commands:
add-user
get-user
其中Commands就是我们的成员函数的调用命令,需要注意一下它将函数原来的"_"变为了“-”。
然后我们就可以调用查询方法了
python3demo.pyget-user-u123
然后我们就可以得到结果
searchuser:123
同样的调用添加用户信息的方法。
python3demo.pyadd-user-u123-p"1234qwer"
因为-t不是必传参数所以我们可以忽略,使用默认值"phone"。
好了,这就是今天要说的内容,基本上够日常操作了。
更多内容,感兴趣的朋友可以参考官方文档。
到此这篇关于深入浅析Python命令行模块Click的文章就介绍到这了,更多相关Python命令行模块Click内容请搜索毛票票以前的文章或继续浏览下面的相关文章希望大家以后多多支持毛票票!
声明:本文内容来源于网络,版权归原作者所有,内容由互联网用户自发贡献自行上传,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任。如果您发现有涉嫌版权的内容,欢迎发送邮件至:czq8825#qq.com(发邮件时,请将#更换为@)进行举报,并提供相关证据,一经查实,本站将立刻删除涉嫌侵权内容。