[SQL-CVS] r4529 - in SQLObject/branches/1.2: docs sqlobject

[<sub...@co...>]

Author: phd
Date: Fri May 25 10:28:24 2012
New Revision: 4529

Log:
Fixed a bug in cache handling while unpickling (bug 3525829).

Modified:
   SQLObject/branches/1.2/docs/News.txt
   SQLObject/branches/1.2/sqlobject/main.py

Modified: SQLObject/branches/1.2/docs/News.txt
==============================================================================
--- SQLObject/branches/1.2/docs/News.txt	Thu May 24 13:37:48 2012	(r4528)
+++ SQLObject/branches/1.2/docs/News.txt	Fri May 25 10:28:24 2012	(r4529)
@@ -11,7 +11,9 @@
 ===============
 
 * Fixed a minor bug in PostgreSQL introspection: VIEWs don't have
-  PRIMARY KEYs - use sqlmeta.idName as the key if there is no one.
+  PRIMARY KEYs - use sqlmeta.idName as the key.
+
+* Fixed a bug in cache handling while unpickling.
 
 SQLObject 1.2.2
 ===============

Modified: SQLObject/branches/1.2/sqlobject/main.py
==============================================================================
--- SQLObject/branches/1.2/sqlobject/main.py	Thu May 24 13:37:48 2012	(r4528)
+++ SQLObject/branches/1.2/sqlobject/main.py	Fri May 25 10:28:24 2012	(r4529)
@@ -1670,6 +1670,8 @@
         return NotImplemented
 
 
+    # (De)serialization (pickle, etc.)
+
     def __getstate__(self):
         if self.sqlmeta._perConnection:
             from pickle import PicklingError
@@ -1680,10 +1682,16 @@
         return d
 
     def __setstate__(self, d):
+        id = d['id']
+        cls = self.__class__
+        cache = self._connection.cache
+        if cache.get(id, cls) is not None:
+            raise ValueError(
+                "Cannot unpickle %s row with id=%s - the id already exists in the cache" % (cls.__name__, id))
         self.__init__(_SO_fetch_no_create=1)
         self._SO_writeLock = threading.Lock()
         self.__dict__.update(d)
-        self.__class__._connection.cache.put(self.id, self.__class__, self)
+        cache.created(id, cls, self)
 
 
 def setterName(name):