encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-07-18 04:02:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add support for allowed file extensions

Browse Source
This commit is contained in:
amirhusein1996 2024-01-16 15:50:18 +03:30
parent 21bb21b65b
commit 1805bb42b9
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
rest_framework/fields.py
View File

@ -16,7 +16,7 @@ from django.core.exceptions import ValidationError as DjangoValidationError
from django.core.validators import ( from django.core.validators import (
EmailValidator, MaxLengthValidator, MaxValueValidator, MinLengthValidator, EmailValidator, MaxLengthValidator, MaxValueValidator, MinLengthValidator,
MinValueValidator, ProhibitNullCharactersValidator, RegexValidator, MinValueValidator, ProhibitNullCharactersValidator, RegexValidator,
URLValidator URLValidator, FileExtensionValidator
) )
from django.forms import FilePathField as DjangoFilePathField from django.forms import FilePathField as DjangoFilePathField
from django.forms import ImageField as DjangoImageField from django.forms import ImageField as DjangoImageField
@ -1516,9 +1516,17 @@ class FileField(Field):
def __init__(self, **kwargs): def __init__(self, **kwargs):
self.max_length = kwargs.pop('max_length', None) self.max_length = kwargs.pop('max_length', None)
self.allow_empty_file = kwargs.pop('allow_empty_file', False) self.allow_empty_file = kwargs.pop('allow_empty_file', False)
allowed_extensions = kwargs.pop('allowed_extensions', None)
assert allowed_extensions is None or isinstance(allowed_extensions, (list, tuple)), (
'Invalid value for allowed_extensions. It must be either None or a list/tuple, but %s provided.'
% (allowed_extensions.__name__ if inspect.isclass(allowed_extensions) else allowed_extensions.__class__.__name__)
)
if 'use_url' in kwargs: if 'use_url' in kwargs:
self.use_url = kwargs.pop('use_url') self.use_url = kwargs.pop('use_url')
super().__init__(**kwargs) super().__init__(**kwargs)
if allowed_extensions is not None:
self.validators.append(
FileExtensionValidator(allowed_extensions))
def to_internal_value(self, data): def to_internal_value(self, data):
try: try: