diff options
author | Magnus Hagander | 2018-12-14 11:43:50 +0000 |
---|---|---|
committer | Magnus Hagander | 2018-12-14 11:43:50 +0000 |
commit | 5e8a4a7ab78a9939b21d11094234231bda59567a (patch) | |
tree | 9d32974ce783aeecbae020d6391baef4fa376a95 /postgresqleu/util/diffablemodel.py | |
parent | 5e500d7d55ef048adc1dcf009db42d10c7ad53fc (diff) |
Replace tabs with spaces
In an effort to close up with PEP8, we should use spaces for indent
rather than tabs... Time to update your editor config!
Diffstat (limited to 'postgresqleu/util/diffablemodel.py')
-rw-r--r-- | postgresqleu/util/diffablemodel.py | 66 |
1 files changed, 33 insertions, 33 deletions
diff --git a/postgresqleu/util/diffablemodel.py b/postgresqleu/util/diffablemodel.py index bcdc1e4d..47d692d9 100644 --- a/postgresqleu/util/diffablemodel.py +++ b/postgresqleu/util/diffablemodel.py @@ -3,41 +3,41 @@ import django.db.models.fields.related class DiffableModel(object): - """ - Make it possible to diff a model. + """ + Make it possible to diff a model. """ - def __init__(self, *args, **kwargs): - super(DiffableModel, self).__init__(*args, **kwargs) - self.__initial = self._dict + 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]]) - # Foreign key lookups - for k,v in diffs.items(): - if type(self._meta.get_field(k)) is django.db.models.fields.related.ForeignKey: - # If it's a foreign key, look up the name again on ourselves. - # Since we only care about the *new* value, it's easy enough. - diffs[k] = (v[0], getattr(self, k)) - # Many to many lookups - 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 + @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]]) + # Foreign key lookups + for k,v in diffs.items(): + if type(self._meta.get_field(k)) is django.db.models.fields.related.ForeignKey: + # If it's a foreign key, look up the name again on ourselves. + # Since we only care about the *new* value, it's easy enough. + diffs[k] = (v[0], getattr(self, k)) + # Many to many lookups + 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 + 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) + @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) |