Python3之命令行参数处理

sys模块
常用单元
getopt模块
optparse模块
argparse模块

sys模块

sys模块代表了Python解释器，主要用于获取和Python解释器相关的信息，其中 sys.argv 可以获取命令行参数

在Python交互式解释器中可以先导入sys模块 import sys，再输入 dir(sys) 查看sys模块所包含的全部程序单元（包括变量、函数等），或者 [i for i in dir(sys) if not i.startswith('_')] 过滤掉隐藏的单元

常用单元

sys.argv：以列表的方式获取运行 Python 程序的命令行参数存放其中。其中 sys.argv[0] 通常就是指该 Python程序本身，sys.argv[1] 代表第一个参数，sys.argv[2] 代表第二个参数，以此类推。类似于 Shell 中的 $0、$1、$2

import sys
print('程序名称为：{}，第一个参数为：{}，第二个参数为：{}'.format(sys.argv[0], sys.argv[1], sys.argv[2]))

sys.byteorder：显示本地字节序的指示符。如果本地字节序的大端模式，则该属性返回big，否则返回little

sys.copyright：该属性返回与 Python 解释器有关的版权信息

sys.executable：该属性返回 Python 解释器在磁盘上的存储路径

sys.exit()：通过引发SystemExit异常来退出程序，将其放在try块中不能阻止finally块的执行。

sys.flags：该只读属性返回运行Python命令时指定的旗标

sys.getfilesystemencodiing()：返回在当前系统中保存文件所用的字符集

sys.getrefcount()：返回指定对象的引用计数。

sys.getrecursionlimit()：返回Python解释器当前支持的递归深度。该属性可通过setrecursionlimit()方法重新设置

sys.getswitchinterval()：返回在当前Python解释器中线程切换的时间间隔。该属性可通过setswitchinterval()函数改变

sys.implementation：返回当前Python解释器的实现

sys.maxsize：返回Python整数支持的最大值。在32位平台上，该属性值为 2 ×× 31-1；在64位平台上，该属性值为 2 ×× 63-1

sys.path：该属性指定Python查找模块的路径列表。程序可通过修改该属性来动态增加Python加载模块的路径

sys.platfrom：返回Python解释器所在的平台的标识符

sys.stdin：返回系统的标准输入流——一个类文件对象

sys.stdout：返回系统的标准输出流——一个类文件对象

sys.stderr：返回系统的错误输出流——一个类文件对象

sys.version：返回当前Python解释器的版本信息

sys.winver：返回当前Python解释器的主版本号

sys.copyright：返回当前Python解释器的版权信息

getopt模块

getopt是对sys.argv获取的命令行参数进行二次处理。在运行程序时，可能需要根据不同的条件，输入不同的命令行选项来实现不同的功能，比如 -u 代表后面是用户，-p 代表后面是密码。

举个例子

## fileName: test.py
import sys, getopt
opts, args = getopt.getopt(sys.argv[1:], "hu:", ["help", "file="])
print("opts输出结果为", opts)
print("args输出结果为", args)

调用getopt.getopt()会返回两个列表，第一个列表赋值给opts，第二个列表赋值给args：
- opts：是 (选项, 选项值) 的元组为一个元素的列表。如果没有选项值为空字符串
- args：存放没用选项单参数的列表。不属于格式信息的剩余的命令行参数
使用 sys.argv[1:] 是为过滤第一个参数（第一个参数为Python程序本身）
"hu:" 为定义短格式选项（-）。这里的h和u为两个选项。"h" 后面没有 ":" 表示h这个选项没有选项值，是作为开关选项（比如：有 -h 为显示帮助信息，没有 -h 为不显示帮助信息）。"u:" 后面有 ":" 为必须有选项值
["help", "file="] 为定义长格式选项（--）。这里的 "=" 和上面的短格式选项中的 ":" 一样

执行的例子

python test.py -h -u root --help --file=file.py file2.py
## 输出结果
opts输出结果为 [('-h', ''), ('-u', 'root'), ('--help', ''), ('--file', 'file.py')]
args输出结果为 ['file2.py']

optparse模块

optparse模块主要用来为脚本传递命令参数，采用预先定义好的选项来解析命令行参数，自动生成帮助信息，不需要sys.argv

先简单写个例子

import optparse

usage="python %prog -u/--user <target user> -p/--password <target password>"
parser=optparse.OptionParser(usage) ## 写入上面定义的帮助信息
parser.add_option('-u', '--user',dest='User',type='string',help='target user', default='root')
parser.add_option('-p','--password',dest='Pwd',type='string',help='target password')
options, args=parser.parse_args()
print('options为', options)
print("用户名为", options.User)
print("密码为", options.Pwd)
print('args为', args)

add_option参数：

可以设置多个选项字符串，比如上面的 '-u', '--user' 这两个，用的时候任选其一
dest用于定义选项值存放的位置，作为后面的第一个字典（options）的键，值为命令输入的参数
type用于检测命令行输入参数的数据类型是否符合要求，有 string（字符串）、int（整数）、float（浮点数）等类型
help用于定义帮助信息
default设置默认值

帮助信息里面的选项字符串都在一个组中，我们可以创建不同的组用来给选项字符串分类

import optparse

usage="python %prog -u/--user <target user> -p/--password <target password>"
parser=optparse.OptionParser(usage) ## 写入上面定义的帮助信息
parser.add_option('-u', '--user',dest='User',type='string',help='target user', default='root')
parser.add_option('-p','--password',dest='Pwd',type='string',help='target password')
group = optparse.OptionGroup(parser, 'Debug Options') ## 创建group分组
parser.add_option_group(group)  ## 把创建的组添加到parser中
group.add_option('-d', '--debug', action='store_true', help='Print debug information')
options, args=parser.parse_args()

action有三种类型：

action='store' 默认类型，可以忽略不写。用户必须给出一个明确的参数值，该类型定义了将实际参数值保存到dest指定变量中
action='store_true' 用户不需给出参数值，该类型定义了将布尔值true保存到dest指定的变量中(和上面getopt中不加 ':' 、'=')
action='store_false' 用户不需给出参数值，该类型定义了将布尔值false保存到dest指定的变量中

argparse模块

optparse模块的升级版，optparse被Python2.7和Python3.2弃用，所有有理由相信未来是属于argparse模块的

编辑帮助信息

import argparse

parser = argparse.ArgumentParser(description='帮助描述信息', prog='程序显示名称', usage='顶部显示信息，会覆盖prog', epilog='底部显示信息')  ## 括号里面可以什么都不写，会走默认
parser.parse_args()    ## 命令行参数解析后存放的位置

还有一些不常用的参数：

add_help = False 禁止显示帮助界面，只接收参数
argument_default=argparse.SUPPRESS 限制只能从命令行传递参数，程序中指定无效
allow_abbrev=False 例如--fooo，输入--foo，则不可以识别参数，默认可以
conflict_handler='resolve' 若参数重名，则不报错

添加命令行参数

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-u', '--user', dest='User', type=str,default='root', help='target User')
parser.add_argument('-s', '--sex', dest='Sex', type=str, choices=['男', '女'], default='男', help='target Sex')
parser.add_argument('-n', '--number', dest='Num', nargs=2, required=True,type=int, help='target Two Numbers')
print(parser.parse_args())  ## 字典的方式接收参数

参数大多数和上面的optparse相同，不同的有：