DSRC-15: Table Block (#1679)

nationalarchives · Jul 16, 2024 · 8d3db08 · 8d3db08
1 parent 08644f7
commit 8d3db08
Show file tree

Hide file tree

Showing 5 changed files with 694 additions and 0 deletions.
diff --git a/config/settings/base.py b/config/settings/base.py
@@ -68,6 +68,7 @@
     "etna.whatson",
     "wagtail.contrib.forms",
     "wagtail.contrib.redirects",
+    "wagtail.contrib.table_block",
     "wagtail.embeds",
     "wagtail.sites",
     "wagtail.users",

diff --git a/etna/core/blocks/__init__.py b/etna/core/blocks/__init__.py
@@ -20,6 +20,7 @@
 )
 from .quote import QuoteBlock
 from .section import SectionBlock, SubHeadingBlock
+from .tables import ContentTableBlock
 from .text import InsetTextBlock, WarningTextBlock
 from .video import YouTubeBlock
 
@@ -29,6 +30,7 @@
     "ButtonBlock",
     "CallToActionBlock",
     "ContentImageBlock",
+    "ContentTableBlock",
     "DocumentsBlock",
     "DoDontListBlock",
     "FeaturedRecordArticleBlock",

diff --git a/etna/core/blocks/tables.py b/etna/core/blocks/tables.py
@@ -0,0 +1,34 @@
+from wagtail import blocks
+from wagtail.contrib.table_block.blocks import TableBlock
+
+
+class ContentTableBlock(blocks.StructBlock):
+    title = blocks.CharBlock(
+        max_length=100,
+        help_text="Title of the table",
+        label="Title",
+        required=False,
+    )
+    table = TableBlock(
+        table_options={
+            "contextMenu": [
+                "row_above",
+                "row_below",
+                "---------",
+                "col_left",
+                "col_right",
+                "---------",
+                "remove_row",
+                "remove_col",
+                "---------",
+                "undo",
+                "redo",
+                "---------",
+                "alignment",
+            ]
+        }
+    )
+
+    class Meta:
+        icon = "table"
+        label = "Table"
diff --git a/etna/generic_pages/blocks.py b/etna/generic_pages/blocks.py
@@ -4,6 +4,7 @@
     ButtonBlock,
     CallToActionBlock,
     ContentImageBlock,
+    ContentTableBlock,
     DocumentsBlock,
     DoDontListBlock,
     FeaturedRecordArticleBlock,
@@ -37,6 +38,7 @@ class SectionContentBlock(blocks.StreamBlock):
     quote = QuoteBlock()
     record_links = RecordLinksBlock()
     sub_heading = SubHeadingBlock()
+    table = ContentTableBlock()
     warning_text = WarningTextBlock()
     youtube_video = YouTubeBlock()