X-Git-Url: http://git.veekun.com/zzz-pokedex.git/blobdiff_plain/ec65b5ae366174912cedf0e263d425378220d70b..c1ba23b0c45a0873cd27456c0950c7f4f3253fe4:/pokedex/db/__init__.py diff --git a/pokedex/db/__init__.py b/pokedex/db/__init__.py index a6c8f6e..e2790da 100644 --- a/pokedex/db/__init__.py +++ b/pokedex/db/__init__.py @@ -2,7 +2,7 @@ from sqlalchemy import MetaData, Table, engine_from_config, orm from ..defaults import get_default_db_uri from .tables import metadata -from .multilang import MultilangSession +from .multilang import MultilangSession, MultilangScopedSession def connect(uri=None, session_args={}, engine_args={}, engine_prefix=''): @@ -42,6 +42,6 @@ def connect(uri=None, session_args={}, engine_args={}, engine_prefix=''): all_session_args = dict(autoflush=True, autocommit=False, bind=engine) all_session_args.update(session_args) sm = orm.sessionmaker(class_=MultilangSession, **all_session_args) - session = orm.scoped_session(sm) + session = MultilangScopedSession(sm) return session