Add custom get_api_representation for CaptionedImageBlock

Author: laymonage

bakerydemo/base/blocks.py

@@ -12,6 +12,17 @@
 from wagtail.images.blocks import ImageChooserBlock
 
 
+def get_image_api_representation(image):
+    return {
+        "id": image.pk,
+        "title": image.title,
+        "meta": {
+            "type": type(image)._meta.label,
+            "download_url": image.file.url,
+        },
+    }
+
+
 class CaptionedImageBlock(StructBlock):
     """
     Custom `StructBlock` for utilizing images with associated caption and
@@ -34,6 +45,11 @@ def get_preview_value(self):
             "caption": self.preview_image.description,
         }
 
+    def get_api_representation(self, value, context=None):
+        data = super().get_api_representation(value, context)
+        data["image"] = get_image_api_representation(value["image"])
+        return data
+
     class Meta:
         icon = "image"
         template = "blocks/captioned_image_block.html"

bakerydemo/recipes/blocks.py

@@ -13,7 +13,18 @@
 from wagtail.embeds.blocks import EmbedBlock
 from wagtail.images.blocks import ImageBlock
 
-from bakerydemo.base.blocks import BlockQuote, HeadingBlock
+from bakerydemo.base.blocks import (
+    BlockQuote,
+    HeadingBlock,
+    get_image_api_representation,
+)
+
+
+class CustomImageBlock(ImageBlock):
+    def get_api_representation(self, value, context=None):
+        data = super().get_api_representation(value, context)
+        data["image"] = get_image_api_representation(value)
+        return data
 
 
 class RecipeStepBlock(StructBlock):
@@ -57,7 +68,7 @@ class RecipeStreamBlock(StreamBlock):
             ("text", CharBlock()),
             ("numeric", FloatBlock()),
             ("rich_text", RichTextBlock()),
-            ("image", ImageBlock()),
+            ("image", CustomImageBlock()),
         ],
         group="Content",
         description=(
@@ -104,7 +115,7 @@ class RecipeStreamBlock(StreamBlock):
         },
     )
 
-    image_block = ImageBlock(group="Media")
+    image_block = CustomImageBlock(group="Media")
     embed_block = EmbedBlock(
         help_text="Insert an embed URL e.g https://www.youtube.com/watch?v=SGJFWirQ3ks",
         icon="media",