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來放問參數值。

 

更多幫助：https://docs.python.org/2/library/optparse.html

##========================================================##

對於參數較多或者參數值較大的情況，個人還是比較喜歡使用參數配置文件來實現，簡單而且方便編輯，如創建一個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()

簡單粗暴，沒那么麻煩，土豹子的做法哈！

 

##===================================================##