[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:
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
if self.action in ['list', 'metadata'] and self.request.query_params.get('draw'):
serializer_class = self.serializer_classes.get('display')
if serializer_class is None:
serializer_class = self.serializer_classes.get(
self.action, self.serializer_classes.get('default')
)
if hasattr(self, 'serializer_classes') and isinstance(self.serializer_classes, dict):
if self.action in ['list', 'metadata'] and self.request.query_params.get('draw'):
serializer_class = self.serializer_classes.get('display')
if serializer_class is None:
serializer_class = self.serializer_classes.get(
self.action, self.serializer_classes.get('default')
)
print(serializer_class)
if serializer_class:
return serializer_class
return super().get_serializer_class()