From 4b103b2ddaac8d0a71efaf873e4ec89c74105839 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Laura=20Kl=C3=BCnder?= Date: Wed, 19 Jul 2017 15:24:12 +0200 Subject: [PATCH] WrappedQueryset: exclude submodel-deleted LocationSlug instances --- src/c3nav/editor/wrappers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/c3nav/editor/wrappers.py b/src/c3nav/editor/wrappers.py index d097d568..ba8ab9f8 100644 --- a/src/c3nav/editor/wrappers.py +++ b/src/c3nav/editor/wrappers.py @@ -715,7 +715,8 @@ class ManagerWrapper(BaseQueryWrapper): make sure that the database does not return objects that have been deleted in this changeset """ qs = self._wrap_queryset(self._obj.model.objects.all()) - return qs.exclude(pk__in=self._changeset.deleted_existing.get(self._obj.model, ())) + return qs.exclude(pk__in=tuple(chain(*(self._changeset.deleted_existing.get(submodel, ()) + for submodel in get_submodels(self._obj.model))))) def delete(self): self.get_queryset().delete()