diff options
Diffstat (limited to 'lib/sqlalchemy/mods/threadlocal.py')
| -rw-r--r-- | lib/sqlalchemy/mods/threadlocal.py | 7 |
1 files changed, 2 insertions, 5 deletions
diff --git a/lib/sqlalchemy/mods/threadlocal.py b/lib/sqlalchemy/mods/threadlocal.py index b67329612..6f122a409 100644 --- a/lib/sqlalchemy/mods/threadlocal.py +++ b/lib/sqlalchemy/mods/threadlocal.py @@ -29,18 +29,15 @@ class Objectstore(SessionContext): def assign_mapper(class_, *args, **kwargs): assignmapper.assign_mapper(objectstore, class_, *args, **kwargs) -def _mapper_extension(): - return SessionContext._get_mapper_extension(objectstore) - objectstore = Objectstore(Session) def install_plugin(): sqlalchemy.objectstore = objectstore - global_extensions.append(_mapper_extension) + global_extensions.append(objectstore.mapper_extension) engine.default_strategy = 'threadlocal' sqlalchemy.assign_mapper = assign_mapper def uninstall_plugin(): engine.default_strategy = 'plain' - global_extensions.remove(_mapper_extension) + global_extensions.remove(objectstore.mapper_extension) install_plugin() |