diff options
author | Magnus Hagander | 2013-08-18 17:12:11 +0000 |
---|---|---|
committer | Magnus Hagander | 2013-08-18 17:12:11 +0000 |
commit | fc14d745d311b43ac993af91a5914f545f2b27e5 (patch) | |
tree | 05c60158a7efbc656068ba24bb03b6de46e1b292 /pgcommitfest/commitfest/util.py | |
parent | 3fd84633c36e06958aef6f628c3e68422b973515 (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.py | 35 |
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) |