Python命令行参数解析

Python命令行参数解析

平时用Python写一些小工具,解析命令行参数是非常有必要的。

获取命令行参数

在Python中命令行参数通过 sys.argv 传递,它是一个list,其中的元素为字符串。

import sys

def cli_parser():
    print(f"参数个数: {len(sys.argv)}")
    print(f"参数列表:{str(sys.argv)}")
    print(f"脚本名:{sys.argv[0]}")
    for i in range(1, len(sys.argv)):
        print(f"参数{i}:{sys.argv[i]}")
        
if __name__ == "__main__":
    cli_parser()
(base)PS D:\桌面\test\python\test1> python console.py --username tom -p haha123
参数个数: 5
参数列表:['console.py', '--username', 'tom', '-p', 'haha123']
脚本名:console.py
参数1:--username
参数2:tom
参数3:-p
参数4:haha123

解析命令行参数

Python标准库提供了2个模块可用于解析命令行参数,getoptargparse

  • getopt是C函数getopt()的移植,适合C程序员
  • argparse更加智能,推荐使用

getopt模块

getopt --- C 风格的命令行选项解析器 — Python 3.12.1 文档

getopt.getopt 方法用于解析命令行参数列表,语法格式如下:

getopt.getopt(args, options[, long_options])

方法参数说明:

  • args: 要解析的命令行参数列表。
  • options: 以字符串的格式定义,options后的冒号(:)表示该选项必须有附加的参数,不带冒号表示该选项不附加参数。
  • long_options: 以列表的格式定义,long_options 后的等号(=)表示如果设置该选项,必须有附加的参数,否则就不附加参数。
  • 该方法返回值由两个元素组成: 第一个是 (option, value) 元组的列表。 第二个是参数列表,包含那些没有---的参数。

抛出异常Exception getopt.GetoptError

  • 在没有找到参数列表,或选项的需要的参数为空时会触发该异常。

  • 异常的参数是一个字符串,表示错误的原因。属性 msg 和 opt 为相关选项的错误信息。

import getopt

def cli_parser():
    """
    命令行参数解析
    """
    username = ""
    password = ""

    try:
        """
            options, args = getopt.getopt(args, shortopts, longopts=[])

            参数args:一般是sys.argv[1:]。过滤掉sys.argv[0],它是执行脚本的名字,不算做命令行参数。
            参数shortopts:短格式分析串。例如:"hp:i:",h后面没有冒号,表示后面不带参数;p和i后面带有冒号,表示后面带参数。
            参数longopts:长格式分析串列表。例如:["help", "ip=", "port="],help后面没有等号,表示后面不带参数;ip和port后面带冒号,表示后面带参数。

            返回值options是以元组为元素的列表,每个元组的形式为:(选项串, 附加参数),如:('-i', '192.168.0.1')
            返回值args是个列表,其中的元素是那些不含'-'或'--'的参数。
        """
        opts, args = getopt.getopt(sys.argv[1:], "hu:p:", ["help", "username=", "password="])
    except getopt.GetoptError:
        print('Error: console.py -u <username> -p <password>')
        print('   or: console.py --username=<username> --password=<password>')
        sys.exit(2)

    # 处理 返回值options是以元组为元素的列表。
    for opt, arg in opts:
        if opt in ("-h", "--help"):
            print('console.py -u <username> -p <password>')
            print('or: console.py --username=<username> --password=<password>')
            sys.exit()
        elif opt in ("-u", "--username"):
            username = arg
        elif opt in ("-p", "--password"):
            password = arg
    print('username为:', username)
    print('password为:', password)

    # 打印 返回值args列表,即其中的元素是那些不含'-'或'--'的参数。
    for i in range(0, len(args)):
        print('参数 %s 为:%s' % (i + 1, args[i]))
        
if __name__ == "__main__":
    cli_parser()

argparse模块

python之argparse模块常见用法包含实例(超详细)-CSDN博客

argparse --- 用于命令行选项、参数和子命令的解析器 — Python 3.12.4 文档

argparse模块提供了ArgumentParser类来构造命令行参数解析器对象,并提供add_argument()成员方法来向这个解析器添加可识别的参数信息格式。解析后的结果通过Namespace类型(一种类似于字典的类型)来返回,使用.参数名的反射方式来提取特定的参数。

arparse模块中用到了Namespace容器类型,用于存储对象和属性。类似于字典,但是可以使用.来访问和设置属性。在实际应用中,常常使用命名空间对象来管理和传递配置参数、命令行参数等。

from argparse import Namespace

# 创建一个命名空间对象
person = Namespace()

# 设置属性值
person.name = "Alice"
person.age = 25

# 访问属性值
print(person.name)    # Alice
print(person.age)     # 25

基本使用示例代码如下:

import argparse
# 构造ArgumentParser对象,该对象完成本模块的全部功能。默认会提供一个-h,--help选项,显示的信息就是description
parser = argparse.ArgumentParser(description='演示命令行参数解析')
# 往parser添加命令行选项
parser.add_argument('-i','--ip', type=str, default='localhost', help='IP地址')
parser.add_argument('-p','--port', type=int, default='8090', help='端口号')
parser.add_argument('user', required=True, help='用户名')
parser.add_argument('passwd', help='密码')
# 解析命令行参数得到Namespace对象,此步会阻塞等待命令行输入
args = parser.parse_args()

# 从Namespace对象中提取各命令行选项对应的参数值
print(args)
print(args.ip)
print(args.port)
print(args.user)
print(args.passwd)
(base)PS G:\Code\Python> python arg.py xiaoming -i 127.0.0.1 --port 9090 mima123
Namespace(ip='127.0.0.1', port=9090, user='xiaoming', passwd='mima123')
127.0.0.1
9090
xiaoming
mima123

上述代码中:

  • ipport是命名参数,可以乱序输入,因为输入时指定了-i--port这样的长短选项

  • userpasswd是位置参数,只能顺序输入,因为输入时没有指定任何信息

  • 同时user是必需参数,而其他参数默认是非必需参数,没有值时会赋值为None

posted @ 2023-12-30 23:18  3的4次方  阅读(18)  评论(0编辑  收藏  举报