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 <image_dir> -c <code_dir> -o <out_put>'
        sys.exit(2)

    for opt, arg in opts:
        if opt == '-h':
            print 'SearchUnusedImages.py -i <image_dir> -c <code_dir> -o <out_put>'
            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:])