fix attribute errors

This commit is contained in:
Laura Klünder 2017-06-22 21:33:22 +02:00
parent dee040734c
commit c6ed3105b0
2 changed files with 2 additions and 2 deletions

View file

@ -16,6 +16,7 @@ from c3nav.editor.utils import is_created_pk
from c3nav.editor.wrappers import ModelWrapper
from c3nav.mapdata.models import LocationSlug
from c3nav.mapdata.models.locations import LocationRedirect
from c3nav.mapdata.utils.models import get_submodels
class ChangeSet(models.Model):
@ -283,7 +284,7 @@ class ChangeSet(models.Model):
objects = self.ever_created_objects if allow_deleted else self.created_objects
objects = tuple(obj for obj in ((submodel, objects.get(submodel, {}).get(pk, None))
for submodel in ModelWrapper.get_submodels(model)) if obj[1] is not None)
for submodel in get_submodels(model)) if obj[1] is not None)
if not objects:
raise model.DoesNotExist
if len(objects) > 1:

View file

@ -340,7 +340,6 @@ class BaseQueryWrapper(BaseWrapper):
"""
Get all created pks for this query's model an submodels.
"""
self.model.get_submodels(self.model._obj)
return reduce(operator.or_, (self._changeset.get_created_pks(model) for model in self.model._submodels))
def _wrap_queryset(self, queryset, created_pks=None, add_extra=()):