validate cover format on document create

[redakcja.git] / apps / catalogue / forms.py

diff --git a/apps/catalogue/forms.py b/apps/catalogue/forms.py
index 9301d9b..e6a6412 100644 (file)
--- a/apps/catalogue/forms.py
+++ b/apps/catalogue/forms.py
@@ -51,6 +51,12 @@ class DocumentCreateForm(forms.Form):
     #
     #     return self.cleaned_data
 
+    def clean_cover(self):
+        cover = self.cleaned_data['cover']
+        if cover.name.rsplit('.', 1)[-1].lower() not in ('jpg', 'jpeg', 'png', 'gif', 'tif', 'tiff'):
+            raise forms.ValidationError('The cover should be an image file (jpg/png/gif)')
+        return file
+
 
 class DocumentsUploadForm(forms.Form):
     """