1 files changed, 4 insertions, 2 deletions
diff --git a/ausbutler/db.py b/ausbutler/db.py
index 94f07de..353c9e0 100644
--- a/ausbutler/db.py
+++ b/ausbutler/db.py
@@ -4,9 +4,11 @@ from sqlalchemy.orm import sessionmaker
 from .config import load_config
 
 
-def get_session():
+def get_session(config=None):
+    if config is None:
+        config = load_config('db')
     session = sessionmaker(bind=create_engine(
         "mysql+mysqlconnector://{0[user]}:{0[pass]}@{0[host]}/{0[db]}?charset=utf8".format(
-            load_config('db')
+            config
         )))
     return session()