python 之 分割参数getopt

python 之 分割参数getopt
点击打开原文链接



os下有个方法walk，非常的好用，用来生成一个generator。每次可以得到一个三元tupple，其中第一个为起始路径，第二个为起始路径下的文件夹，第三个是起始路径下的文件。

1. 导入getopt, sys 模块

2. 分析命令行参数

3. 处理结果

第一步很简单，只需要：

import getopt, sys

第二步处理方法如下（以Python 手册上的例子为例）：

Python代码








try:
opts, args = getopt.getopt(sys.argv[1:],
"ho:", ["help",
"output="])
except getopt.GetoptError:

# print help information and exit:

try:
    opts, args = getopt.getopt(sys.argv[1:], "ho:", ["help", "output="])
except getopt.GetoptError:
    # print help information and exit:

1. 处理所使用的函数叫getopt() ，因为是直接使用import 导入的getopt 模块，所以要加上限定getopt 才可以。

2. 使用sys.argv[1:] 过滤掉第一个参数（它是执行脚本的名字，不应算作参数的一部分）。

3. 使用短格式分析串"ho:" 。当一个选项只是表示开关状态时，即后面不带附加参数时，在分析串中写入选项字符。当选项后面是带一个附加参数时，在分析串中写入选项字符同时后面加一个":" 号 。所以"ho:" 就表示"h" 是一个开关选项；"o:" 则表示后面应该带一个参数。

4. 使用长格式分析串列表：["help", "output="] 。长格式串也可以有开关状态，即后面不跟"=" 号。如果跟一个等号则表示后面还应有一个参数 。这个长格式表示"help" 是一个开关选项；"output=" 则表示后面应该带一个参数。

5. 调用getopt 函数。函数返回两个列表：opts 和args 。opts 为分析出的格式信息。args 为不属于格式信息的剩余的命令行参数。opts 是一个两元组的列表。每个元素为：( 选项串, 附加参数) 。如果没有附加参数则为空串'' 。

6. 整个过程使用异常来包含，这样当分析出错时，就可以打印出使用信息来通知用户如何使用这个程序。

如上面解释的一个命令行例子为：

'-h -o file --help --output=out file1 file2'

在分析完成后，opts 应该是：

[('-h', ''), ('-o', 'file'), ('--help', ''), ('--output', 'out')]

而args 则为：

['file1', 'file2']

第三步主要是对分析出的参数进行判断是否存在，然后再进一步处理。主要的处理模式为：

Python代码








for o, a
in opts:
if o
in ("-h", "--help"):

usage()
sys.exit()
if o
in ("-o", "--output"):

output = a

for o, a in opts:
    if o in ("-h", "--help"):
        usage()
        sys.exit()
    if o in ("-o", "--output"):
        output = a

使用一个循环，每次从opts 中取出一个两元组，赋给两个变量。o 保存选项参数，a 为附加参数。接着对取出的选项参数进行处理。（例子也采用手册的例子）

实践篇：

Python代码








#!/usr/bin/env python

import sys;

import getopt;

def usage():

print("Usage:%s [-a|-o|-c] [--help|--output] args...." %Dsys.argv[0]);

if
"__main__" == __name__:
#lsArgs = [""];

try:
opts,args = getopt.getopt(sys.argv[1:],
"ao:c", ["help",
"output="]);

print("============ opts ==================");

print(opts);

print("============ args ==================");

print(args);

#check all param

for opt,arg
in opts:
if opt
in ("-h", "--help"):

usage();
sys.exit(1);

elif opt
in ("-t", "--test"):

print("for test option");

else:
print("%s ==> %s" %(opt, arg));

except getopt.GetoptError:

print("getopt error!");

usage();
sys.exit(1);

#!/usr/bin/env python

import sys;
import getopt;

def usage():
    print("Usage:%s [-a|-o|-c] [--help|--output] args...." %Dsys.argv[0]);

if "__main__" == __name__:
    #lsArgs = [""];
    
    try:
        opts,args = getopt.getopt(sys.argv[1:], "ao:c", ["help", "output="]);
    
        print("============ opts ==================");       
        print(opts);
    
        print("============ args ==================");
        print(args);
        
        #check all param
        for opt,arg in opts:
            if opt in ("-h", "--help"):
                usage();
                sys.exit(1);
            elif opt in ("-t", "--test"):
                print("for test option");
            else:
                print("%s  ==> %s" %(opt, arg));        
        
    except getopt.GetoptError:
        print("getopt error!");
        usage();
        sys.exit(1);

运行结果：

$ ./test_getopt.py -a -oaaa -caa --output=out file1 t file2 -d

============ opts ==================

[('-a', ''), ('-o', 'aaa'), ('-c', ''), ('-a', ''), ('-a', ''), ('--output', 'out')]

============ args ==================

['file1', 't', 'file2', '-d']

-a ==>

-o ==> aaa

-c ==>

-a ==>

-a ==>

--output ==> out