python获得命令行输入的参数
外部直接执行python文件时,我们有时需要获得命令行的参数
获得命令行参数的两种方式
1、通过sys.argv
sys.argv:获得一个参数列表,第一个值为文件名本身,通过sys.argv[1]获得第文件名后的第一个参数 ,多个参数使用空格隔开
测试代码:
import sys
print(sys.argv)
print(len(sys.argv))
print(len(sys.argv[1]))
测试数据:
python3 test.py 第一个参数 第二个参数
执行结果:
["test.py", "第一个参数", "第二个参数"]
3
5
[]
["第一个参数", "第二个参数"]
2、通过getopt()获得参数值
如果命令行需要通过linux那中指定参数名的方法运行,可以使用getopt()来获得参数内容
命令【getopt.getopt(参数列表,”短选项参数”,[长选项参数])】
参数有两种维度区分:短选项,长选项;带参数值(开关)和不带参数值
短选项参数(命令行使用-参数名):由1个字母组成,如果字母后面跟冒号:表示这个参数需要有参数值,否则表示不输入参数值
例如:io:
表示入参要输入-i -o=xxx
长选项参数(命令行使用–参数名):由[参数名=]表示,有值的参数一定要写=,否则会抛错
例如:["help","input=","output="]
表示入参要输入 --help --input=xxx --output=xxx
测试代码:
#Python学习交流群:778463939
import getopt
import sys
opts,args=getopt.getopt(sys.argv[1:],"i:ho:",["help","input=","output="])
print(opts)
print(args)
for opts,arg in opts:
print(opts)
if opts=="-h" or opts=="--help":
print("我只是一个说明文档")
elif opts=="-i" or opts=="--input":
print("我只是输入,输入内容如下:")
print(arg)
elif opts=="-o" or opts=="--output":
print("我只是输出,输出内容如下:")
print(arg)
测试数据:
python3 test.py -h -o=输出 --input=输入 最后一个参数
测试结果:
[("-h", ""), ("-o", "=输出"), ("--input", "输入")]
["最后一个参数"]
-h
我只是一个说明文档
-o
我只是输出,输出内容如下:
=输出
--input
我只是输入,输入内容如下:
输入
说明:使用getopt()需要注意一下,不能把未定义的参数放到短选项和长选项的前面,否则-h –help这种不能读取到参数值(源码中不是很严谨,遇到这种情况直接就break了),放到后面可以,也能通过args获取到
getopt()中第一个参数写sys.argv[1:]的缘故是命令行第一个参数为文件名本身,这里没有意义