Add config file option to the command line

2 files changed, 16 insertions, 1 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
diff --git a/helpers/mapping.py b/helpers/mapping.py
index e800778..dd51246 100644
--- a/helpers/mapping.py
+++ b/helpers/mapping.py
@@ -10,6 +10,7 @@ import sys
 
 from .lock import *
 from .music_file import *
+from . import yml_file
 
 class Mapping(RelativeLayout):
     expected_keys = NumericProperty(0)
@@ -83,7 +84,7 @@ class Mapping(RelativeLayout):
             self.running.remove((key, start_time))
 
     def parse_config(self):
-        stream = open("config.yml", "r")
+        stream = open(yml_file(), "r")
         config = yaml.load(stream)
         stream.close()