python获得命令行输入的参数

  • Z时代
  • 2024-01-10
  • 分类：综合

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:]的缘故是命令行第一个参数为文件名本身，这里没有意义

以上是 python获得命令行输入的参数 的全部内容， 来源链接： utcz.com/z/529879.html

回到顶部