Add config file option to the command line

1 files changed, 14 insertions, 0 deletions

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