controle de l'extention du fichier

zestedesavoir · Sep 1, 2014 · c846ed4 · c846ed4
1 parent 8a41a39
commit c846ed4
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/zds/gallery/forms.py b/zds/gallery/forms.py
@@ -171,6 +171,20 @@ def __init__(self, *args, **kwargs):
                 u'href="{{ gallery.get_absolute_url }}">Annuler</a>'),
             ),
         )
+
+    def clean(self):
+        cleaned_data = super(ArchiveImageForm, self).clean()
+
+        file = cleaned_data.get('file')
+        extension=file.name.split('.')[-1]
+
+        if extension != "zip":
+            self._errors['file'] = self.error_class(
+                [u"Le champ n'accepte que les fichiers zip"])
+            if 'file' in cleaned_data:
+                del cleaned_data['file']
+
+        return cleaned_data
 
 
 class UpdateImageForm(ImageForm):