2025-summer/team-3
12
Fork 0
Code Issues Pull requests Projects Releases Activity

correctly process relation updates in ModelInstanceWrapper

Browse source
This commit is contained in:
Laura Klünder 2017-06-13 23:38:56 +02:00
parent 5b3d6ffbc4
commit 9277c32afc
1 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/c3nav/editor/wrappers.py
View file

@ -121,9 +121,15 @@ class ModelInstanceWrapper(BaseWrapper):
self._initial_values[field] = getattr(self._obj, field.name)
elif (field.many_to_one or field.one_to_one) and not field.primary_key:
if field.name in updates:
obj = self._wrap_model(field.model).get(pk=updates[field.name])
setattr(self._obj, field.name, obj)
setattr(self._obj, field.attname, obj.pk)
value_pk = updates[field.name]
class_value = getattr(type(self._obj), field.name, None)
if isinstance(value_pk, str):
obj = self._wrap_model(field.model).get(pk=value_pk)
setattr(self._obj, class_value.cache_name, obj)
setattr(self._obj, field.attname, obj.pk)
else:
delattr(self._obj, class_value.cache_name, obj)
setattr(self._obj, field.attname, value_pk)
self._initial_values[field] = getattr(self._obj, field.attname)
def __eq__(self, other):