holm/django-helpdesk
1
0
Fork 0
forked from extern/django-helpdesk
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updated validate_file_extension() to look for defined VALID_EXTENTIONS in settings.py so its configurable

Browse Source
This commit is contained in:
Thomas Wheeler 2021-12-31 14:59:37 -08:00
parent 94902ec44f
commit 345a713777
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
helpdesk/validators.py
View File

@ -2,14 +2,22 @@
# #
# validators for file uploads, etc. # validators for file uploads, etc.
from django.conf import settings
def validate_file_extension(value): def validate_file_extension(value):
import os import os
from django.core.exceptions import ValidationError from django.core.exceptions import ValidationError
ext = os.path.splitext(value.name)[1] # [0] returns path+filename ext = os.path.splitext(value.name)[1] # [0] returns path+filename
valid_extensions = ['.txt', '.pdf', '.doc', '.docx', '.odt', '.jpg', '.png']
# TODO: we might improve this with more thorough checks of file types # TODO: we might improve this with more thorough checks of file types
# rather than just the extensions. # rather than just the extensions.
if not ext.lower() in valid_extensions:
# check if VALID_EXTENSTIONS is defined in settings.py
# if not use defaults
if settings.VALID_EXTENSTIONS:
valid_extenstions = settings.VALID_EXTENSTIONS
else:
valid_extenstions = ['.txt', '.pdf', '.doc', '.docx', '.odt', '.jpg', '.png']
if not ext.lower() in valid_extenstions:
raise ValidationError('Unsupported file extension.') raise ValidationError('Unsupported file extension.')