Add toolbar item on main govplan page

This commit is contained in:
Stefan Wehrmeyer 2022-03-16 13:11:02 +01:00
parent 475f2fe4c1
commit e7d5496254

View file

@ -7,22 +7,30 @@ from cms.toolbar_pool import toolbar_pool
class GovPlanToolbar(CMSToolbar): class GovPlanToolbar(CMSToolbar):
def populate(self): def populate(self):
if not hasattr(self.request, "govplan"): if self.request.current_page.application_namespace != "govplan":
return return
menu = self.toolbar.get_or_create_menu("govplan-menu", "Koalitionstracker")
menu = self.toolbar.get_or_create_menu("govplan-menu", _("GovPlan"))
govplan = self.request.govplan
url = reverse( url = reverse(
"admin:froide_govplan_governmentplan_change", "admin:app_list",
args=(govplan.pk,), kwargs={"app_label": "froide_govplan"},
current_app="govplanadmin", current_app="govplanadmin",
) )
menu.add_modal_item(_("Edit government plan"), url=url) menu.add_modal_item(_("Edit plans and updates"), url=url)
url = reverse(
"admin:froide_govplan_governmentplanupdate_add", if hasattr(self.request, "govplan"):
current_app="govplanadmin", govplan = self.request.govplan
) url = reverse(
menu.add_modal_item(_("Add update"), url=url) "admin:froide_govplan_governmentplan_change",
args=(govplan.pk,),
current_app="govplanadmin",
)
menu.add_modal_item(_("Edit government plan"), url=url)
url = reverse(
"admin:froide_govplan_governmentplanupdate_add",
current_app="govplanadmin",
)
menu.add_modal_item(_("Add update"), url=url)
toolbar_pool.register(GovPlanToolbar) toolbar_pool.register(GovPlanToolbar)