new file system updates

cookbook/serializer.py

@@ -1,13 +1,12 @@
 from decimal import Decimal
+from gettext import gettext as _
 
 from django.contrib.auth.models import User
-from django.db.models import QuerySet
+from django.db.models import QuerySet, Sum
 from drf_writable_nested import (UniqueFieldsMixin,
                                  WritableNestedModelSerializer)
 from rest_framework import serializers
-from rest_framework.exceptions import ValidationError, NotAuthenticated, NotFound, ParseError
-from rest_framework.fields import ModelField
-from rest_framework.serializers import BaseSerializer, Serializer
+from rest_framework.exceptions import ValidationError, NotFound
 
 from cookbook.models import (Comment, CookLog, Food, Ingredient, Keyword,
                              MealPlan, MealType, NutritionInformation, Recipe,
@@ -105,11 +104,28 @@ class UserPreferenceSerializer(serializers.ModelSerializer):
 
 class UserFileSerializer(serializers.ModelSerializer):
 
+    def check_file_limit(self, validated_data):
+        if self.context['request'].space.max_file_storage_mb == -1:
+            raise ValidationError(_('File uploads are not enabled for this Space.'))
+
+        try:
+            current_file_size_mb = UserFile.objects.filter(space=self.context['request'].space).aggregate(Sum('file_size_kb'))['file_size_kb__sum'] / 1000
+        except TypeError:
+            current_file_size_mb = 0
+
+        if (validated_data['file'].size / 1000 / 1000 + current_file_size_mb - 5) > self.context['request'].space.max_file_storage_mb != 0:
+            raise ValidationError(_('You have reached your file upload limit.'))
+
     def create(self, validated_data):
+        self.check_file_limit(validated_data)
         validated_data['created_by'] = self.context['request'].user
         validated_data['space'] = self.context['request'].space
         return super().create(validated_data)
 
+    def update(self, instance, validated_data):
+        self.check_file_limit(validated_data)
+        return super().update(instance, validated_data)
+
     class Meta:
         model = UserFile
         fields = ('name', 'file', 'file_size_kb', 'id',)
@@ -117,6 +133,20 @@ class UserFileSerializer(serializers.ModelSerializer):
         extra_kwargs = {"file": {"required": False, }}
 
 
+class UserFileViewSerializer(serializers.ModelSerializer):
+
+    def create(self, validated_data):
+        raise ValidationError('Cannot create File over this view')
+
+    def update(self, instance, validated_data):
+        return instance
+
+    class Meta:
+        model = UserFile
+        fields = ('name', 'file', 'id',)
+        read_only_fields = ('id', 'file')
+
+
 class StorageSerializer(SpacedModelSerializer):
 
     def create(self, validated_data):
@@ -265,7 +295,7 @@ class StepSerializer(WritableNestedModelSerializer):
     ingredients = IngredientSerializer(many=True)
     ingredients_markdown = serializers.SerializerMethodField('get_ingredients_markdown')
     ingredients_vue = serializers.SerializerMethodField('get_ingredients_vue')
-    file = UserFileSerializer(allow_null=True)
+    file = UserFileViewSerializer(allow_null=True)
 
     def get_ingredients_vue(self, obj):
         return obj.get_instruction_render()