phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Added auto type

Browse Source
This commit is contained in:
dormantman 2020-01-31 00:35:51 +03:00
parent f83c7273d2
commit 4d2d08a6f4
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
apps/establishment/serializers/back.py
View File

@ -701,6 +701,7 @@ class MenuGallerySerializer(serializers.ModelSerializer):
class MenuFilesSerializers(ProjectModelSerializer): class MenuFilesSerializers(ProjectModelSerializer):
type = serializers.CharField(read_only=True)
class Meta: class Meta:
model = models.MenuFiles model = models.MenuFiles
@ -715,6 +716,11 @@ class MenuFilesSerializers(ProjectModelSerializer):
instance = models.MenuFiles.objects.create(**validated_data) instance = models.MenuFiles.objects.create(**validated_data)
return instance return instance
def validate(self, attrs):
extension = attrs['file'].name.split('.')[-1]
attrs['type'] = 'image' if extension in ('jpg', 'jpeg', 'png') else 'file'
return attrs
class MenuBackOfficeSerializer(serializers.ModelSerializer): class MenuBackOfficeSerializer(serializers.ModelSerializer):
name = serializers.CharField(source='category_translated') name = serializers.CharField(source='category_translated')