From 9b9dd12a0253f2e65c5934068d91b544f7679f12 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Isma=C3=ABl=20Bouya?= Date: Thu, 14 Jul 2016 13:25:41 +0200 Subject: Add config file option to the command line --- helpers/__init__.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'helpers/__init__.py') diff --git a/helpers/__init__.py b/helpers/__init__.py index f3e5967..2131c2a 100644 --- a/helpers/__init__.py +++ b/helpers/__init__.py @@ -2,6 +2,13 @@ import sys import os +class Config: + def __init__(self, **kwargs): + for arg in kwargs: + setattr(self, arg, kwargs[arg]) + +config = Config(yml_file="config.yml") + def path(): if getattr(sys, 'frozen', False): return sys._MEIPASS + "/" @@ -11,6 +18,10 @@ def path(): def parse_args(): for arg in sys.argv: + if arg[0:2] == "-c": + sys.argv.remove(arg) + config.yml_file = arg[2:] + if arg == "-v": sys.argv.remove(arg) @@ -22,6 +33,9 @@ def parse_args(): print("option '-v' can only be used in bundled package") sys.exit() +def yml_file(): + return config.yml_file + def duration_to_min_sec(duration): minutes = int(duration / 60) seconds = int(duration) % 60 -- cgit v1.2.3