[Update] 修改 SerializerMixin 还原之前的更改

pull/4033/head
ibuler 2020-05-21 20:35:44 +08:00
parent 3320e6105c
commit 2ff2266417
1 changed files with 8 additions and 9 deletions

View File

@ -27,16 +27,15 @@ class JSONResponseMixin(object):
class SerializerMixin: class SerializerMixin:
def get_serializer_class(self): def get_serializer_class(self):
if not hasattr(self, 'serializer_classes') or isinstance(self.serializer_classes, dict):
return super().get_serializer_class()
serializer_class = None serializer_class = None
if hasattr(self, 'serializer_classes') and isinstance(self.serializer_classes, dict):
if self.action in ['list', 'metadata'] and self.request.query_params.get('draw'): if self.action in ['list', 'metadata'] and self.request.query_params.get('draw'):
serializer_class = self.serializer_classes.get('display') serializer_class = self.serializer_classes.get('display')
if serializer_class is None: if serializer_class is None:
serializer_class = self.serializer_classes.get( serializer_class = self.serializer_classes.get(
self.action, self.serializer_classes.get('default') self.action, self.serializer_classes.get('default')
) )
print(serializer_class)
if serializer_class: if serializer_class:
return serializer_class return serializer_class
return super().get_serializer_class() return super().get_serializer_class()