fix ModelInstanceWrapper foreignkey values

This commit is contained in:
Laura Klünder 2017-06-13 13:03:36 +02:00
parent 4e1469c7d4
commit 66596aac63
2 changed files with 6 additions and 4 deletions

View file

@ -98,7 +98,7 @@ class ChangeSet(models.Model):
for name, value in kwargs.items():
setattr(change, name, value)
change.save()
# print(repr(change))
print(repr(change))
return change
def add_create(self, obj, author=None):

View file

@ -77,7 +77,7 @@ class ModelInstanceWrapper(BaseWrapper):
continue
self._initial_values[field] = getattr(self, field.name)
elif (field.many_to_one or field.one_to_one) and not field.primary_key:
self._initial_values[field] = getattr(self, field.name).pk
self._initial_values[field] = getattr(self, field.name)
def __eq__(self, other):
if type(other) == ModelWrapper:
@ -117,8 +117,10 @@ class ModelInstanceWrapper(BaseWrapper):
for field, initial_value in self._initial_values.items():
new_value = getattr(self._obj, field.name)
if field.related_model:
if new_value.pk != initial_value.pk:
self._changeset.add_update(self, name=field.name, value=new_value.pk, author=author)
new_pk = None if new_value is None else new_value.pk
initial_pk = None if initial_value is None else initial_value.pk
if new_pk != initial_pk:
self._changeset.add_update(self, name=field.name, value=new_pk, author=author)
continue
if new_value == initial_value: