summaryrefslogtreecommitdiff
path: root/pgcommitfest/commitfest/util.py
diff options
context:
space:
mode:
authorMagnus Hagander2013-08-18 17:12:11 +0000
committerMagnus Hagander2013-08-18 17:12:11 +0000
commitfc14d745d311b43ac993af91a5914f545f2b27e5 (patch)
tree05c60158a7efbc656068ba24bb03b6de46e1b292 /pgcommitfest/commitfest/util.py
parent3fd84633c36e06958aef6f628c3e68422b973515 (diff)
Create history entries for fields that are changed on a patch
Diffstat (limited to 'pgcommitfest/commitfest/util.py')
-rw-r--r--pgcommitfest/commitfest/util.py35
1 files changed, 35 insertions, 0 deletions
diff --git a/pgcommitfest/commitfest/util.py b/pgcommitfest/commitfest/util.py
new file mode 100644
index 0000000..862ee3d
--- /dev/null
+++ b/pgcommitfest/commitfest/util.py
@@ -0,0 +1,35 @@
+from django.forms.models import model_to_dict
+
+
+class DiffableModel(object):
+ """
+ Make it possible to diff a model.
+ """
+
+ def __init__(self, *args, **kwargs):
+ super(DiffableModel, self).__init__(*args, **kwargs)
+ self.__initial = self._dict
+
+ @property
+ def diff(self):
+ manytomanyfieldnames = [f.name for f in self._meta.many_to_many]
+ d1 = self.__initial
+ d2 = self._dict
+ diffs = dict([(k, (v, d2[k])) for k, v in d1.items() if v != d2[k]])
+ if hasattr(self, 'map_manytomany_for_diff'):
+ for k,v in diffs.items():
+ if k in manytomanyfieldnames and self.map_manytomany_for_diff.has_key(k):
+ # Try to show the display name instead here
+ newvalue = getattr(self, self.map_manytomany_for_diff[k])
+ diffs[k] = (v[0], newvalue)
+ return diffs
+
+ def save(self, *args, **kwargs):
+ super(DiffableModel, self).save(*args, **kwargs)
+ self.__initial = self._dict
+
+ @property
+ def _dict(self):
+ fields = [field.name for field in self._meta.fields]
+ fields.extend([field.name for field in self._meta.many_to_many])
+ return model_to_dict(self, fields=fields)