-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
add option to limit the file size for image and file uploads
- add a validator to validate the max file size (FileMaxSizeValidator) - return the first form error in a JsonResponse so it is actually shown to the user instead of a generic error message. - add separate test_settings for pytest (to allow testing with small file sizes) - add tests for file extension and file size
- Loading branch information
Showing
11 changed files
with
1,019 additions
and
357 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,40 @@ | ||
from django import get_version | ||
from django.core.exceptions import ValidationError | ||
from django.utils.deconstruct import deconstructible | ||
|
||
if get_version() >= "4.0": | ||
from django.utils.translation import gettext_lazy as _ | ||
else: | ||
from django.utils.translation import ugettext_lazy as _ | ||
|
||
|
||
@deconstructible() | ||
class FileMaxSizeValidator: | ||
"""Validate that a file is not bigger than max_size mb, otherwise raise ValidationError. | ||
If zero is passed for max_size any file size is allowed. | ||
""" | ||
|
||
message = _("File should be at most %(max_size)s MB.") | ||
code = "invalid_size" | ||
|
||
def __init__(self, max_size): | ||
self.max_size = max_size * 1024 * 1024 | ||
self.orig_max_size = max_size | ||
|
||
def __call__(self, value): | ||
if value.size > self.max_size > 0: | ||
raise ValidationError( | ||
self.message, | ||
code=self.code, | ||
params={ | ||
"max_size": self.orig_max_size, | ||
}, | ||
) | ||
|
||
def __eq__(self, other): | ||
return ( | ||
isinstance(other, self.__class__) | ||
and self.max_size == other.max_size | ||
and self.message == other.message | ||
and self.code == other.code | ||
) |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.