web: Add base controllers

master
Dustin 2015-12-31 21:39:27 -06:00
parent dbd6d12499
commit dc9c5ef137
1 changed files with 26 additions and 0 deletions

View File

@ -0,0 +1,26 @@
from __future__ import unicode_literals
from . import model
import milla.controllers
import sqlalchemy
# pylint: disable=too-few-public-methods
class SQLAlchemyController(milla.controllers.Controller):
def __before__(self, request):
super(SQLAlchemyController, self).__before__(request)
request.db = model.Session(
bind=sqlalchemy.engine_from_config(request.config, 'sqlalchemy.'),
)
def __after__(self, request):
super(SQLAlchemyController, self).__after__(request)
if hasattr(request, 'db') and request.db:
request.db.close()
del request.db
class BaseController(milla.controllers.HTTPVerbController,
SQLAlchemyController):
pass