def create(self, validated_data): validated_data['filetype'] = Tools.checkMime(validated_data['attachment']) if validated_data['filetype'] == '': raise ValidationError({'detail': 'File type not supported'}) attach = Attach(**validated_data) attach.save() return attach
def update(self, instance, validated_data): if 'attachment' in validated_data: validated_data['filetype'] = Tools.checkMime(validated_data['attachment']) if validated_data['filetype'] == '': raise ValidationError({'detail': 'File type not supported'}) instance.__dict__.update(validated_data) instance.save() return instance