This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
authentik/passbook/admin/api/v1/utils.py

18 lines
501 B
Python
Raw Normal View History

2018-11-16 08:10:35 +00:00
from django.db.models import Model
from rest_framework.serializers import ModelSerializer
class LookupSerializer(ModelSerializer):
mapping = {}
def to_representation(self, instance):
for __model, __serializer in self.mapping.items():
if isinstance(instance, __model):
return __serializer(instance=instance).to_representation(instance)
raise KeyError(instance.__class__.__name__)
class Meta:
model = Model
fields = '__all__'