From 2ff226641734ca05897b28d22ac4b19bbd2c0abf Mon Sep 17 00:00:00 2001 From: ibuler Date: Thu, 21 May 2020 20:35:44 +0800 Subject: [PATCH] =?UTF-8?q?[Update]=20=E4=BF=AE=E6=94=B9=20SerializerMixin?= =?UTF-8?q?=20=E8=BF=98=E5=8E=9F=E4=B9=8B=E5=89=8D=E7=9A=84=E6=9B=B4?= =?UTF-8?q?=E6=94=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/common/mixins/api.py | 17 ++++++++--------- 1 file changed, 8 insertions(+), 9 deletions(-) diff --git a/apps/common/mixins/api.py b/apps/common/mixins/api.py index 544e24852..c49535ad6 100644 --- a/apps/common/mixins/api.py +++ b/apps/common/mixins/api.py @@ -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()