WEB-4122 : Override ILeadImageBehavior to limit image formats availab…

…le for upload

CHANGES.rst

@@ -5,7 +5,7 @@ Changelog
 1.2.18 (unreleased)
 -------------------
 
-- Nothing changed yet.
+- WEB-4122 : Override ILeadImageBehavior to limit image formats available for upload
 
 
 1.2.17 (2024-06-06)

src/imio/smartweb/common/behaviors/__init__.py

@@ -1 +1,2 @@
 # -*- coding: utf-8 -*-
+from .behaviors import ICustomLeadImage  # NOQA

src/imio/smartweb/common/behaviors/behaviors.py

@@ -0,0 +1,45 @@
+from plone.app.contenttypes import _
+from plone.autoform.interfaces import IFormFieldProvider
+from plone.namedfile.interfaces import INamedImage
+from plone.namedfile.field import NamedBlobImage
+from plone.app.contenttypes.behaviors.leadimage import ILeadImage
+from plone.app.contenttypes.behaviors.leadimage import ILeadImageBehavior
+from zope.interface import alsoProvides
+from zope.schema import ValidationError
+from zope.interface import implementer
+from zope.interface import provider
+
+
+class ICustomLeadImage(ILeadImage):
+    pass
+
+
+class InvalidImageFormat(ValidationError):
+    __doc__ = "Unsupported image format."
+
+
+def validate_image_format(value):
+    import pdb
+
+    pdb.set_trace()
+    allowed_extensions = [".gif", ".jpg", ".jpeg", ".png", ".svg", ".webp"]
+    if value:
+        filename = value.filename.lower()
+        if not any(filename.endswith(ext) for ext in allowed_extensions):
+            raise InvalidImageFormat
+    return True
+
+
+@implementer(ILeadImage)
+@provider(IFormFieldProvider)
+class ICustomLeadImageBehavior(ILeadImageBehavior):
+
+    image = NamedBlobImage(
+        title=_("label_leadimage", default="Lead Image"),
+        description="",
+        required=False,
+        constraint=validate_image_format,
+    )
+
+
+alsoProvides(ICustomLeadImageBehavior, INamedImage)

src/imio/smartweb/common/behaviors/configure.zcml

@@ -19,4 +19,15 @@
       provides=".iam.IAm"
       />
 
+  <plone:behavior
+      name="imio.smartweb.leadimage"
+      title="Lead Image with Custom Validation"
+      description="A lead image field with custom format validation."
+      factory="plone.app.contenttypes.behaviors.leadimage.LeadImage"
+      provides=".behaviors.ICustomLeadImageBehavior"
+      for="plone.dexterity.interfaces.IDexterityContent"
+      marker="plone.app.contenttypes.behaviors.leadimage.ILeadImage"
+      former_dotted_names="imio.smartweb.common.behaviors.ICustomLeadImage"
+      />
+
 </configure>