Refactoring - config loading helper

2 files changed, 11 insertions, 7 deletions

diff --git a/ausbutler/config.py b/ausbutler/config.py
new file mode 100644
index 0000000..7a0f57d
--- /dev/null
+++ b/ausbutler/config.py
@@ -0,0 +1,8 @@
+from os import path
+import json
+import __main__
+
+def load_config(filename):
+    return json.load(
+        open(path.join(path.dirname(__main__.__file__),
+                       'config', filename + '.json')))
diff --git a/ausbutler/db.py b/ausbutler/db.py
index 940c69d..e40fee5 100644
--- a/ausbutler/db.py
+++ b/ausbutler/db.py
@@ -1,15 +1,11 @@
-import json
-from os import path
-
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
 
-import __main__
+from .config import load_config
 
 def get_session():
     session = sessionmaker(bind=create_engine(
         "mysql://{0[user]}:{0[pass]}@{0[host]}/{0[db]}".format(
-            json.load(open(
-                path.join(path.dirname(__main__.__file__), 'config', 'db.json')
-            )))))
+            load_config('db')
+        )))
     return session()