From 8a6b5981795fc65c994d8e42c78f23a6c8d10498 Mon Sep 17 00:00:00 2001
From: emkael <emkael@tlen.pl>
Date: Thu, 5 Jan 2017 13:17:18 +0100
Subject: Refactoring - config loading helper

---
 ausbutler/db.py | 10 +++-------
 1 file changed, 3 insertions(+), 7 deletions(-)

(limited to 'ausbutler/db.py')

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()
-- 
cgit v1.2.3