From 77e843a9bcd04d1f4b8a68b8a94baaafc06bed72 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9B=B9=E5=89=8D=E6=98=8E?= <909355014@qq.com> Date: Tue, 21 Jun 2022 18:30:36 +0800 Subject: [PATCH] customemodelserializer create bug --- apps/utils/serializers.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/apps/utils/serializers.py b/apps/utils/serializers.py index 4ea620da..3b9a1004 100755 --- a/apps/utils/serializers.py +++ b/apps/utils/serializers.py @@ -24,12 +24,13 @@ class CustomModelSerializer(DynamicFieldsMixin, serializers.ModelSerializer): def create(self, validated_data): if self.request: if getattr(self.request, 'user', None): - validated_data['create_by'] = self.request.user - if 'belong_dept' in validated_data: - pass - elif getattr(self.request.user, 'belong_dept', None): - if hasattr(self.Meta.model, 'belong_dept'): - validated_data['belong_dept'] = self.request.user.belong_dept + if getattr(self.request.user, 'create_by', None): + validated_data['create_by'] = self.request.user + if 'belong_dept' in validated_data: + pass + elif getattr(self.request.user, 'belong_dept', None): + if hasattr(self.Meta.model, 'belong_dept'): + validated_data['belong_dept'] = self.request.user.belong_dept return super().create(validated_data) def update(self, instance, validated_data):