fix keyerror introduced by #3985

This commit is contained in:
Oskar Hahn 2018-12-17 13:34:17 +01:00
parent 6c9fb713e1
commit e09c5f415a
2 changed files with 10 additions and 1 deletions

View File

@ -13,6 +13,7 @@ class UsersAppConfig(AppConfig):
def ready(self):
# Import all required stuff.
from . import serializers # noqa
from ..core.signals import post_permission_creation, permission_change
from ..utils.rest_api import router
from .projector import get_projector_elements

View File

@ -135,5 +135,13 @@ class RESTModelMixin:
"""
Returns the full_data of the instance.
"""
serializer_class = model_serializer_classes[type(self)]
try:
serializer_class = model_serializer_classes[type(self)]
except KeyError:
# Because of the order of imports, it can happen, that the serializer
# for a model is not imported yet. Try to guess the name of the
# module and import it.
module_name = type(self).__module__.rsplit(".", 1)[0] + ".serializers"
__import__(module_name)
serializer_class = model_serializer_classes[type(self)]
return serializer_class(self).data