Python命令行参数解析Demo

文章转自 Python–命令行参数解析Demo

---

写没有操作界面的程序时，最讨厌的就是参数解析问题，尤其是很多参数那种，下面是一个小Demo，拿出来与各位分享：

# -*- coding:utf8 -*-
import os
import datetime
import sys
from optparse import OptionParser


def get_user_paras():
    try:
        opt = OptionParser()
        opt.add_option('--host_ip',
                       dest='host_ip',
                       type=str,
                       help='the ip of the check host')
        opt.add_option('--run',
                       action="store_true",
                       dest="is_run",
                       default=False,
                       help="run the scripts")
        opt.add_option('--view',
                       action="store_false",
                       dest="is_run",
                       default=False,
                       help="only view but not run the scripts")
        opt.add_option('--show_type',
                       dest="show_type",
                       type=int,
                       default=0,
                       help="0 or 1, 0 only show the simple data, 1 show the full data")
        (options, args) = opt.parse_args()
        is_valid_paras = True
        error_messages = []
        host_ip = options.host_ip
        is_run = options.is_run
        show_type = options.show_type
        if not host_ip:
            error_messages.append("host_ip must be set;")
            is_valid_paras = False
        if show_type not in [0, 1]:
            error_messages.append("show_type only can be 0 or 1;")
            is_valid_paras = False

        if is_valid_paras:
            user_paras = {"host_ip": host_ip, "is_run": is_run, "show_type": show_type}
            return user_paras
        else:
            for error_message in error_messages:
                print(error_message)
               opt.print_help()
            return None
    except Exception as ex:
        print("exception :{0}".format(str(ex)))
        return None


def main():
    user_paras = get_user_paras()
    if user_paras is None:
        sys.exit(0)
    info = "host_ip:{0}, is_run:{1}, show_type:{2}"
    info = info.format(user_paras["host_ip"],
                       user_paras["is_run"],
                       user_paras["show_type"])
    print(info)


if __name__ == '__main__':
    main()

当使用OptionParser时，会自动增加–help和-h参数，也会自动生成参数帮助，如：

对于代码：

opt.add_option('--run',
               action="store_true",
               dest="is_run",
               default=False,
               help="run the scripts")

–run 表示参数名

action表示将参数值如何处理，常用的有store/store_true/store_false，store即字面意思，store_true即将True作为参数值传递给参数，store_false将False作为参数值传递给参数

dest表示命令行参数解析后的参数名，

上面代码中–run作为命令行参数传递进来，由于action为store_true，因此参数值为True，解析后的参数名为is_run，通过(options, args) = opt.parse_args() 赋值后，便可以使用options.is_run来放问参数值。

---

对于参数较多或者参数值较大的情况，个人还是比较喜欢使用参数配置文件来实现，简单而且方便编辑，如创建一个run_config.py文件:

# -*- coding:utf8 -*-


# run config
class RunConfig(object):
    is_run = True
    show_status = 1
    host_ip = "192.167.1.1"
    run_scripts = """
    SELECT *
    FROM TB001
    WHERE ID >1000
    AND C1<300
    """

然后在其他文件中访问：

# -*- coding:utf8 -*-
from run_config import RunConfig


def main():
    print("is_run:{0},  host_ip:{1}".format(RunConfig.is_run,RunConfig.host_ip))
    print("run_scripts:{0}".format(RunConfig.run_scripts))


if __name__ == '__main__':
    main()