注册 登录
  • 欢迎访问"运维那点事",推荐使用Google浏览器访问,可以扫码关注本站的"微信公众号"。
  • 如果您觉得本站对你有帮助,那么可以扫码捐助以帮助本站更好地发展。

Python模块:getopt

Python模块 彭东稳 7463次浏览 已收录 0个评论

有时候我们需要写一些脚本处理一些任务,这时候往往需要提供一些命令行参数,根据不同参数进行不同的处理,在Python里,命令行的参数和C语言很类似(因为标准Python是用C语言实现的)。在C语言里,main函数的原型为int main(int argc, char **argv),这里主要指Linux平台, argv指的是命令行传入的参数个数(程序的name为第一个参数),而argv则是一个指针数组,每一个元素为指向一个命令行参数的指针。在Python里的命令行参数是存储在sys.argv里,argv是一个列表,第一个元素也为程序名称。

看下面的例子:

在命令行运行./test.py  1 2 3,结果如下:

用这种方式可以进行命令行参数的判断从而进行不同的动作。而Python的标准库里实际上有专门处理命令行参数的getopt模块,getopt模块用于抽出命令行选项和参数,也就是sys.argv。命令行选项使得程序的参数更加灵活。支持短选项模式和长选项模式。getopt里面的提供了2个函数和一个类,我们主要使用getopt函数,先看下函数原型:

( [命令行参数列表], “短选项”, [长选项列表] )

先看一个例子,这样会便于理解。

1. 处理所使用的函数叫getopt() ,因为是直接使用import导入的getopt模块,所以要加上限定getopt才可以。

2. 使用sys.argv[1:]过滤掉第一个参数(它是执行脚本的名字,不应算作参数的一部分)。

3. 使用短格式分析串”ho:”,当一个选项只是表示开关状态时,即后面不带附加参数时,在分析串中写入选项字符。当选项后面是带一个附加参数时,在分析串中写入选项字符同时后面加一个”:” 号 。所以”ho:v” 就表示”h”是一个开关选项;”o:” 则表示后面应该带一个参数,”v”也是一个开关选项。

4. 使用长格式分析串列表:[“help”, “output=”],长格式串也可以有开关状态,即后面不跟”=” 号。如果跟一个等号则表示后面还应有一个参数,这个长格式表示”help”是一个开关选项;”output=” 则表示后面应该带一个参数。

5. 调用getopt函数,函数返回两个列表:opts和args,opts为分析出的格式信息,args为不属于格式信息的剩余的命令行参数。opts是一个两元组的列表,每个元素为:( 选项串, 附加参数) ,如果没有附加参数则为空串” 。

6. 整个过程使用异常来包含,这样当分析出错时,就可以打印出使用信息来通知用户如何使用这个程序。

如上面解释的一个命令行例子为:

在分析完成后,opts应该是:

而args则为:

下面就是根据不同参数处理:

注意,如果你输入的短选项或者长选项不在getopt.getopt定义的参数内,执行时就会报错的哦。

模拟一个登陆脚本:

执行脚本

可以看出,这个脚本非常死,任何一个选项或参数不正常就会异常输出。当然,一个登陆选项还有很多东西要做,这里只是使用一个getopt模块而已。


如果您觉得本站对你有帮助,那么可以支付宝扫码捐助以帮助本站更好地发展,在此谢过。
喜欢 (1)or分享 (0)
关于作者:

您必须 登录 才能发表评论!