Don't allow file uploads bigger than 10MB

This commit is contained in:
Andrew Godwin 2022-12-02 04:02:38 -07:00
parent 630c9df113
commit 9f3ad0be16
2 changed files with 12 additions and 1 deletions

View File

@ -156,6 +156,17 @@ class ImageUpload(FormView):
image = forms.ImageField() image = forms.ImageField()
description = forms.CharField(required=False) description = forms.CharField(required=False)
def clean_image(self):
value = self.cleaned_data["image"]
if value.size > 1024 * 1024 * 1:
# Erase the file from our data to stop trying to show it again
self.files = {}
raise forms.ValidationError("File must be 10MB or less")
return value
def form_invalid(self, form):
return super().form_invalid(form)
def form_valid(self, form): def form_valid(self, form):
# Make a PostAttachment # Make a PostAttachment
thumbnail_file = resize_image(form.cleaned_data["image"], size=(400, 225)) thumbnail_file = resize_image(form.cleaned_data["image"], size=(400, 225))