Ticket #3369: patch_1.diff

django/db/models/manager.py

@@ -101 +101 @@
 
     def select_related(self, *args, **kwargs):
         return self.get_query_set().select_related(*args, **kwargs)
+    
+    def auto_cache(self, *args, **kwargs):
+        return self.get_query_set().auto_cache(*args, **kwargs)
 
     def values(self, *args, **kwargs):
         return self.get_query_set().values(*args, **kwargs)

django/db/models/fields/related.py

@@ -135 +135 @@
         # Set the value of the related field
         setattr(value, self.related.field.rel.get_related_field().attname, instance)
 
-        # Clear the cache, if it exists
+        # Update the cache, if it exists
         try:
-            delattr(value, self.related.field.get_cache_name())
+            setattr(value, self.related.field.get_cache_name(), value)
         except AttributeError:
             pass
 
@@ -182 +182 @@
             val = None
         setattr(instance, self.field.attname, val)
 
-        # Clear the cache, if it exists
+        # Update the cache, if it exists
         try:
-            delattr(instance, self.field.get_cache_name())
+            setattr(instance, self.field.get_cache_name(), value)
         except AttributeError:
             pass
 

django/db/models/query.py

@@ -91 +91 @@
         self._tables = []            # List of extra tables to use.
         self._offset = None          # OFFSET clause.
         self._limit = None           # LIMIT clause.
+        self._auto_cache = {}        # Dictionary of items to automatically cache on the fetched objects.
         self._result_cache = None
 
     ########################
@@ -192 +193 @@
                     obj = self.model(*row[:index_end])
                 for i, k in enumerate(extra_select):
                     setattr(obj, k[0], row[index_end+i])
+                # auto cache items
+                for key in self._auto_cache:
+                  setattr(obj, key, self._auto_cache[key])
                 yield obj
 
     def count(self):
@@ -386 +390 @@
     def distinct(self, true_or_false=True):
         "Returns a new QuerySet instance with '_distinct' modified."
         return self._clone(_distinct=true_or_false)
+    
+    def auto_cache(self, **kwargs):
+        "Returns a new QuerySet instance with _auto_cache modified."
+        return self._clone(_auto_cache=kwargs)
 
     def extra(self, select=None, where=None, params=None, tables=None):
         assert self._limit is None and self._offset is None, \
@@ -416 +424 @@
         c._tables = self._tables[:]
         c._offset = self._offset
         c._limit = self._limit
+        c._auto_cache = self._auto_cache.copy()
         c.__dict__.update(kwargs)
         return c
 

tests/modeltests/many_to_one/models.py

@@ -245 +245 @@
 >>> Reporter.objects.filter(article__reporter=r).distinct()
 [<Reporter: John Smith>]
 
+# Auto caching
+>>> r_autocache = Reporter.objects.all()[0]
+>>> r_autocache
+<Reporter: John Smith>
+>>> a_autocache = r_autocache.article_set.auto_cache(reporter=r_autocache)[0]
+>>> a_autocache
+<Article: John's second story>
+>>> a_autocache.reporter is r_autocache
+True
+
 # If you delete a reporter, his articles will be deleted.
 >>> Article.objects.all()
 [<Article: John's second story>, <Article: Paul's story>, <Article: This is a test>, <Article: This is a test>, <Article: This is a test>]