environment class

get_session

couchpotato/__init__.py

@@ -1,10 +1,14 @@
 from couchpotato.core.auth import requires_auth
 from couchpotato.core.logger import CPLog
+from couchpotato.environment import Env
 from flask.app import Flask
 from flask.globals import request
 from flask.helpers import url_for
 from flask.module import Module
 from flask.templating import render_template
+from sqlalchemy.engine import create_engine
+from sqlalchemy.orm import scoped_session
+from sqlalchemy.orm.session import sessionmaker
 from werkzeug.utils import redirect
 import os
 
@@ -12,6 +16,12 @@ app = Flask(__name__)
 log = CPLog(__name__)
 web = Module(__name__, 'web')
 
+def get_session(engine):
+    engine = engine if engine else get_engine()
+    return scoped_session(sessionmaker(autoflush = True, transactional = True, bind = engine))
+
+def get_engine():
+    return create_engine('sqlite:///' + Env.get('db_path'), echo = Env.get('debug'))
 
 @web.route('/')
 @requires_auth
@@ -23,16 +33,3 @@ def page_not_found(error):
     index_url = url_for('web.index')
     url = request.path[len(index_url):]
     return redirect(index_url + '#' + url)
-
-@web.route('/exit')
-@requires_auth
-def exit():
-    # stopping code
-    pass
-
-
-@web.route('/restart')
-@requires_auth
-def restart():
-    # restart code
-    pass