详解python中的三种命令行模块(sys.argv,argparse,click)
Python作为一门脚本语言,经常作为脚本接受命令行传入参数,Python接受命令行参数大概有三种方式。因为在日常工作场景会经常使用到,这里对这几种方式进行总结。
命令行参数模块
这里命令行参数模块平时工作中用到最多就是这三种模块:sys.argv,argparse,click。sys.argv和argparse都是内置模块,click则是第三方模块。
sys.argv模块(内置模块)
先看一个简单的示例:
#!/usr/bin/python importsys defhello(name,age,sex,*args): print("Hello,Mynameis{name}.".format(name=name)) print("I'm{age}yearsold.".format(age=age)) print("I'ma{sex}".format(sex=sex)) print("Otherword:\n{args}".format(args="\n".join(args))) if__name__=="__main__": file_name=sys.argv[0] name=sys.argv[1] age=sys.argv[2] sex=sys.argv[3] other=sys.argv[4:] hello(name,age,sex,*other)
调用脚本:
pythontest_sysargv.pyzhangsan13mannibiss
脚本输出:
Hello,Mynameiszhangsan.
I'm13yearsold.
I'maman
Otherword:
nibi
ss
sys.argv模块不难理解,命令参数作为列表传入Python脚本中,argv[0]是脚本的名字,argv[1]则是第一个参数,后面以此类推。所以在脚本中只需要提取列表中的参数即可使用。上面演示的是正确调用Python脚本的情况,下面则是调用失败的情况。
错误调用脚本:
pythontest_sysargv.pyzhangsan13
错误输出:
Traceback(mostrecentcalllast): File"test_sysargv.py",line16,insex=sys.argv[3] IndexError:listindexoutofrange
关于错误也很好理解,经典的列表索引超出范围,之所列表索引超出范围,没有传入足够的参数。当然你可以使用try...except捕获错误。但是这种做法太过死板,因为在命令行中必须按照脚本规定的参数顺序输入参数,所以这种模块使用一般是针对一些需要的参数比较少并且固定的脚本。
argparse模块(内置模块)
同样的先看一个简单的示例:
#!/usr/bin/python importargparse parser=argparse.ArgumentParser(description='Testforargparsemodule.')#构建命令参数实例 parser.add_argument("--name","-n",help="nameattribute:非必要属性") parser.add_argument("--age","-a",help="ageattribute:非必要属性") parser.add_argument("--sex","-s",help="sexattribute:非必要属性") parser.add_argument("--type","-t",help="typeattribute:非必要属性",required=True) args=parser.parse_args() defhello(name,age,sex,*args): print("Hello,Mynameis{name}.".format(name=name)) print("I'm{age}yearsold.".format(age=age)) print("I'ma{sex}".format(sex=sex)) print("Otherword:\n{args}".format(args="\n".join(args))) if__name__=="__main__": print("Formatoftransferfile:{type}".format(type=args.type)) ifargs.nameandargs.ageandargs.sex: hello(args.name,args.age,args.sex)
执行脚本:
python3test_argparse.py-tjson-nzhangsan-a13-sman
脚本成功输出:
Formatoftransferfile:json
Hello,Mynameiszhangsan.
I'm13yearsold.
I'maman
Otherword:
关于argparse模块的使用,首先需要生成一个命令行参数的实例,然后通过对这个对象添加属性,添加需要从命令行获取的参数,包括哪些是必要参数(required=True),哪些非必要参数,同时也可以对每个参数进行帮助提示(help="")。
而上面示例中分别添加了四个属性,--name和-n同时可以再命令行中使用,都表示了参数name。ArgumentParser通过parse_ags()方法解析参数,检查命令行,将每个参数转换为适当的类型,所以在脚本中同时也可以使用args.n和args.name获取到参数,相对应的如果没有传入该参数,脚本中则获取到None。
查看命令行参数之后脚本的帮助提示:
python3test_argparse.py-h usage:test_argparse.py[-h][--nameNAME][--ageAGE][--sexSEX]--typeTYPE Testforargparsemodule. optionalarguments: -h,--helpshowthishelpmessageandexit --nameNAME,-nNAMEnameattribute:非必要属性 --ageAGE,-aAGEageattribute:非必要属性 --sexSEX,-sSEXsexattribute:非必要属性 --typeTYPE,-tTYPEtypeattribute:非必要属性
另外在添加命令行参数的属性中,还有更多的设置,多余的可以参考Python官方文档,里面都有详细的标注,这里就不展开来讲,总结的就是关于argparse模块的使用非常简便,同时十分人性化,也很符合日常工作的需要。
click模块
先开一个简单的使用示例:
#!/usr/bin/python importclick @click.command() @click.option("--name",default="zhangsan",help="nameattribute:非必要属性") @click.option("--age",help="ageattribute",type=int) @click.option("--sex",help="sexattribute") @click.option("-t",help="typeattribute:必要属性",required=True) defhello(t,name,age,sex,*args): print("Formatoftransferfile:{type}".format(type=t)) print("Hello,Mynameis{name}.".format(name=name)) print("I'm{age}yearsold.".format(age=age)) print("I'ma{sex}".format(sex=sex)) print("Otherword:\n{args}".format(args="\n".join(args))) if__name__=="__main__": hello()
执行脚本:
python3test_click.py-t1--age13--sexman
脚本输出:
Formatoftransferfile:1
Hello,Mynameiszhangsan.
I'm13yearsold.
I'maman
Otherword:
click模块是Flask团队优秀的开源项目,使用方法和argparse模块很相似,同样为命令行封装了大量的方法,使用者只需要专注代码功能的实现。
click模块和argparse模块不同的地方就是,click模块使用装饰器的方式给函数添加命令行属性,关于装饰器简单来讲就是能够在不修改原有函数的基础上添加功能。虽然使用装饰器但是添加命令行属性的方式和argparse模块很相似,包括options中常用的参数含义也有很多类似的地方。值得注意的就是一开始需要通过command()将函数成为命令行的接口。
关于的click模块的就大致讲到这里,其余有兴趣的可以再去了解一下。
总结
关于这三个模块值得注意的是,尽量贴近自己应用场景去选择,真正的做到自己能够方便使用才是你去使用这些模块的原因。
以上就是详解python中的三种命令行模块(sys.argv,argparse,click)的详细内容,更多关于python命令行模块的资料请关注毛票票其它相关文章!