Wagtail menus

Author: marcoacierno

backend/api/cms/menu/__init__.py

@@ -0,0 +1,28 @@
+from api.cms.utils import get_site_by_host
+from api.context import Context
+import strawberry
+
+from wagtailmenus.models import MainMenu, FlatMenu
+from api.cms.menu.types import CMSMenu
+
+
+@strawberry.field
+def cms_menu(
+    info: strawberry.Info[Context],
+    hostname: str,
+    handle: str,
+) -> CMSMenu | None:
+    site = get_site_by_host(hostname)
+
+    if not site:
+        return None
+
+    if handle == "main":
+        menu = MainMenu.get_for_site(site)
+    else:
+        menu = FlatMenu.get_for_site(site=site, handle=handle)
+
+    if not menu:
+        return None
+
+    return CMSMenu.from_model(menu)

backend/api/cms/menu/queries/__init__.py

@@ -0,0 +1,34 @@
+from typing import Self
+from strawberry import ID
+import strawberry
+from wagtailmenus.models import MainMenu, FlatMenu
+
+
+@strawberry.type
+class CMSMenuItem:
+    id: ID
+    title: str
+
+    @classmethod
+    def from_model(cls, item) -> Self:
+        return cls(
+            id=item.id,
+            title=item.title,
+        )
+
+
+@strawberry.type
+class CMSMenu:
+    id: ID
+    title: str
+    items: list[CMSMenuItem]
+
+    @classmethod
+    def from_model(cls, menu: MainMenu | FlatMenu) -> Self:
+        title = menu.title if isinstance(menu, FlatMenu) else "Main"
+
+        return cls(
+            id=menu.id,
+            title=title,
+            items=[CMSMenuItem.from_model(item) for item in menu.items],
+        )

backend/api/cms/menu/queries/cms_menu.py

@@ -5,7 +5,9 @@
 from api.cms.page.queries.cms_pages import cms_pages
 from api.cms.news.queries.news_articles import news_articles
 from api.cms.news.queries.news_article import news_article
+from api.cms.menu.queries.cms_menu import cms_menu
 
 CMSQuery = create_type(
-    "CMSQuery", fields=[cms_page, cms_pages, news_articles, news_article, page_preview]
+    "CMSQuery",
+    fields=[cms_page, cms_pages, news_articles, news_article, page_preview, cms_menu],
 )

backend/api/cms/menu/types.py

@@ -75,6 +75,7 @@
     "modelcluster",
     "taggit",
     "wagtail_headless_preview",
+    "wagtailmenus",
     # --
     "schedule.apps.ScheduleConfig",
     "custom_admin",
@@ -180,6 +181,7 @@
                 "django.contrib.auth.context_processors.auth",
                 "django.contrib.messages.context_processors.messages",
                 "custom_admin.context_processors.admin_settings",
+                "wagtailmenus.context_processors.wagtailmenus",
             ],
             "builtins": [
                 "custom_admin.templatetags.to_json_for_prop",

backend/api/cms/schema.py

@@ -82,7 +82,7 @@ dependencies = [
     "stripe==10.5.0",
     "djangorestframework==3.15.2",
     "l18n<2022.0,>=2021.3",
-    "wagtail==6.2.2",
+    "wagtail==7.0.1",
     "wagtail-localize==1.10",
     "celery==5.4.0",
     "wagtail-headless-preview==0.8.0",
@@ -104,6 +104,7 @@ dependencies = [
     "weasyprint>=63.1",
     "opencv-python-headless>=4.10.0.84",
     "psycopg[c]>=3.2.3",
+    "wagtailmenus>=4.0.2",
 ]
 name = "backend"
 version = "0.1.0"