mirror of
https://github.com/Hopiu/wagtail.git
synced 2026-05-18 20:21:11 +00:00
91 lines
3.3 KiB
Python
91 lines
3.3 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from six import text_type, with_metaclass
|
|
|
|
try:
|
|
# renamed util -> utils in Django 1.7; try the new name first
|
|
from django.forms.utils import flatatt
|
|
except ImportError:
|
|
from django.forms.util import flatatt
|
|
|
|
from django.forms import MediaDefiningClass, Media
|
|
from django.utils.text import slugify
|
|
from django.utils.html import format_html
|
|
from django.utils.safestring import mark_safe
|
|
|
|
from wagtail.wagtailcore import hooks
|
|
|
|
|
|
class MenuItem(with_metaclass(MediaDefiningClass)):
|
|
def __init__(self, label, url, name=None, classnames='', attrs=None, order=1000):
|
|
self.label = label
|
|
self.url = url
|
|
self.classnames = classnames
|
|
self.name = (name or slugify(text_type(label)))
|
|
self.order = order
|
|
|
|
if attrs:
|
|
self.attr_string = flatatt(attrs)
|
|
else:
|
|
self.attr_string = ""
|
|
|
|
def is_shown(self, request):
|
|
"""
|
|
Whether this menu item should be shown for the given request; permission
|
|
checks etc should go here. By default, menu items are shown all the time
|
|
"""
|
|
return True
|
|
|
|
def render_html(self, request):
|
|
return format_html(
|
|
"""<li class="menu-{0}"><a href="{1}" class="{2}"{3}>{4}</a></li>""",
|
|
self.name, self.url, self.classnames, self.attr_string, self.label)
|
|
|
|
|
|
class Menu(object):
|
|
def __init__(self, register_hook_name, construct_hook_name=None):
|
|
self.register_hook_name = register_hook_name
|
|
self.construct_hook_name = construct_hook_name
|
|
# _registered_menu_items will be populated on first access to the
|
|
# registered_menu_items property. We can't populate it in __init__ because
|
|
# we can't rely on all hooks modules to have been imported at the point that
|
|
# we create the admin_menu and settings_menu instances
|
|
self._registered_menu_items = None
|
|
|
|
@property
|
|
def registered_menu_items(self):
|
|
if self._registered_menu_items is None:
|
|
self._registered_menu_items = [fn() for fn in hooks.get_hooks(self.register_hook_name)]
|
|
return self._registered_menu_items
|
|
|
|
def menu_items_for_request(self, request):
|
|
return [item for item in self.registered_menu_items if item.is_shown(request)]
|
|
|
|
@property
|
|
def media(self):
|
|
media = Media()
|
|
for item in self.registered_menu_items:
|
|
media += item.media
|
|
return media
|
|
|
|
def render_html(self, request):
|
|
menu_items = self.menu_items_for_request(request)
|
|
|
|
# provide a hook for modifying the menu, if construct_hook_name has been set
|
|
if self.construct_hook_name:
|
|
for fn in hooks.get_hooks(self.construct_hook_name):
|
|
fn(request, menu_items)
|
|
|
|
rendered_menu_items = []
|
|
for item in sorted(menu_items, key=lambda i: i.order):
|
|
try:
|
|
rendered_menu_items.append(item.render_html(request))
|
|
except TypeError:
|
|
# fallback for older render_html methods that don't accept a request arg
|
|
rendered_menu_items.append(item.render_html())
|
|
|
|
return mark_safe(''.join(rendered_menu_items))
|
|
|
|
|
|
admin_menu = Menu(register_hook_name='register_admin_menu_item', construct_hook_name='construct_main_menu')
|
|
settings_menu = Menu(register_hook_name='register_settings_menu_item')
|