#13: Exit on missing config values

ral-facilities · Jul 29, 2019 · eec6821 · eec6821
1 parent ff81709
commit eec6821
Showing 1 changed file with 13 additions and 3 deletions.
diff --git a/common/config.py b/common/config.py
@@ -1,4 +1,5 @@
 import json
+import sys
 
 
 class Config(object):
@@ -9,13 +10,22 @@ def __init__(self):
         target.close()
 
     def get_db_url(self):
-        return self.config["DB_URL"]
+        try:
+            return self.config["DB_URL"]
+        except:
+            sys.exit("Missing config value, DB_URL")
 
     def get_log_level(self):
-        return self.config["log_level"]
+        try:
+            return self.config["log_level"]
+        except:
+            sys.exit("Missing config value, log_level")
 
     def is_debug_mode(self):
-        return self.config["debug_mode"]
+        try:
+            return self.config["debug_mode"]
+        except:
+            sys.exit("Missing config value, debug_mode")
 
 
 config = Config()