From 59a7e4778e0d7669f05b028e900a70f25a3febb9 Mon Sep 17 00:00:00 2001 From: wenqin Date: Wed, 16 Jan 2019 10:22:32 +0800 Subject: [PATCH] Update ticketConf.py MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 增加指定配置文件运行的参数，这样方便抢来回的车票 --- config/ticketConf.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/config/ticketConf.py b/config/ticketConf.py index 9c55da7..1ba6b91 100755 --- a/config/ticketConf.py +++ b/config/ticketConf.py @@ -9,7 +9,9 @@ def _get_yaml(): 解析yaml :return: s 字典 """ - path = os.path.join(os.path.dirname(__file__) + '/ticket_config.yaml') + opts, args = getopt.getopt(sys.argv[1:], 'f:') + config_name = opts[0][1] if opts else 'ticket_config.yaml' + path = os.path.join(os.path.dirname(__file__) + '/' + config_name) f = open(path) s = yaml.load(f) f.close() @@ -17,4 +19,4 @@ def _get_yaml(): if __name__ == '__main__': - print(_get_yaml()) \ No newline at end of file + print(_get_yaml())