2015-02-14 14:23:35 +00:00
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
from cms.menu_bases import CMSAttachMenu
|
2015-02-25 22:37:58 +00:00
|
|
|
from menus.base import Modifier, NavigationNode
|
2015-02-14 14:23:35 +00:00
|
|
|
from menus.menu_pool import menu_pool
|
2015-02-15 14:45:48 +00:00
|
|
|
from django.db.models.signals import post_save, post_delete
|
2015-02-14 14:23:35 +00:00
|
|
|
from django.utils.translation import ugettext_lazy as _, get_language
|
|
|
|
from .models import BlogCategory
|
|
|
|
|
|
|
|
|
|
|
|
class BlogCategoryMenu(CMSAttachMenu):
|
|
|
|
name = _('Blog Category menu')
|
|
|
|
|
|
|
|
def get_nodes(self, request):
|
|
|
|
nodes = []
|
|
|
|
qs = BlogCategory.objects.translated(get_language())
|
2015-02-14 14:23:35 +00:00
|
|
|
qs = qs.order_by('parent_id', 'translations__name')
|
2015-02-14 14:23:35 +00:00
|
|
|
for category in qs:
|
|
|
|
kwargs = { 'category': category.slug }
|
|
|
|
node = NavigationNode(
|
|
|
|
category.name,
|
|
|
|
category.get_absolute_url(),
|
|
|
|
category.pk,
|
|
|
|
category.parent_id
|
|
|
|
)
|
|
|
|
nodes.append(node)
|
|
|
|
return nodes
|
|
|
|
|
|
|
|
menu_pool.register_menu(BlogCategoryMenu)
|
|
|
|
|
2015-02-15 14:45:48 +00:00
|
|
|
|
2015-02-25 22:37:58 +00:00
|
|
|
class BlogNavModifier(Modifier):
|
|
|
|
"""
|
|
|
|
This navigation modifier makes sure that when
|
|
|
|
a particular blog post is viewed,
|
|
|
|
a corresponding category is selected in menu
|
|
|
|
"""
|
|
|
|
def modify(self, request, nodes, namespace, root_id, post_cut, breadcrumb):
|
|
|
|
if not post_cut: return nodes
|
|
|
|
if not hasattr(request, 'toolbar'):
|
|
|
|
return nodes
|
|
|
|
if request.toolbar.get_object_model() != 'djangocms_blog.post':
|
|
|
|
return nodes
|
|
|
|
cat = request.toolbar.obj.categories.first()
|
|
|
|
if not cat: return nodes
|
|
|
|
|
|
|
|
for node in nodes:
|
|
|
|
if (node.namespace == BlogCategoryMenu.__name__ and
|
|
|
|
cat.pk == node.id):
|
|
|
|
node.selected = True
|
|
|
|
break
|
|
|
|
return nodes
|
|
|
|
|
|
|
|
menu_pool.register_modifier(BlogNavModifier)
|
|
|
|
|
2015-02-15 14:45:48 +00:00
|
|
|
def clear_menu_cache(**kwargs):
|
|
|
|
menu_pool.clear(all=True)
|
|
|
|
|
|
|
|
post_save.connect(clear_menu_cache, sender=BlogCategory)
|
|
|
|
post_delete.connect(clear_menu_cache, sender=BlogCategory)
|