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
+Update
-        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
+Update
-        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>]