Ticket #4102: 4102-dirty.4.patch

django/db/models/base.py

@@ -94 +94 @@
-    def __ne__(self, other):
-        return not self.__eq__(other)
-
+    _dirty = {}
+
+    def __setattr__(self, name, value):
+        if name != '_dirty' and (not hasattr(self, name) or value != getattr(self, name)):
+            self._dirty[name] = True
+        super(Model, self).__setattr__(name, value)
+
+    def _wash(self):
+        self._dirty = {}
+
-    def __init__(self, *args, **kwargs):
-        dispatcher.send(signal=signals.pre_init, sender=self.__class__, args=args, kwargs=kwargs)
-
+        wash = kwargs.pop('wash', False)
+        dirty = {}
+
-        # There is a rather weird disparity here; if kwargs, it's set, then args
-
+.
-        # The reason for the kwargs check is that standard iterator passes in by
-
+i
-        args_len = len(args)
-        if args_len > len(self._meta.fields):
-            # Daft, but matches old exception sans the err msg.
@@ -114 +127 @@
-            # without changing the logic.
-            for val, field in izip(args, fields_iter):
-                setattr(self, field.attname, val)
+                dirty[field.attname] = True
-        else:
-            # Slower, kwargs-ready version.
-            for val, field in izip(args, fields_iter):
-                setattr(self, field.attname, val)
+                dirty[field.attname] = True
-                kwargs.pop(field.name, None)
-                # Maintain compatibility with existing calls.
-                if isinstance(field.rel, ManyToOneRel):
@@ -164 +179 @@
-                    pass
-            if kwargs:
-                raise TypeError, "'%s' is an invalid keyword argument for this function" % kwargs.keys()[0]
+
+        self._wash()
+        if not wash:
+            self._dirty = dirty
-        dispatcher.send(signal=signals.post_init, sender=self.__class__, instance=self)
-
-    def add_to_class(cls, name, value):
@@ -199 +218 @@
-    _prepare = classmethod(_prepare)
-
-    def save(self):
+        "Save model object to database, creating if doesn't exist, otherwise updating"
-        dispatcher.send(signal=signals.pre_save, sender=self.__class__, instance=self)
-
-
+
+
-        cursor = connection.cursor()
-
-        # First, try an UPDATE. If that doesn't update anything, do an INSERT.

django/db/models/query.py

@@ -201 +201 @@
-                    obj, index_end = get_cached_row(klass=self.model, row=row,
-                                                    index_start=0, max_depth=self._max_related_depth)
-                else:
-
+wash=True, 
-                for i, k in enumerate(extra_select):
-                    setattr(obj, k[0], row[index_end+i])
-                yield obj

tests/modeltests/update_fields/models.py

@@ +1 @@
+"""
+#. Save only modified fields on update
+"""
+from django.db import models
+
+class Article(models.Model):
+    headline = models.CharField(maxlength=100, default='Default headline')
+    body = models.TextField()
+    pub_date = models.DateTimeField()
+
+    class Meta:
+        ordering = ('pub_date','headline')
+
+    def __str__(self):
+        return self.headline
+
+__test__ = {'API_TESTS':"""
+# Create a couple of Articles.
+>>> from datetime import datetime
+>>> a = Article(headline='Article 1', body='Body 1', pub_date=datetime(2007, 5, 5))
+>>> a.save()
+
+# Change headline and body on different instances, then save both to verify
+# that each only updated the modified field
+>>> a1 = Article.objects.get(pk=a.id)
+>>> a2 = Article.objects.get(pk=a.id)
+>>> a1.headline = 'Changed article 1'
+>>> a2.body = 'Changed body 1'
+>>> a1.save()
+>>> a2.save()
+>>> a3 = Article.objects.get(pk=a.id)
+>>> a3.headline
+'Changed article 1'
+>>> a3.body
+'Changed body 1'
+"""}