python中getopt函数详解
责任编辑:梦想飞行     时间:2023-08-07     来源:转载于:CSDN
责任编辑:梦想飞行
时间:2023-08-07  来源:转载于:CSDN
分类: 技术分享
浏览量: 423

在运行程序时,可能需要根据不同的条件,输入不同的命令行选项来实现不同的功能。目前有短选项和长选项两种格式。短选项格式为"-"加上单个字母选项;长选项为"--"加上一个单词。长格式是在Linux下引入的。许多Linux程序都支持这两种格式。在Python中提供了getopt模块很好的实现了对这两种用法的支持,而且使用简单。

一、getopt模块有两个函数,两个属性:

函数:

  • getopt.getopt
  • getopt.gnu_getopt

属性:

  • getopt.error
  • getopt.GetoptError
    这两个属性主要是用来抛出错误信息的,非常友好不是吗?

我们经常用到的是:getopt函数:

getopt.getopt(args, shortopts, longopts=[])
args指的是当前脚本接收的参数,它是一个列表,可以通过sys.argv获得
shortopts 是短参数  啥是短参数啊?  类似于 这样:python test.py -h # 输出帮助信息
longopts 是长参数  啥是长参数啊?  类似于 这样:python test.py -help # 输出帮助信息


  1. import getopt


  2. import sys


  3. arg = getopt.getopt(sys.argv[1:],'-h',['help'])


  4. print(arg)

结果如下:


  1. root@Kali:~/python# python3.5 test.py -h


  2. ([('-h', '')], [])


  3. root@Kali:~/python# python3.5 test.py --help


  4. ([('--help', '')], [])


  5. root@Kali:~/python#

可以看到已经接收了参数。并且做了处理,为啥我传入sys.argv[1:]?

因为sys.argv里的argv[0]是当前脚本的文件名,不需要它去参与,要不然你的选项和选项值无法匹配,问题多多。
假设我要接收一个参数+参数值的选项怎么办?


  1. #!/usr/bin/env python3.5


  2. import urllib.request


  3. import getopt


  4. import sys



  5. opts,args = getopt.getopt(sys.argv[1:],'-h-f:-v',['help','filename=','version'])


  6. for opt_name,opt_value in opts:


  7. if opt_name in ('-h','--help'):


  8. print("[*] Help info")


  9. exit()


  10. if opt_name in ('-v','--version'):


  11. print("[*] Version is 0.01 ")


  12. exit()


  13. if opt_name in ('-f','--filename'):


  14. fileName = opt_value


  15. print("[*] Filename is ",fileName)


  16. # do something


  17. exit()

运行测试结果如下:


  1. root@Kali:~/python# python3.5 test.py --filename=test


  2. [*] Filename is test


  3. root@Kali:~/python# python3.5 test.py --filename=


  4. [*] Filename is


  5. root@Kali:~/python# python3.5 test.py --help


  6. [*] Help info


  7. root@Kali:~/python# python3.5 test.py --version


  8. [*] Version is 0.01


  9. root@Kali:~/python# python3.5 test.py -v


  10. [*] Version is 0.01


  11. root@Kali:~/python# python3.5 test.py -f test


  12. [*] Filename is test


  13. root@Kali:~/python#

来详细解释一下这几行代码
首先从短参数名开始。
我定义了'-h-f:-v' 大家发现没有,在-f后面多了一个":"
这个":"代表了当前参数是有值的,是一个参数名+参数值的参数
如果我再加一个-o: 那么证明-o后面是可以接收一个值,这个值就是-o的参数值,将会保存到opts变量中。
长参数名的方式和短参数差不多,唯一的区别就是长参数如果要接收值,那必须得在后面加上一个"="


短参数使用的时候是参数名[空格]参数值
长参数使用的时候是参数名=参数值
PS:不理解可以看我的测试结果

opts与args:

  • opts 为分析出的格式信息。args 为不属于格式信息的剩余的命令行参数。
  • opts 是一个两元组的列表。每个元素为:( 选项串, 附加参数) 。如果没有附加参数则为空串''

eg:


  1. #!/usr/bin/env python


  2. #-*- coding:utf-8 -*-



  3. import sys


  4. import getopt



  5. list=['-t','192.168.26.216','-r',256,'-p',80]


  6. args=['-h','-o','file','--help','--output=out','file1','file2']


  7. opts, arg = getopt.getopt(args, "ho:", ["help", "output="])#"ho:"也可以写成'-h-o:'


  8. print(opts)


  9. print(arg)



  10. for o, a in opts:


  11. if o in ("-h", "--help"):


  12. sys.exit()


  13. if o in ("-o", "--output"):


  14. output = a

输出结果:


  1. E:\testPython\venv\Scripts\python.exe E:/testPython/a111/test1.py


  2. [('-h', ''), ('-o', 'file'), ('--help', ''), ('--output', 'out')]


  3. ['file1', 'file2']



  4. Process finished with exit code 0

 

 

 

 

 


来源:转载于:CSDN

回复:

Copyright © 2021 .长沙麦涛网络科技有限公司 All rights reserved. 湘ICP备20015126号-2
联系我们