initial code repo

[stor4nfv.git] / src / ceph / src / brag / server / ceph_brag / model / __init__.py

diff --git a/src/ceph/src/brag/server/ceph_brag/model/__init__.py b/src/ceph/src/brag/server/ceph_brag/model/__init__.py
new file mode 100644 (file)
index 0000000..ad1a942
--- /dev/null
+++ b/src/ceph/src/brag/server/ceph_brag/model/__init__.py
@@ -0,0 +1,28 @@
+from sqlalchemy import create_engine
+from pecan import conf  # noqa
+from .db import Session, Base
+import sys
+
+def create_from_conf():
+    configs = dict(conf.sqlalchemy)
+    url = configs.pop('url')
+    return create_engine(url, **configs)
+
+def init_model():
+    engine = create_from_conf()
+    conf.sqlalchemy.engine = engine
+    engine.connect()
+    #create the tables if not existing
+    Base.metadata.create_all(engine)
+
+def start():
+    Session.bind = conf.sqlalchemy.engine
+
+def commit():
+    Session.commit()
+
+def rollback():
+    Session.rollback()
+
+def clear():
+    Session.remove()