9 lines
338 B
Python
9 lines
338 B
Python
import os
|
|
from django.core.exceptions import ValidationError
|
|
|
|
def validate_file_extension(value):
|
|
ext = os.path.splitext(value.name)[1]
|
|
valid_extensions = ['.pdf', '.jpg', '.jpeg', '.png']
|
|
if not ext.lower() in valid_extensions:
|
|
raise ValidationError('Unsupported file extension. Only PDF, JPG, and PNG are allowed.')
|