我们在编写python程序时,程序中经常会提供多种功能或者模式,在实际使用时根据不同的参数使用不同的功能。那么如何获取命令行传入进来的参数呢?
一般情况下,我们会使用 sys 模块,如👇
import sys
# 打印 sys 模块获取到的命令行参数
print(sys.argv)
或者,我们会使用 getopt 模块,如👇
import getopt
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=="optName":
print("optName is value is:", arg)
再或者,我们使用 argparse 模块,如👇
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('-a', '--arg1', help='argument 1')
parser.add_argument('-b', '--arg2', help='argument 2')
parser.add_argument('-c', '--arg3', help='argument 3')
args = parser.parse_args()
print(args.arg1)
print(args.arg2)
print(args.arg3)
但,以👆这些,都不优雅,如果我们需要在不同的函数或者模块中传递使用命令行进来的参数,那这些零散的参数处理代码,将会带来不小的麻烦。我们需要通过一个专门的类来封装命令行的参数。
pip install DebugInfo
定义一个入参类,继承自 入参基类,来专门负责解析和管理我们的传入参数,示例见👇:
# -*- coding:UTF-8 -*-
# region 引入必要依赖
from DebugInfo.DebugInfo import *
# endregion
class 入参类(入参基类):
def __init__(self):
# 初始化基类
super().__init__(接口说明='这个脚本是用来演示如何使用【入参基类】来管理传入参数的')
if __name__ == '__main__':
# 打印一下 入参基类 的说明文档
print(入参基类.__doc__)
👆上面的代码初步建立了入参类,继承自入参基类,我们通过打印入参基类的 doc 信息,可以查阅到相关的使用说明,如👇
如👆,我们通过 doc 的说明信息得知,这个入参基类其实是在 argparse 的基础上做了二次封装,然后提供了一些易用的接口。🤭
👇下面的代码为入参类添加了我们需要接收的参数,并实例化了入参类,并解析了命令行参数:
# -*- coding:UTF-8 -*-
# region 引入必要依赖
from DebugInfo.DebugInfo import *
# endregion
class 入参类(入参基类):
def __init__(self):
# 初始化基类
super().__init__(接口说明='这个脚本是用来演示如何使用【入参基类】来管理传入参数的')
# 添加需要接收的参数
self._添加参数('a', int, '这是 a 参数, 请输入一个数字', 0)
self._添加参数('b', int, '这是 b 参数, 请输入一个数字', 0)
self._添加参数('运算', ['和', '差'], '请输入运算形式', '和')
if __name__ == '__main__':
# 实例化入参类
入参 = 入参类()
# 解析命令行参数
入参.解析入参()
# 打印获取到的参数
print(入参.get('a'))
print(入参.get('b'))
print(入参.get('运算'))
👆上面的代码运行效果如👇:
当我们在命令行中使用 -h 参数时,还会有相应的参数帮助提示,如👇
当我们输入的参数不正确时,还有相应的错误提示,如👇
这么多功能,相对于我们输入的代码量来说,是不是物超所值?优雅至极?
上面的代码中,我们通过 入参.get(‘a’) 这样的方式获取了参数 a 的值.这不够优雅.
如👇的代码中,我们通过在 入参类 中定义property的方法,将每一个参数成员定义对应的getter接口,这样就可以通过 入参.a 这种方式获取和使用参数值了.同时也获得了 IDE 的成员提示和代码补全支持.
# -*- coding:UTF-8 -*-
# region 引入必要依赖
from DebugInfo.DebugInfo import *
# endregion
class 入参类(入参基类):
def __init__(self):
# 初始化基类
super().__init__(接口说明='这个脚本是用来演示如何使用【入参基类】来管理传入参数的')
# 添加需要接收的参数
self._添加参数('a', int, '这是 a 参数, 请输入一个数字', 0)
self._添加参数('b', int, '这是 b 参数, 请输入一个数字', 0)
self._添加参数('运算', ['和', '差'], '请输入运算形式', '和')
# region 访问器
@property
def a(self) -> int:
if 'a' in self._参数字典:
return self._参数字典['a'].值
else:
return 0
@a.setter
def a(self, 值: int):
if 'a' in self._参数字典:
if type(值) in [int, float]:
self._参数字典['a'].值 = int(值)
@property
def b(self) -> int:
if 'b' in self._参数字典:
return self._参数字典['b'].值
else:
return 0
@b.setter
def b(self, 值: int):
if 'b' in self._参数字典:
if type(值) in [int, float]:
self._参数字典['b'].值 = int(值)
@property
def 运算(self) -> str:
if '运算' in self._参数字典:
return self._参数字典['运算'].值
else:
return ''
@运算.setter
def 运算(self, 值: str):
if '运算' in self._参数字典:
self._参数字典['运算'].值 = str(值)
# endregion
if __name__ == '__main__':
# 实例化入参类
入参 = 入参类()
# 解析命令行参数
入参.解析入参()
# 打印获取到的参数
print(入参.a)
print(入参.b)
print(入参.运算)
如此优雅的命令行参数解析和管理方式,我相信你从来没有见过,独此一份了.
上面的代码中,我们看到我们为每一个参数写一个 property 的接口,好麻烦啊.优雅个毛线.
no!no!no! 你能想到的需求,作者当然要支持上了.
👇下面的代码中, 我们通过 入参.转换为属性范式() 将每一个命令行参数自动生成其对应的访问器接口,并且自动送到您的粘贴板里,您唯一需要做的就是 ctrl-V, 惊不惊喜? 意不意外?
# -*- coding:UTF-8 -*-
# region 引入必要依赖
from DebugInfo.DebugInfo import *
# endregion
class 入参类(入参基类):
def __init__(self):
# 初始化基类
super().__init__(接口说明='这个脚本是用来演示如何使用【入参基类】来管理传入参数的')
# 添加需要接收的参数
self._添加参数('a', int, '这是 a 参数, 请输入一个数字', 0)
self._添加参数('b', int, '这是 b 参数, 请输入一个数字', 0)
self._添加参数('运算', ['和', '差'], '请输入运算形式', '和')
if __name__ == '__main__':
# 实例化入参类
入参 = 入参类()
# 对每一个命令行参数,生成其对应的 访问器接口
入参.转换为属性范式()
👆上面的代码中, 入参.转换为属性范式() 的效果如👇:
共生成了32行代码,每个参数的setter, getter属性都给你准备好了, ctrl+V 是您唯一需要做的事情了.
不让告诉我你不知道在哪里 ctrl+V 👈
如果你不希望命令行参数被setter, 你可以通过函数参数控制生成,如👇
if __name__ == '__main__':
# 实例化入参类
入参 = 入参类()
# 对每一个命令行参数,生成其对应的 访问器接口
入参.转换为属性范式(setter=False)
以上所分享的命令行参数解析+管理的方式,提供自 DebugInfo 模块内的 入参基类, 虽然是基于 argparse 的一个二次封装,但相对于直接使用 argparse, 确实方便和清晰不少.