1 files changed, 36 insertions, 0 deletions

diff --git a/src/mailman_pgp/database/__init__.py b/src/mailman_pgp/database/__init__.py
new file mode 100644
index 0000000..b799942
--- /dev/null
+++ b/src/mailman_pgp/database/__init__.py
@@ -0,0 +1,36 @@
+""""""
+
+from contextlib import contextmanager
+
+from mailman.config import config as mailman_config
+from mailman.utilities.string import expand
+from public import public
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+
+from mailman_pgp.config import config
+from mailman_pgp.model.base import Base
+
+
+@public
+class Database:
+    def __init__(self):
+        url = config.get('db', 'url')
+        self.url = expand(url, None, mailman_config.paths)
+        self.engine = create_engine(self.url)
+        Session = sessionmaker(bind=self.engine)
+        self.session = Session()
+        Base.metadata.create_all(self.engine)
+        self.session.commit()
+
+
+@public
+@contextmanager
+def transaction():
+    try:
+        yield
+    except:
+        config.db.session.abort()
+        raise
+    else:
+        config.db.session.commit()