From 369a1b558b87e5540ece9753fc2c7aa7e82a4fcf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Laura=20Kl=C3=BCnder?= <laura@codingcatgirl.de>
Date: Fri, 7 Jul 2017 15:02:50 +0200
Subject: [PATCH] get-changed_object: accept ModelInstanceWrapper

---
 src/c3nav/editor/models/changeset.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/c3nav/editor/models/changeset.py b/src/c3nav/editor/models/changeset.py
index fab8c0af..0fed5387 100644
--- a/src/c3nav/editor/models/changeset.py
+++ b/src/c3nav/editor/models/changeset.py
@@ -20,7 +20,7 @@ from rest_framework.exceptions import PermissionDenied
 
 from c3nav.editor.models.changedobject import ChangedObject
 from c3nav.editor.utils import is_created_pk
-from c3nav.editor.wrappers import ModelWrapper
+from c3nav.editor.wrappers import ModelInstanceWrapper, ModelWrapper
 from c3nav.mapdata.models import LocationSlug, MapUpdate
 from c3nav.mapdata.models.locations import LocationRedirect
 from c3nav.mapdata.utils.models import get_submodels
@@ -317,6 +317,8 @@ class ChangeSet(models.Model):
         return r
 
     def get_changed_object(self, obj) -> ChangedObject:
+        if isinstance(obj, ModelInstanceWrapper):
+            obj = obj._obj
         model = obj.__class__
         pk = obj.pk
         if pk is None: