python处理命令行参数的方法介绍

发布时间:2020-11-20编辑:脚本学堂
首先,总结下自己对sys模块的理解。手册上对sys的描述是系统参数和系统函数,这里的系统实际上是python解释器。这个模块提供了用户可以访问的解释器变量和一些可以与解释器

首先,总结下自己对sys模块的理解。手册上对sys的描述是系统参数和系统函数,这里的系统实际上是python解释器。这个模块提供了用户可以访问的解释器变量和一些可以与解释器进行交互的功能函数。这里主要关注的就是sys.argv,其实它就是一个解释器变量,通过sys模块,我们就可以访问,并且根据自己的需要来解析。
sys.argv
描述:记录命令行参数
类别:列表(可以使用列表方法来访问)
我用的系统是RHL 5.3,python 版本为2.4,写了个测试程序PyTest.py来观察python解释器调用时,sys.agrv的元素的值:
 

复制代码 代码如下:

#!/usr/bin/env python

import sys

def main(argv):
    for arg in argv:
        print arg
if __name__ == '__main__':
    main(sys.argv)

那我们开始调用测试脚本
python PyTest.py 输出结果:PyTest.py, 原始列表为['PyTest.py']
python PyTest.py -c 0 a:1:b 输出结果为:PyTest.py -c 0 a:1:b。说明解释器是以空格或者制表符来区分参数。
python /root/python_scripts/PyTest.py 输出结果为: /root/python_scripts/PyTest.py。记录的是绝对路径
chmod +x PyTest.py;./PyTest.py 输出结果为: ./PyTest.py
/root/python_scripts/PyTest.py 输出结果为: /root/python_scripts/PyTest.py

注:说明sys.argv[0]保存的是可执行程序名,后面是命令行参数,以空格为区分点。
下面介绍下C语言风格的命令行解析模块getopt。这个支持短命令(-a)和长命令模式(--add)。getopt的调用方式为:getopt.getopt(args, 'hvo:')或者getopt.getopt(args,'hvo:', ['help', 'version', 'output=']),如果只使用长命令,短命令参数必须是个空字符串。:表示参数后必须有参数,=表示长命令后必须有参数。
示例代码:
  

复制代码 代码如下:

  #!/usr/bin/env python

    import sys
    import getopt

    def Usage():
        print 'PyTest.py usage:'
        print '-h,--help: print help message.'
        print '-v, --version: print script version'
        print '-o, --output: input an output verb'
        print '--foo: Test option '
        print '--fre: another test option'
    def Version():
        print 'PyTest.py 1.0.0.0.1'
    def OutPut(args):
        print 'Hello, %s'%args
    def main(argv):
        try:
            opts, args = getopt.getopt(argv[1:], 'hvo:', ['output=', 'foo=', 'fre='])
        except getopt.GetoptError, err:
            print str(err)
            Usage()
            sys.exit(2)
        for o, a in opts:
            if o in ('-h', '--help'):
                Usage()
                sys.exit(1)
            elif o in ('-v', '--version'):
                Version()
                sys.exit(0)
            elif o in ('-o', '--output'):
                OutPut(a)
                sys.exit(0)
            elif o in ('--foo',):
                Foo=a
            elif o in ('--fre',):
                Fre=a
            else:
                print 'unhandled option'
                sys.exit(3)

    if __name__ == '__main__':
        main(sys.argv)

19行: 调用函数时返回两个列表, 第一个列表保存参数和值的元组集合,第二个列表保存其他值,不在参数指定的所有值。
[root@node5 python_scripts]# ./PyTest.py --f
option --f not a unique prefix
PyTest.py usage:
-h,--help: print help message.
-v, --version: print script version
-o, --output: input an output verb
--foo: Test option
--fre: another test option
[root@node5 python_scripts]# ./PyTest.py --fo
option --foo requires argument
PyTest.py usage:
-h,--help: print help message.
-v, --version: print script version
-o, --output: input an output verb
--foo: Test option
--fre: another test option
这两个调用过程可以看到getopt可以进行唯一匹配。
[root@node5 python_scripts]# ./PyTest.py -h -v
PyTest.py usage:
-h,--help: print help message.
-v, --version: print script version
-o, --output: input an output verb
--foo: Test option
--fre: another test option
[root@node5 python_scripts]# ./PyTest.py -v -h
PyTest.py 1.0.0.0.1
这里可以看到解析顺序从前向后。
 

复制代码 代码如下:
[root@node5 python_scripts]# ./PyTest.py -o a --output ab
Hello, a
[root@node5 python_scripts]# ./PyTest.py --output ab -o a
Hello, ab


这里可以看到实际上-o和--output是分开处理的。
从这里我们可以总结下getopt的特点:
1 getopt是从前到后解析
2 getopt不检查额外参数的合法性,需要自行检查
3 短命令行和长命令行是分开解析的。
稍后完善
模块optparse提供了OptionParser用来解析命令行,optparse在2.3引进,2.7弃用,2.7以后用argparse替代。
OptionParser将自动生成帮助信息。
OptionParser的使用过程
 

复制代码 代码如下:
from optparse import OptionParser


导入OptionParser。
 

复制代码 代码如下:
parser = OptionParser()

生成OptionParser的一个实例
 

复制代码 代码如下:
parser.add_option('c', 'clear', dest='clear_option', default=False, action='store_true', help = 'Clean origin configurations and databases.')

添加命令行选项
 

复制代码 代码如下:
(options, args) = parser.parse_args(sys.argv[1:])

解析命令行
options.clear_option 使用命令行。
python 创建目录
os.mkdir()  创建一个非递归目录
os.makedirs() 创建一个递归目录

您可能感兴趣的文章:
python获取命令行参数的方法介绍