python3获得命令行输入的参数

Love The Way You Lie 2023-01-23 11:45 182阅读 0赞

外部直接执行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学习交流QQ群：531509025 寻找有志同道合的小伙伴，互帮互助,群里还有不错的视频学习教程和PDF电子书！ '''
    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:]`的缘故是命令行第一个参数为文件名本身，这里没有意义