fix attribute errors
This commit is contained in:
parent
dee040734c
commit
c6ed3105b0
2 changed files with 2 additions and 2 deletions
|
@ -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:
|
||||
|
|
|
@ -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=()):
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue