Add CMS toolbar menu for govplan

froide_govplan/cms_toolbars.py

@@ -0,0 +1,28 @@
+from django.urls import reverse
+from django.utils.translation import gettext_lazy as _
+
+from cms.toolbar_base import CMSToolbar
+from cms.toolbar_pool import toolbar_pool
+
+
+class GovPlanToolbar(CMSToolbar):
+    def populate(self):
+        if not hasattr(self.request, "govplan"):
+            return
+
+        menu = self.toolbar.get_or_create_menu("govplan-menu", _("GovPlan"))
+        govplan = self.request.govplan
+        url = reverse(
+            "govplan:admin:froide_govplan_governmentplan_change",
+            args=(govplan.pk,),
+            current_app="govplan",
+        )
+        menu.add_modal_item(_("Edit government plan"), url=url)
+        url = reverse(
+            "govplan:admin:froide_govplan_governmentplanupdate_add",
+            current_app="govplan",
+        )
+        menu.add_modal_item(_("Add update"), url=url)
+
+
+toolbar_pool.register(GovPlanToolbar)

froide_govplan/views.py

@@ -52,6 +52,8 @@ class GovPlanDetailView(GovernmentMixin, DetailView):
             "-timestamp"
         )
         context["section"] = self.get_section()
+        # For CMS toolbar
+        self.request.govplan = self.object
         return context