OpenSlides/openslides/utils/main_menu.py

from django.core.urlresolvers import reverse
from django.dispatch import Signal, receiver

from .dispatch import SignalConnectMetaClass
from .signals import template_manipulation


class MainMenuEntry(object, metaclass=SignalConnectMetaClass):
    """
    Base class for a main menu entry.

    Every app which wants to add entries has to create a class subclassing
    from this base class. For the appearance the verbose_name, the
    pattern_name and the icon-css-class attribute have to be set. The
    metaclass (SignalConnectMetaClass) does the rest of the
    magic.

    For the appearance there are some optional attributes and methods like
    required_permission, default_weight, stylesheets, javascript_files,
    check_permission, get_url, get_default_weight, get_icon_css_class,
    get_stylesheets and get_javascript_files.
    """
    signal = Signal(providing_args=['request'])
    verbose_name = None
    required_permission = None
    default_weight = 0
    pattern_name = None
    icon_css_class = 'icon-home'
    stylesheets = None
    javascript_files = None

    def __init__(self, sender, request, **kwargs):
        """
        Initializes the main menu entry instance. This is done when the signal
        is sent.

        Only the required request argument is used. Because of Django's signal
        API, we have to take also a sender argument and wildcard keyword
        arguments. But they are not used here.
        """
        self.request = request

    def __str__(self):
        if self.verbose_name is None:
            raise NotImplementedError(
                'The main menu entry class %s must provide a verbose_name '
                'attribute or override the __str__ method.' % type(self).__name__)
        return str(self.verbose_name)

    @classmethod
    def get_dispatch_uid(cls):
        """
        Returns the classname as a unique string for each class. Returns None
        for the base class so it will not be connected to the signal.
        """
        if not cls.__name__ == 'MainMenuEntry':
            return cls.__name__

    def check_permission(self):
        """
        Returns True if the request user is allowed to see the entry.
        """
        return self.required_permission is None or self.request.user.has_perm(self.required_permission)

    def get_icon_css_class(self):
        """
        Returns the css class name of the icon. Default is 'icon-home'.
        """
        return self.icon_css_class

    def get_url(self):
        """
        Returns the url of the entry.
        """
        if self.pattern_name is None:
            raise NotImplementedError(
                'The main menu entry class %s must provide a pattern_name '
                'attribute or override the get_url method.' % type(self).__name__)
        return reverse(self.pattern_name)

    def is_active(self):
        """
        Returns True if the entry is selected at the moment.
        """
        try:
            return_value = isinstance(self, self.request.active_main_menu_class)
        except AttributeError:
            return_value = self.request.path.startswith(self.get_url())
        return return_value

    def get_stylesheets(self):
        """
        Returns an interable of stylesheets to be loaded.
        """
        return iter(self.stylesheets or [])

    def get_javascript_files(self):
        """
        Returns an interable of javascript files to be loaded.
        """
        return iter(self.javascript_files or [])


def main_menu_entries(request):
    """
    Adds all main menu entries to the request context as template context
    processor.
    """
    return {'main_menu_entries': MainMenuEntry.get_all(request)}


@receiver(template_manipulation, dispatch_uid="add_main_menu_context")
def add_main_menu_context(sender, request, context, **kwargs):
    """
    Adds all stylefiles from all main menu entries to the context.
    """
    for main_menu_entry in MainMenuEntry.get_all(request):
        context['extra_stylefiles'].extend(main_menu_entry.get_stylesheets())
        context['extra_javascript'].extend(main_menu_entry.get_javascript_files())