web: Add base controllers
parent
dbd6d12499
commit
dc9c5ef137
|
@ -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
|
Loading…
Reference in New Issue