customemodelserializer create bug

This commit is contained in:
曹前明 2022-06-21 18:30:36 +08:00
parent 3a76ff7bed
commit 77e843a9bc
1 changed files with 7 additions and 6 deletions

View File

@ -24,12 +24,13 @@ class CustomModelSerializer(DynamicFieldsMixin, serializers.ModelSerializer):
def create(self, validated_data): def create(self, validated_data):
if self.request: if self.request:
if getattr(self.request, 'user', None): if getattr(self.request, 'user', None):
validated_data['create_by'] = self.request.user if getattr(self.request.user, 'create_by', None):
if 'belong_dept' in validated_data: validated_data['create_by'] = self.request.user
pass if 'belong_dept' in validated_data:
elif getattr(self.request.user, 'belong_dept', None): pass
if hasattr(self.Meta.model, 'belong_dept'): elif getattr(self.request.user, 'belong_dept', None):
validated_data['belong_dept'] = self.request.user.belong_dept if hasattr(self.Meta.model, 'belong_dept'):
validated_data['belong_dept'] = self.request.user.belong_dept
return super().create(validated_data) return super().create(validated_data)
def update(self, instance, validated_data): def update(self, instance, validated_data):