-
需求:在命令行中通过参数控制日志的输出级别
-
思路分析:
1.如果想要通过命令行传入参数,那么就要在被执行文件中写入方法能够获取到命令行中的参数
2.如果能获取到参数,那么我们就能通过参数进行简单的if 。。。else 判断给出不同的选项
3.既然有命令行了,那么-h (help)就一定不能少
4.还有一个比较重要的问题就是怎样将参数传入其它相关引用模块中
-
明确这四个功能点后,那就分头行动,寻找解决方案
answer python中内置函数模块 sys 就能实现命令行的抓取
- 方法一:
eg:
文件名为:test.py
import sys
print "脚本名:", sys.argv[0]
for i in range(1, len(sys.argv)):
print "参数", i, sys.argv[i]
命令行输出:
python test.py hello world
输出结果:
脚本名:test.py
参数 1 hello
参数 2 world
这种方法可以实现简单的参数传入,但是无法实现{key:value}的形式
例如这样的更加灵活些:
python test.py -l info - 方法二:
eg:
test.py
import getopt,sys
def usage():
"""使用说明"""
print """
参数使用说明: -l [info][debug][error][warning]
(分别控制日志输出级别为:info/debug/error/warning)
!默认日志输出级别为:warning
eg: python Run.py -l info
"""
opts, args = getopt.getopt(sys.argv[1:], 'hl:')
"""说明:参数h为开关,l:为可识别输出参数;如果新增p参数,则可以'hl:p:'这样写 """
level = 'w'
for op, value in opts:
if op == '-l':
level = value
elif op == '-h':
#帮助文档
usage()
sys.exit()
命令行:
python test.py -h
python test.py -l info
a) sys.argv[1:]为要处理的参数列表,sys.argv[0]为脚本名,所以用sys.argv[1:]过滤掉脚本名。
b) "hi:o:": 当一个选项只是表示开关状态时,即后面不带附加参数时,在分析串中写入选项字符。当选项后面是带一个附加参数时,在分析串中写入选项字符同时后面加一个":"号。所以"hi:o:"就表示"h"是一个开关选项;"i:"和"o:"则表示后面应该带一个参数。
c) 调用getopt函数。函数返回两个列表:opts和args。opts为分析出的格式信息。args为不属于格式信息的剩余的命令行参数。opts是一个两元组的列表。每个元素为:(选项串,附加参数)。如果没有附加参数则为空串''。
getopt函数的第三个参数[, long_options]为可选的长选项参数,上面例子中的都为短选项(如-i -o)长选项格式举例:
--version
--file=error.txt
让一个脚本同时支持短选项和长选项
getopt.getopt(sys.argv[1:], "hi:o:", ["version", "file="])
- 最后一个问题
- 通过全局模块方法传递参数
eg:
在test2.py中引用test中传入的参数level
test2.py
import test.py
parameter = test.level
print parameter
@晴天--2016-09-13 10:48:34