feat: 仅支持fields_size=mini,small

2020-06-01 10:10:28 +08:00 · 2020-06-01 10:10:28 +08:00 · ad1c17aa7b
parent f8142e23cd
commit ad1c17aa7b
1 changed files with 2 additions and 0 deletions
--- a/apps/common/mixins/serializers.py
+++ b/apps/common/mixins/serializers.py
@ -194,6 +194,8 @@ class SizedModelFieldsMixin(BaseDynamicFieldsPlugin):
        size = query_params.get(self.arg_name)
        if not size:
            return []
+        if size not in ['mini', 'small']:
+            return []
        size_fields = getattr(self.serializer.Meta, 'fields_{}'.format(size), None)
        if not size_fields or not isinstance(size_fields, Iterable):
            return []