Use the new database class layout

1 files changed, 14 insertions, 0 deletions

diff --git a/portato/db/__init__.py b/portato/db/__init__.py
index 5787c56..122940a 100644
--- a/portato/db/__init__.py
+++ b/portato/db/__init__.py
@@ -9,3 +9,17 @@
 # There is NO WARRANTY, to the extent permitted by law.
 #
 # Written by René 'Necoro' Neumann <necoro@necoro.net>
+
+from __future__ import absolute_import
+
+from ..constants import USE_SQL
+from ..helper import debug
+
+if USE_SQL:
+    debug("Using SQLDatabase")
+    from .sql import SQLDatabase
+    Database = SQLDatabase
+else:
+    debug("Using DictDatabase")
+    from .dict import DictDatabase
+    Database = DictDatabase