Python - 命令行参数

Python - 命令行参数

命令行参数
1. sys.argv[1:]为要处理的参数列表, sys.argv[0]为脚本名, sys.argv[1:]过滤脚本名.
2. “hc:o:”当选项(h)表示开关状态时，后面不带附加参数. 当选项(c:或o:)带附加参数时，选项字符后面加一个”:”号.
3. getopt函数返回两个列表: opts和args. opts是两元组的列表, 选项列表和参数列表.
4. getopt()的第三个参数[, long_options]为可选的长选项参数, 配合短选项使用, 长选项格式如: –image_dir –image_dir=./hello/

代码

import sys
import getopt

def main(argv):
    image_dir = ''  # 图片文件夹
    code_dir = ''  # 代码文件夹
    out_put = ''  # 输出文件
    result = list()

    try:
        opts, args = getopt.getopt(argv, hi:c:o:, [image_dir=, code_dir=, out_put=])
    except getopt.GetoptError:
        print 'SearchUnusedImages.py -i  -c  -o '
        sys.exit(2)

    for opt, arg in opts:
        if opt == '-h':
            print 'SearchUnusedImages.py -i  -c  -o '
            sys.exit()
        elif opt in (-i, --image_dir):
            image_dir = arg
        elif opt in (-c, --code_dir):
            code_dir = arg
        elif opt in (-o, --out_put):
            out_put = arg

    out_file = open(out_put, 'w')
    print 'scan start.'
    for name, path in search_unused_images(image_dir, code_dir).items():
        result.append(path)
    for sort_path in sorted(result):
        print >> out_file, sort_path
    print 'scan over.'
    out_file.close()


if __name__ == __main__:
    main(sys.argv[1:])