Show devenez membre contact form submenu

This commit is contained in:
PCoder 2021-09-08 20:24:00 +05:30
parent fa76b10f73
commit df0a219f62

View file

@ -2,8 +2,8 @@
from django import template from django import template
from django.utils import translation from django.utils import translation
from wagtail.core.models import Site from wagtail.core.models import Site, Page
from ..models.forms import ContactForm
register = template.Library() register = template.Library()
@register.simple_tag() @register.simple_tag()
@ -74,19 +74,22 @@ def menuitems_children(parent):
if 'Über uns' in parent.title: if 'Über uns' in parent.title:
remove_devenez = True remove_devenez = True
menuitems_children = parent.get_children().live().in_menu().specific() menuitems_children = parent.get_children().live().in_menu().specific()
out_menuitems_children = [] items_to_remove = []
for menuitem in menuitems_children: for menuitem in menuitems_children:
try: try:
if type(menuitem) == ContactForm:
menuitem.title = menuitem.title.title()
else:
menuitem.title = menuitem.trans_title menuitem.title = menuitem.trans_title
if 'devenez' in menuitem.title.lower() and remove_devenez: if 'devenez' in menuitem.title.lower() and remove_devenez:
continue items_to_remove.append(menuitem)
elif 'mitglied werden' in menuitem.title.lower() and remove_mitglied: elif 'mitglied werden' in menuitem.title.lower() and remove_mitglied:
continue items_to_remove.append(menuitem)
else:
out_menuitems_children.append(menuitem)
except AttributeError: except AttributeError:
pass pass
return out_menuitems_children for item in items_to_remove:
menuitems_children = menuitems_children & Page.objects.not_page(item)
return menuitems_children
# Retrieves the children of the top menu items for the drop downs # Retrieves the children of the top menu items for the drop downs
@register.inclusion_tag('tags/top_menu_children.html', takes_context=True) @register.inclusion_tag('tags/top_menu_children.html', takes_context=True)