1 files changed, 12 insertions, 2 deletions
diff --git a/archivist/server/__init__.py b/archivist/server/__init__.py
index f5ae194..913cff3 100644
--- a/archivist/server/__init__.py
+++ b/archivist/server/__init__.py
@@ -4,6 +4,9 @@ from flask.cli import FlaskGroup, ScriptInfo
 from flask import Flask
 from flask_restplus import Api, Resource, fields
 
+from ..model import db
+
+app = Flask('archivist')
 api = Api(version='0.1', title='Archivist API',
         description='API for the Archivist Document System')
 
@@ -12,9 +15,16 @@ NAMESPACES=(
     'prefix',
 )
 
-def create_app(info):
-    app = Flask('archivist')
+@app.before_request
+def before_request():
+    db.connect()
 
+@app.after_request
+def after_request(response):
+    db.close()
+    return response
+
+def create_app(info):
     for ns in NAMESPACES:
         mod = import_module('.' + ns, __name__)
         api.add_namespace(mod.api)