lemoncurry/lemoncurry/templatetags/lemoncurry_tags.py

from django import template
from django.conf import settings
from django.contrib.sites.models import Site
from django.urls import reverse
from ..utils import load_package_json, origin, uri

from .. import breadcrumbs
from entries import kinds

register = template.Library()


class MenuItem:
    def __init__(self, label, icon, url):
        self.label = label
        self.icon = icon
        self.url = reverse(url)


@register.simple_tag
def get_package_json():
    return load_package_json()


@register.simple_tag
def request_origin(request):
    return origin(request)


@register.simple_tag
def request_uri(request):
    return uri(request)


@register.simple_tag
def site_name():
    return Site.objects.get_current().name


@register.inclusion_tag('lemoncurry/tags/nav.html')
def nav_left(request):
    items = (MenuItem(
        label=k.plural,
        icon=k.icon,
        url='entries:'+k.plural+'_index'
    ) for k in kinds.all)
    return {'items': items, 'request': request}


@register.inclusion_tag('lemoncurry/tags/nav.html')
def nav_right(request):
    if request.user.is_authenticated():
        items = (
            MenuItem(label='admin', icon='fa fa-gear', url='admin:index'),
            MenuItem(label='log out', icon='fa fa-sign-out', url='lemonauth:logout'),
        )
    else:
        items = (
            MenuItem(label='log in', icon='fa fa-sign-in', url='lemonauth:login'),
        )
    return {'items': items, 'request': request}


@register.inclusion_tag('lemoncurry/tags/breadcrumbs.html', takes_context=True)
def nav_crumbs(context, route):
    crumbs = breadcrumbs.find(route)
    current = crumbs.pop()

    item_list_element = [{
        '@type': 'ListItem',
        'position': i + 1,
        'item': {
            '@id': context['origin'] + reverse(crumb['route']),
            '@type': 'WebPage',
            'name': crumb['label']
        }
    } for i, crumb in enumerate(crumbs)]
    item_list_element.append({
        '@type': 'ListItem',
        'position': len(item_list_element) + 1,
        'item': {
            'id': context['uri'],
            '@type': 'WebPage',
            'name': current['label'] or context.get('title'),
        }
    })

    breadcrumb_list = {
        '@context': 'http://schema.org',
        '@type': 'BreadcrumbList',
        'itemListElement': item_list_element
    }

    return {
        'breadcrumb_list': breadcrumb_list,
        'crumbs': crumbs,
        'current': current,
        'title': context.get('title'),
    }


@register.simple_tag
def get_push_hub():
    return settings.PUSH_HUB