fix pickling of ObjectId by defining __getstate__() and __setstate__() because __slots__() is defined

shearic · behackett · commit c9c54575ff1a · 2011-04-02T11:10:26.000-07:00
diff --git a/bson/objectid.py b/bson/objectid.py
@@ -187,6 +187,17 @@ def generation_time(self):
         t = struct.unpack(">i", self.__id[0:4])[0]
         return datetime.datetime.fromtimestamp(t, utc)
 
+    def __getstate__(self):
+        """return value of object for pickling.
+        needed explicitly because __slots__() defined.
+        """
+        return self.__str__()
+
+    def __setstate__(self, oid):
+        """explicit state set from pickling
+        """
+        self.__validate(oid)
+
     def __str__(self):
         return self.__id.encode("hex")
 
