DSRC-65: Call to action block

etna/api/tests/expected_results/article.json

@@ -214,7 +214,7 @@
                     {
                         "type": "featured_record_article",
                         "value": {
-                            "page": {}
+                            "page": null
                         },
                         "id": "b505f636-f3d1-4d4b-b368-69183e324e6e"
                     },

etna/core/blocks/__init__.py

@@ -1,5 +1,5 @@
 from .base import SectionDepthAwareStructBlock
-from .cta import LargeCardLinksBlock
+from .cta import ButtonBlock, CallToActionBlock, LargeCardLinksBlock
 from .featured_content import (
     FeaturedCollectionBlock,
     FeaturedRecordArticleBlock,
@@ -22,6 +22,8 @@
 __all__ = [
     "APIPageChooserBlock",
     "AuthorPromotedPagesBlock",
+    "ButtonBlock",
+    "CallToActionBlock",
     "ContentImageBlock",
     "FeaturedRecordArticleBlock",
     "FeaturedCollectionBlock",

etna/core/blocks/cta.py

@@ -1,8 +1,11 @@
+from django.conf import settings
+from django.core.exceptions import ValidationError
 from django.utils.translation import gettext_lazy as _
 
 from wagtail import blocks
 
 from .page_chooser import APIPageChooserBlock
+from .paragraph import APIRichTextBlock
 
 
 class LargeCardLinksBlock(blocks.StructBlock):
@@ -29,3 +32,52 @@ def get_context(self, value, parent_context=None):
             link_pages.append(page_2.specific)
         context["link_pages"] = link_pages
         return context
+
+
+class ButtonBlock(blocks.StructBlock):
+    label = blocks.CharBlock()
+    link = APIPageChooserBlock(required=False)
+    external_link = blocks.URLBlock(required=False)
+    accented = blocks.BooleanBlock(
+        required=False,
+        help_text="Use the accented button style",
+        label="Accented",
+    )
+
+    def clean(self, value):
+        data = super().clean(value)
+
+        if data.get("link") and data.get("external_link"):
+            raise ValidationError(
+                "You must provide either a page link or an external link, not both."
+            )
+        elif not (data.get("link") or data.get("external_link")):
+            raise ValidationError(
+                "You must provide either a page link or an external link."
+            )
+
+        return data
+
+    def get_api_representation(self, value, context=None):
+        representation = {
+            "label": value["label"],
+            "href": value.get("external_link") or value["link"].full_url,
+            "accent": value.get("accented") or False,
+        }
+
+        return representation
+
+    class Meta:
+        icon = "link"
+        label = "Button"
+
+
+class CallToActionBlock(blocks.StructBlock):
+    body = APIRichTextBlock(
+        max_length=100, features=settings.RESTRICTED_RICH_TEXT_FEATURES
+    )
+    button = ButtonBlock()
+
+    class Meta:
+        icon = "link"
+        label = "Call to action"

etna/core/serializers/pages.py

@@ -62,7 +62,7 @@ def get_api_data(object, required_api_fields: list = []) -> dict:
                 api_representation[field.name] = get_field_data(
                     object=specific, field=field
                 )
-    return api_representation
+    return api_representation or None
 
 
 class DefaultPageSerializer(serializers.Serializer):

etna/generic_pages/blocks.py

@@ -1,6 +1,8 @@
 from wagtail import blocks
 
 from etna.core.blocks import (
+    ButtonBlock,
+    CallToActionBlock,
     ContentImageBlock,
     FeaturedRecordArticleBlock,
     ParagraphBlock,
@@ -16,6 +18,8 @@
 
 
 class SectionContentBlock(blocks.StreamBlock):
+    button = ButtonBlock()
+    call_to_action = CallToActionBlock()
     featured_record_article = FeaturedRecordArticleBlock()
     image = ContentImageBlock()
     media = MediaBlock()