f

lmsege/lms/validators.py

@@ -0,0 +1,8 @@
+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.')