Import - New/Edit: verify associated images if valid zip #5330

1 files changed, 9 insertions, 0 deletions

diff --git a/ishtar_common/forms_common.py b/ishtar_common/forms_common.py
index 57efa72cd..df0a6d911 100644
--- a/ishtar_common/forms_common.py
+++ b/ishtar_common/forms_common.py
@@ -26,6 +26,7 @@ import requests
 import shutil
 import tempfile
 from urllib.parse import urlparse, quote
+from zipfile import ZipFile, BadZipFile
 
 from django import forms
 from django.conf import settings
@@ -221,6 +222,14 @@ class BaseImportForm(BSForm, forms.ModelForm):
                     "but not both."
                 )
             )
+        if data.get("imported_images"):
+            try:
+                images = data.get("imported_images")
+                zf = ZipFile(images)
+                zf.testzip()
+            except BadZipFile:
+                raise forms.ValidationError(_("Associated images must be zip file."))
+
         return data
 
     def clean_imported_images_link(self):