Got rid of (apparently) unnecessary _mapper_extension() function.


-Sean
Index: lib/sqlalchemy/mods/threadlocal.py
===================================================================
--- lib/sqlalchemy/mods/threadlocal.py	(revision 1502)
+++ lib/sqlalchemy/mods/threadlocal.py	(working copy)
@@ -29,18 +29,15 @@
 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()

Reply via email to