From 6d74c26eedc16640897bcc17612f82132f604156 Mon Sep 17 00:00:00 2001 From: Iacopo Spalletti Date: Fri, 29 Apr 2016 10:12:41 +0200 Subject: [PATCH] Add some compatibility for cms menu and filer --- djangocms_blog/cms_menus.py | 145 ++++++++++++++++++++++++++++++++++++ djangocms_blog/menu.py | 145 +----------------------------------- tests/base.py | 8 +- 3 files changed, 153 insertions(+), 145 deletions(-) create mode 100644 djangocms_blog/cms_menus.py diff --git a/djangocms_blog/cms_menus.py b/djangocms_blog/cms_menus.py new file mode 100644 index 0000000..6be7aad --- /dev/null +++ b/djangocms_blog/cms_menus.py @@ -0,0 +1,145 @@ +# -*- coding: utf-8 -*- +from __future__ import absolute_import, print_function, unicode_literals + +from cms.apphook_pool import apphook_pool +from cms.menu_bases import CMSAttachMenu +from django.core.urlresolvers import resolve +from django.db.models.signals import post_delete, post_save +from django.utils.translation import get_language_from_request, ugettext_lazy as _ +from menus.base import Modifier, NavigationNode +from menus.menu_pool import menu_pool + +from .cms_appconfig import BlogConfig +from .models import BlogCategory, Post +from .settings import MENU_TYPE_CATEGORIES, MENU_TYPE_COMPLETE, MENU_TYPE_POSTS, get_setting + + +class BlogCategoryMenu(CMSAttachMenu): + """ + Main menu class + + Handles all types of blog menu + """ + name = _('Blog menu') + + def get_nodes(self, request): + """ + Generates the nodelist + + :param request: + :return: list of nodes + """ + nodes = [] + + language = get_language_from_request(request, check_path=True) + + categories_menu = False + posts_menu = False + config = False + if hasattr(self, 'instance') and self.instance: + config = BlogConfig.objects.get(namespace=self.instance.application_namespace) + if config and config.menu_structure in (MENU_TYPE_COMPLETE, MENU_TYPE_CATEGORIES): + categories_menu = True + if config and config.menu_structure in (MENU_TYPE_COMPLETE, MENU_TYPE_POSTS): + posts_menu = True + + if categories_menu: + categories = BlogCategory.objects + if config: + categories = categories.namespace(self.instance.application_namespace) + categories = categories.active_translations(language).distinct() + categories = categories.order_by('parent__id', 'translations__name') + for category in categories: + node = NavigationNode( + category.name, + category.get_absolute_url(), + '{0}-{1}'.format(category.__class__.__name__, category.pk), + ( + '{0}-{1}'.format( + category.__class__.__name__, category.parent.id + ) if category.parent else None + ) + ) + nodes.append(node) + + if posts_menu: + posts = Post.objects + if hasattr(self, 'instance') and self.instance: + posts = posts.namespace(self.instance.application_namespace) + posts = posts.active_translations(language).distinct() + for post in posts: + post_id = None + parent = None + if categories_menu: + category = post.categories.first() + if category: + parent = '{0}-{1}'.format(category.__class__.__name__, category.pk) + post_id = '{0}-{1}'.format(post.__class__.__name__, post.pk), + else: + post_id = '{0}-{1}'.format(post.__class__.__name__, post.pk), + if post_id: + node = NavigationNode( + post.get_title(), + post.get_absolute_url(language), + post_id, + parent + ) + nodes.append(node) + + return nodes + +menu_pool.register_menu(BlogCategoryMenu) + + +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): + """ + Actual modifier function + :param request: request + :param nodes: complete list of nodes + :param namespace: Menu namespace + :param root_id: eventual root_id + :param post_cut: flag for modifier stage + :param breadcrumb: flag for modifier stage + :return: nodeslist + """ + app = None + config = None + if getattr(request, 'current_page', None) and request.current_page.application_urls: + app = apphook_pool.get_apphook(request.current_page.application_urls) + + if app and app.app_config: + namespace = resolve(request.path).namespace + config = app.get_config(namespace) + if config and config.menu_structure != MENU_TYPE_CATEGORIES: + return nodes + if post_cut: + return nodes + current_post = getattr(request, get_setting('CURRENT_POST_IDENTIFIER'), None) + category = None + if current_post and current_post.__class__ == Post: + category = current_post.categories.first() + if not category: + return nodes + + for node in nodes: + if '{0}-{1}'.format(category.__class__.__name__, category.pk) == node.id: + node.selected = True + return nodes + +menu_pool.register_modifier(BlogNavModifier) + + +def clear_menu_cache(**kwargs): + """ + Empty menu cache when saving categories + """ + menu_pool.clear(all=True) + +post_save.connect(clear_menu_cache, sender=BlogCategory) +post_delete.connect(clear_menu_cache, sender=BlogCategory) diff --git a/djangocms_blog/menu.py b/djangocms_blog/menu.py index 6be7aad..00c489f 100644 --- a/djangocms_blog/menu.py +++ b/djangocms_blog/menu.py @@ -1,145 +1,2 @@ # -*- coding: utf-8 -*- -from __future__ import absolute_import, print_function, unicode_literals - -from cms.apphook_pool import apphook_pool -from cms.menu_bases import CMSAttachMenu -from django.core.urlresolvers import resolve -from django.db.models.signals import post_delete, post_save -from django.utils.translation import get_language_from_request, ugettext_lazy as _ -from menus.base import Modifier, NavigationNode -from menus.menu_pool import menu_pool - -from .cms_appconfig import BlogConfig -from .models import BlogCategory, Post -from .settings import MENU_TYPE_CATEGORIES, MENU_TYPE_COMPLETE, MENU_TYPE_POSTS, get_setting - - -class BlogCategoryMenu(CMSAttachMenu): - """ - Main menu class - - Handles all types of blog menu - """ - name = _('Blog menu') - - def get_nodes(self, request): - """ - Generates the nodelist - - :param request: - :return: list of nodes - """ - nodes = [] - - language = get_language_from_request(request, check_path=True) - - categories_menu = False - posts_menu = False - config = False - if hasattr(self, 'instance') and self.instance: - config = BlogConfig.objects.get(namespace=self.instance.application_namespace) - if config and config.menu_structure in (MENU_TYPE_COMPLETE, MENU_TYPE_CATEGORIES): - categories_menu = True - if config and config.menu_structure in (MENU_TYPE_COMPLETE, MENU_TYPE_POSTS): - posts_menu = True - - if categories_menu: - categories = BlogCategory.objects - if config: - categories = categories.namespace(self.instance.application_namespace) - categories = categories.active_translations(language).distinct() - categories = categories.order_by('parent__id', 'translations__name') - for category in categories: - node = NavigationNode( - category.name, - category.get_absolute_url(), - '{0}-{1}'.format(category.__class__.__name__, category.pk), - ( - '{0}-{1}'.format( - category.__class__.__name__, category.parent.id - ) if category.parent else None - ) - ) - nodes.append(node) - - if posts_menu: - posts = Post.objects - if hasattr(self, 'instance') and self.instance: - posts = posts.namespace(self.instance.application_namespace) - posts = posts.active_translations(language).distinct() - for post in posts: - post_id = None - parent = None - if categories_menu: - category = post.categories.first() - if category: - parent = '{0}-{1}'.format(category.__class__.__name__, category.pk) - post_id = '{0}-{1}'.format(post.__class__.__name__, post.pk), - else: - post_id = '{0}-{1}'.format(post.__class__.__name__, post.pk), - if post_id: - node = NavigationNode( - post.get_title(), - post.get_absolute_url(language), - post_id, - parent - ) - nodes.append(node) - - return nodes - -menu_pool.register_menu(BlogCategoryMenu) - - -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): - """ - Actual modifier function - :param request: request - :param nodes: complete list of nodes - :param namespace: Menu namespace - :param root_id: eventual root_id - :param post_cut: flag for modifier stage - :param breadcrumb: flag for modifier stage - :return: nodeslist - """ - app = None - config = None - if getattr(request, 'current_page', None) and request.current_page.application_urls: - app = apphook_pool.get_apphook(request.current_page.application_urls) - - if app and app.app_config: - namespace = resolve(request.path).namespace - config = app.get_config(namespace) - if config and config.menu_structure != MENU_TYPE_CATEGORIES: - return nodes - if post_cut: - return nodes - current_post = getattr(request, get_setting('CURRENT_POST_IDENTIFIER'), None) - category = None - if current_post and current_post.__class__ == Post: - category = current_post.categories.first() - if not category: - return nodes - - for node in nodes: - if '{0}-{1}'.format(category.__class__.__name__, category.pk) == node.id: - node.selected = True - return nodes - -menu_pool.register_modifier(BlogNavModifier) - - -def clear_menu_cache(**kwargs): - """ - Empty menu cache when saving categories - """ - menu_pool.clear(all=True) - -post_save.connect(clear_menu_cache, sender=BlogCategory) -post_delete.connect(clear_menu_cache, sender=BlogCategory) +from djangocms_blog.cms_menus import * # NOQA diff --git a/tests/base.py b/tests/base.py index 74d3a11..b12e368 100644 --- a/tests/base.py +++ b/tests/base.py @@ -3,7 +3,6 @@ from __future__ import absolute_import, print_function, unicode_literals from copy import deepcopy -from cmsplugin_filer_image.models import ThumbnailOption from django.contrib.auth import get_user_model from django.contrib.sites.models import Site from django.core.cache import cache @@ -15,6 +14,13 @@ from parler.utils.context import smart_override from djangocms_blog.cms_appconfig import BlogConfig from djangocms_blog.models import BlogCategory, Post + +try: + from filer.models import ThumbnailOption # NOQA +except ImportError: + from cmsplugin_filer_image.models import ThumbnailOption # NOQA + + User = get_user_model()