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
