Python-argparse命令解析模块详解与代码展示

快来打我* 2022-11-16 08:33 186阅读 0赞

**目录**

示例

思路

ArgumentParser

add\_argument

位置参数

选项参数

选项对应功能，即选项存在就运行某些代码

一个选项对应多个参数值

选项必须存在

选项存在时，参数值只能在一个范围内选择

参数组

互斥参数组（几个选项最多有一个）

总结

全部代码

参考

--------------------

在类似sqlmap这种命令行框架，经常敲这样的命令

> python xxx.py -u http://example.com -p 80

这是怎么实现的呢？这就用到了argparse模块，本文主要讲解思路，主要将选项参数。

# 示例 #

import argparse
    
    parser = argparse.ArgumentParser(description='Process some integers.')
    parser.add_argument('integers', metavar='N', type=int, nargs='+',
                        help='an integer for the accumulator')
    parser.add_argument('--sum', dest='accumulate', action='store_const',
                        const=sum, default=max,
                        help='sum the integers (default: find the max)')
    args = parser.parse_args()
    print(args.accumulate(args.integers))

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70][] \-h 选项

![20210406205504201.PNG][] 位置及选项参数

# 思路 #

如果我们需要自己写一个解析器，需要哪些功能呢？或者说我们有什么需求呢？举几个例子：

通过位置确定参数值

通过选项确定参数值

1.  选项对应功能，即选项存在就运行某些代码
2.  没有选项时，设置默认值
3.  一个选项对应一个参数值
4.  一个选项对应多个参数值
5.  选项存在时，参数值只能在一个范围内选择
6.  某选项必须存在
7.  几个选项最多有一个（互斥）

以上功能，这个模块都能实现。基本流程如下：

1.  import argparse                                       导入模块
2.  parser = argparse.ArgumentParser（） 创建一个解析对象
3.  parser.add\_argument()                           添加命令行选项
4.  parser.parse\_args()                                进行解析

## ArgumentParser ##

*class* `argparse.``ArgumentParser`(*prog=None*, *usage=None*, *description=None*, *epilog=None*, *parents=\[\]*, *formatter\_class=argparse.HelpFormatter*, *prefix\_chars='-'*, *fromfile\_prefix\_chars=None*, *argument\_default=None*, *conflict\_handler='error'*, *add\_help=True*, *allow\_abbrev=True*)

*  **    prog - 程序的名称（默认值：sys.argv\[0\]）**
 *  **    usage - 描述程序用途的字符串（默认值：从添加到解析器的参数生成）**
 *  **    description - 在参数帮助文档之前显示的文本（默认值：无）**
 *      epilog - 在参数帮助文档之后显示的文本（默认值：无）
 *      parents - 一个 ArgumentParser 对象的列表，它们的参数也应包含在内
 *      formatter\_class - 用于自定义帮助文档输出格式的类
 *      prefix\_chars - 可选参数的前缀字符集合（默认值： '-'）
 *      fromfile\_prefix\_chars - 当需要从文件中读取其他参数时，用于标识文件名的前缀字符集合（默认值： None）
 *      argument\_default - 参数的全局默认值（默认值： None）
 *      conflict\_handler - 解决冲突选项的策略（通常是不必要的）
 *      add\_help - 为解析器添加一个 -h/--help 选项（默认值： True）
 *      allow\_abbrev - 如果缩写是无歧义的，则允许缩写长选项 （默认值：True）

## **`add_argument`** ##

`ArgumentParser.``add_argument`(*name or flags...*\[, *action*\]\[, *nargs*\]\[, *const*\]\[, *default*\]\[, *type*\]\[, *choices*\]\[, *required*\]\[, *help*\]\[, *metavar*\]\[, *dest*\])

*  **    name or flags - 一个命名或者一个选项字符串的列表，例如 foo 或 -f, --foo。**
 *  **    action - 当参数在命令行中出现时使用的动作基本类型。**

1.  `store`：存储参数的值。这是默认的动作
2.  `store_const`：存储被const命名参数指定的值。
3.  `store_true和store_false` ：这些是 `'store_const'` 分别用作存储 `True` 和 `False` 值的特殊用例。
4.  `append`：存储一个列表，并且将每个参数值追加到列表中。使用与一个选项多次出现的情况。
5.  `append_const`：这存储一个列表，并将const命名参数指定的值追加到列表中。
6.  `count`：计算一个关键字参数出现的数目或次数。

*  **    nargs - 命令行参数应当消耗的数目。**

1.  N:N个参数形成一个列表
2.  ?:从命令行中消耗一个参数，并产生一个单一项。如果当前没有命令行参数，则会产生default值
3.  \*: 所有当前命令行参数被聚集到一个列表中
4.  \+:所有当前命令行参数被聚集到一个列表中。另外，当前没有至少一个命令行参数时会产生一个错误信息

*      const - 被一些 action 和 nargs 选择所需求的常数。
 *  **    default - 当参数未在命令行中出现时使用的值。**
 *  **    type - 命令行参数应当被转换成的类型。**
 *  **    choices - 可用的参数的容器。**
 *  **    required - 此命令行选项是否可省略 （仅选项可用）。**
 *  **    help - 一个此选项作用的简单描述。**
 *      metavar - 在使用方法消息(-h)中使用的参数值示例。
 *  **    dest - 被添加到 parse\_args() 所返回对象上的属性名。**

# 位置参数 #

parser.add_argument('integers', metavar='N', type=int, nargs='+',
                        help='an integer for the accumulator')

直接写参数名

metavar：使用-h或--help时，参数值示例

type：字符串转为int

nargs：所有当前命令行参数被聚集到一个列表中

help：--h的时候显示的字符串

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70 1][]

可以输出args，当然字典形式更加的清晰。**位置参数作为args的属性。**

# 选项参数 #

## 选项对应功能，即选项存在就运行某些代码 ##

parser.add_argument('--sum', dest='accumulate', action='store_const',
                        const=sum, default=max,
                        help='sum the integers (default: find the max)')

**--sum存在，就运行sum函数，没有--sum选项，就运行max函数**

使用-参数名或--参数名

dest：属性，这里是函数，const或default指定的函数

action：保存sum函数

const：选项存在时的值，这里是函数sum

default：默认的动作

help：显示的帮助值

## 一个选项对应多个参数值 ##

parser.add_argument("-per", "--per", dest='permutations', default=[], nargs='+', help="permutations of the integers ("
                                                                                         "default: find the max)")

dest：属性名为permutations

nargs：所有当前命令行参数被聚集到一个**列表**中

default：默认为空列表

help：-h时显示的帮助

![20210406222610117.PNG][] 选项参数对应多个参数值

## 选项必须存在 ##

parser.add_argument("-min", "--mini", dest='min', default=[], nargs='+', required=True, help="minimum of the integers")

dest：属性名为min

default：默认为空列表

nargs：接收多个值

required：True，必须有这个选项

![20210406223303205.PNG][]

## 选项存在时，参数值只能在一个范围内选择 ##

![20210406224133405.PNG][]

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70 2][]

## 参数组 ##

`ArgumentParser.``add_argument_group`(*title=None*, *description=None*)

当有比官方更好的位置、选项两个分组时，可以添加新的分组，来更好的分类展示

v_group = parser.add_argument_group("v_group", "about version and verose")
    v_group.add_argument("--version", dest="show_version", help="show the version")
    v_group.add_argument("-v", dest="verbose", default=1, choices=[1, 2, 3, 4], help="show more information, 1~4")

![20210407102956611.PNG][] 参数组

参数组的使用：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70 3][] 结果

## 互斥参数组（几个选项最多有一个） ##

从上面可以看到，一个参数组的参数都可以存在，如果一个参数组只能存在一个呢？这就用到了互斥参数组

`ArgumentParser.``add_mutually_exclusive_group`(*required=False*)

required为True时，代表至少有一个必须存在

csdn = parser.add_mutually_exclusive_group(required=True)
    csdn.add_argument("--like", action='store_true', dest="like", help="love csdn")
    csdn.add_argument("--hate", action='store_true', dest="hate", help="hate csdn")

![20210407104648379.PNG][] 互斥作用

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70 4][] 互斥组一个选项

## 总结 ##

*  argpase模块，将参数解析成一个字典，dest就是字典的key，你输入的就是Value
 *  多个参数值使用nargs
 *  参数值有范围，使用choices
 *  参数必须存在使用required
 *  多个参数不能同时存在，至少有一个存在可以使用互斥参数组

有其他的功能，我们可以自己对字典进行扩展和限制，不满足条件时报错。

# 全部代码 #

"""
    -*- coding:utf-8 -*-
    @File: learnarg.py
    @Author:frank yu
    @DateTime: 2021.04.06 20:45
    @Contact: frankyu112058@gmail.com
    @Description:
    """
    import argparse
    from itertools import permutations
    
    parser = argparse.ArgumentParser(description='a demo to learn argparse by lady_killer.')
    # ----------------------- 位置参数 -----------------------
    parser.add_argument('integers', metavar='N', type=int, nargs='+',
                        help='an integer for the accumulator')
    
    # --------选项参数，一个选项一个值(存在为sum函数，不存在为max函数)----------------
    parser.add_argument('--sum', dest='accumulate', action='store_const',
                        const=sum, default=max,
                        help='sum the integers (default: find the max)')
    
    # -------------------- 一个选项多个值，放到列表 ---------------------
    parser.add_argument("-per", "--per", dest='permutations', default=[], nargs='+', help="permutations of the integers")
    
    # ------------------ 选项必须存在，required参数 ---------------------
    parser.add_argument("-min", "--mini", dest="min", default=[], nargs='+', required=True, help="minimum of the "
                                                                                                 "integers,must exist")
    
    # -------------------- 参数在范围内，choices参数 --------------------
    parser.add_argument("-num", "--number", dest="num", type=int, default=1, choices=[x for x in range(10)],
                        help="a integer,0<=num<=9")
    
    # -------------------- 参数组 --------------------
    v_group = parser.add_argument_group("v_group", "about version and verose")
    v_group.add_argument("--version", action='store_true', dest="show_version", help="show the version")
    v_group.add_argument("-v", dest="verbose", default=1, type=int, choices=[1, 2, 3, 4], help="show more information, 1~4")
    
    csdn = parser.add_mutually_exclusive_group(required=True)
    csdn.add_argument("--like", action='store_true', dest="like", help="love csdn")
    csdn.add_argument("--hate", action='store_true', dest="hate", help="hate csdn")
    
    # ------------- 解析参数 -----------------------
    args = parser.parse_args()
    
    # -------------- 解析结果查看 --------------------
    # print(dir(args))
    print(args.__dict__)
    
    # ------------- 位置参数显示 --------------------
    print("位置参数：", args.integers)
    
    # ------------ 选项参数 显示 --------------------
    print("选项参数使用：", args.accumulate(args.integers))
    
    # ----------- 选项参数，一个参数多个值 ----------
    print("一个选项多个值:", [x for x in permutations(args.permutations)])
    
    # ------------------ 必填选项 --------------
    print("必填选项，最小值：", min(args.min))
    
    # ------------ 范围限制选项 ---------------
    print("范围限制选项，参数值：", args.num)
    
    # ------------- 参数组 -------------------
    if args.show_version:
        print("v1.0")
    if args.verbose > 1:
        print("you will get more information")
    
    # ----------- 互斥参数组 ------------------
    if args.like:
        print("I like CSDN")
    if args.hate:
        print("I hate CSDN")

# 参考 #

[Python-argparse][]

更多python相关内容：[【python总结】python学习框架梳理][python_python]

本人b站账号：[lady\_killer9][lady_killer9]

有问题请下方评论，转载请注明出处，并附有原文链接，谢谢！如有侵权，请及时联系。如果您感觉有所收获，自愿打赏，可选择支付宝18833895206（小于），您的支持是我不断更新的动力。

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70]: /images/20221022/dbf77ba180604e1194a77d5ddbe5267c.png
[20210406205504201.PNG]: /images/20221022/b83a860b9c084aa8811720fdef66e9c4.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70 1]: /images/20221022/5c168115745a4f03910e042c4ac06975.png
[20210406222610117.PNG]: /images/20221022/9816455df2de4f8b996a3835a083b807.png
[20210406223303205.PNG]: /images/20221022/56a27e90f4924400acf8fa7420d3a226.png
[20210406224133405.PNG]: /images/20221022/fac33a14999548b2924eab2fc576c52d.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70 2]: /images/20221022/32501fc32d3a4793901a33eff3bcf440.png
[20210407102956611.PNG]: /images/20221022/2a37829117544b448e85dd70f97b38d9.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70 3]: /images/20221022/e21542a59c954f4ea8dcf9683aead123.png
[20210407104648379.PNG]: /images/20221022/6744d02617e84bdda0072817f78b3bdd.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2xhZHlfa2lsbGVyOQ_size_16_color_FFFFFF_t_70 4]: /images/20221022/15b6906b18644b1cad180e056c35f09c.png
[Python-argparse]: https://docs.python.org/zh-cn/3.7/library/argparse.html
[python_python]: https://blog.csdn.net/lady_killer9/article/details/108763489
[lady_killer9]: http://www.smartredirect.de/redir/clickGate.php?u=IgKHHLBT&m=1&p=8vZ5ugFkSx&t=vHbSdnLT&st=&s=&url=https%3A%2F%2Fspace.bilibili.com%2F374432302&r=https%3A%2F%2Fblog.csdn.net%2Flady_killer9%2Farticle%2Fdetails%2F109189772