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):