seserialize QuerySetWrapper as well

This commit is contained in:
Laura Klünder 2018-11-29 18:39:33 +01:00
parent e059d3a9bd
commit e90d02eaa3

View file

@ -17,6 +17,7 @@ from django.utils.translation import ugettext_lazy as _
from rest_framework.response import Response as APIResponse from rest_framework.response import Response as APIResponse
from c3nav.editor.models import ChangeSet from c3nav.editor.models import ChangeSet
from c3nav.editor.wrappers import QuerySetWrapper
from c3nav.mapdata.models.access import AccessPermission from c3nav.mapdata.models.access import AccessPermission
from c3nav.mapdata.models.base import SerializableMixin from c3nav.mapdata.models.base import SerializableMixin
from c3nav.mapdata.utils.user import can_access_editor from c3nav.mapdata.utils.user import can_access_editor
@ -225,7 +226,7 @@ class APIHybridTemplateContextResponse(APIHybridResponse):
def _maybe_serialize_value(self, value): def _maybe_serialize_value(self, value):
if isinstance(value, SerializableMixin): if isinstance(value, SerializableMixin):
value = value.serialize(geometry=False, detailed=False) value = value.serialize(geometry=False, detailed=False)
elif isinstance(value, QuerySet) and issubclass(value.model, SerializableMixin): elif isinstance(value, (QuerySet, QuerySetWrapper)) and issubclass(value.model, SerializableMixin):
value = [item.serialize(geometry=False, detailed=False) for item in value] value = [item.serialize(geometry=False, detailed=False) for item in value]
return value return value