CHORE: YouTube Block

etna/core/blocks/__init__.py

@@ -21,6 +21,7 @@
 from .quote import QuoteBlock
 from .section import SectionBlock, SubHeadingBlock
 from .text import InsetTextBlock, WarningTextBlock
+from .video import YouTubeBlock
 
 __all__ = [
     "APIPageChooserBlock",
@@ -49,4 +50,5 @@
     "SectionDepthAwareStructBlock",
     "SubHeadingBlock",
     "WarningTextBlock",
+    "YouTubeBlock",
 ]

etna/core/blocks/video.py

@@ -0,0 +1,26 @@
+import re
+
+from django.core.exceptions import ValidationError
+
+from wagtail import blocks
+
+from .image import APIImageChooserBlock
+
+
+class YouTubeBlock(blocks.StructBlock):
+    title = blocks.CharBlock(required=True, max_length=100, label="Title")
+    video_id = blocks.CharBlock(required=True, max_length=11, label="YouTube Video ID")
+    preview_image = APIImageChooserBlock(
+        rendition_size="max-640x360", required=False, label="Preview Image"
+    )
+
+    def clean(self, value):
+        if video_id := value.get("video_id"):
+            if not re.match(r"^[a-zA-Z0-9_-]{11}$", video_id):
+                raise ValidationError("Invalid YouTube Video ID")
+
+        return super().clean(value)
+
+    class Meta:
+        label = "YouTube Video"
+        icon = "media"

etna/generic_pages/blocks.py

@@ -15,6 +15,7 @@
     SectionDepthAwareStructBlock,
     SubHeadingBlock,
     WarningTextBlock,
+    YouTubeBlock,
 )
 
 from ..media.blocks import MediaBlock
@@ -37,6 +38,7 @@ class SectionContentBlock(blocks.StreamBlock):
     record_links = RecordLinksBlock()
     sub_heading = SubHeadingBlock()
     warning_text = WarningTextBlock()
+    youtube_video = YouTubeBlock()
 
 
 class ContentSectionBlock(SectionDepthAwareStructBlock):