accountant-ui/api/model/__init__.py

from contextlib import contextmanager
from flask.ext.sqlalchemy import SQLAlchemy
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import scoped_session, sessionmaker

db = SQLAlchemy()

@contextmanager
def session_scope():
    #session = scoped_session(sessionmaker(autocommit = False, autoflush = False, bind = engine))
    session = db.session
    #Base.query = session.query_property()

    try:
        yield session
        session.commit()
    except:
        session.rollback()
        raise
    finally:
        session.close()

import pkgutil

__all__ = []

for loader, module_name, is_pkg in pkgutil.walk_packages(__path__):
    __all__.append(module_name)

from . import *