MainMenu: add extra+extra_evaluate

Author: boxed

iommi/experimental/main_menu.py

@@ -62,12 +62,16 @@ class MainMenu:
         items=EMPTY,
         attrs__style=EMPTY,
         attrs__class=EMPTY,
+        extra=EMPTY,
+        extra_evaluated=EMPTY,
     )
-    def __init__(self, *, items=None, attrs, template=MISSING, paths=None):
+    def __init__(self, *, items=None, attrs, template=MISSING, paths=None, extra, extra_evaluated):
         self.items = items
         self.attrs = attrs
         self.template = template
         self.paths = paths or []
+        self.extra = extra
+        self.extra_evaluated = extra_evaluated
         for name, c in self.items.items():
             c.parent = self
             c._set_name(name)
@@ -178,6 +182,14 @@ def __html__(self):
     def render_items(self):
         return format_html('{}' * len(self.items), *[x.__html__() for x in self.items.values()])
 
+    @property
+    def extra(self):
+        return self.main_menu.extra
+
+    @cached_property
+    def extra_evaluated(self):
+        return Struct(evaluate_as_needed(self.main_menu.extra_evaluated or {}, self._own_evaluate_parameters))
+
     def __repr__(self):
         return '<BoundMainMenu>'
 
@@ -337,6 +349,7 @@ def own_evaluate_parameters(self):
             'request': request,
             'user': request.user if request else None,
             'item': self,
+            'main_menu': self.root,
             **getattr(request, 'iommi_view_params', {}),
         }
 

iommi/experimental/main_menu__tests.py

@@ -301,3 +301,59 @@ def test_m_url_error_case():
         ).urlpatterns()
 
     assert str(e.value) == "External links can't have subitems"
+
+
+def test_m_extra():
+    menu = MainMenu(
+        items=dict(
+            foo=M(
+                view=EXTERNAL,
+                url=lambda item, **_: item.extra.url,
+                extra__url='https://example.com',
+            ),
+        ),
+    ).bind(request=req('get'))
+
+    assert 'https://example.com' in str(menu)
+
+
+def test_m_extra_evaluated():
+    menu = MainMenu(
+        items=dict(
+            foo=M(
+                view=EXTERNAL,
+                url=lambda item, **_: item.extra_evaluated.url,
+                extra_evaluated__url=lambda **_: 'https://example.com',
+            ),
+        ),
+    ).bind(request=req('get'))
+
+    assert 'https://example.com' in str(menu)
+
+
+def test_main_menu_extra():
+    menu = MainMenu(
+        extra__url='https://example.com',
+        items=dict(
+            foo=M(
+                view=EXTERNAL,
+                url=lambda main_menu, **_: main_menu.extra.url,
+            ),
+        ),
+    ).bind(request=req('get'))
+
+    assert 'https://example.com' in str(menu)
+
+
+def test_main_menu_extra_evaluated():
+    menu = MainMenu(
+        extra_evaluated__url=lambda **_: 'https://example.com',
+        items=dict(
+            foo=M(
+                view=EXTERNAL,
+                url=lambda main_menu, **_: main_menu.extra_evaluated.url,
+            ),
+        ),
+    ).bind(request=req('get'))
+
+    assert 'https://example.com' in str(menu)