Allow to pass class to load() explicitly

This will allow to skip expensive class resolution, when class is already known.
zopefoundation · Jun 27, 2014 · f7bdc9c · f7bdc9c
1 parent 98e4326
commit f7bdc9c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/mongopersist/datamanager.py b/src/mongopersist/datamanager.py
@@ -278,8 +278,8 @@ def dump(self, obj):
             del self._registered_objects[id(obj)]
         return res
 
-    def load(self, dbref):
-        return self._reader.get_ghost(dbref)
+    def load(self, dbref, klass=None):
+        return self._reader.get_ghost(dbref, klass)
 
     def reset(self):
         root = self.root