src/pyams_content/zmi/viewlet/toplinks/__init__.py
changeset 339 50c73e42883a
parent 338 f37b5995a48c
child 340 de5a2a9fdf22
equal deleted inserted replaced
338:f37b5995a48c 339:50c73e42883a
     1 #
       
     2 # Copyright (c) 2008-2015 Thierry Florac <tflorac AT ulthar.net>
       
     3 # All Rights Reserved.
       
     4 #
       
     5 # This software is subject to the provisions of the Zope Public License,
       
     6 # Version 2.1 (ZPL).  A copy of the ZPL should accompany this distribution.
       
     7 # THIS SOFTWARE IS PROVIDED "AS IS" AND ANY AND ALL EXPRESS OR IMPLIED
       
     8 # WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
       
     9 # WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS
       
    10 # FOR A PARTICULAR PURPOSE.
       
    11 #
       
    12 
       
    13 __docformat__ = 'restructuredtext'
       
    14 
       
    15 
       
    16 # import standard library
       
    17 
       
    18 # import interfaces
       
    19 from hypatia.interfaces import ICatalog
       
    20 from pyams_content.shared.common.interfaces import IBaseSharedTool, ISharedSite
       
    21 from pyams_content.zmi.interfaces import IUserAddingsMenuLabel
       
    22 from pyams_i18n.interfaces import II18n
       
    23 from pyams_skin.interfaces.viewlet import ITopLinksViewletManager
       
    24 from pyams_zmi.layer import IAdminLayer
       
    25 
       
    26 # import packages
       
    27 from hypatia.catalog import CatalogQuery
       
    28 from hypatia.query import Any, And, Or, NotEq
       
    29 from pyams_catalog.query import CatalogResultSet
       
    30 from pyams_skin.viewlet.toplinks import TopLinksViewlet, TopLinksMenu
       
    31 from pyams_template.template import template_config
       
    32 from pyams_utils.list import unique
       
    33 from pyams_utils.registry import get_local_registry, get_utility
       
    34 from pyams_utils.url import absolute_url
       
    35 from pyams_viewlet.viewlet import viewlet_config
       
    36 from pyramid.threadlocal import get_current_registry
       
    37 
       
    38 from pyams_content import _
       
    39 
       
    40 
       
    41 @viewlet_config(name='shared-sites.menu', layer=IAdminLayer, manager=ITopLinksViewletManager, weight=20)
       
    42 class SharedSitesMenu(TopLinksViewlet):
       
    43     """Shared sites menu"""
       
    44 
       
    45     label = ''
       
    46     css_class = 'top-menu bordered margin-top-10'
       
    47     dropdown_label = _("Shared sites")
       
    48 
       
    49     def update(self):
       
    50         super(SharedSitesMenu, self).update()
       
    51         registry = get_local_registry()
       
    52         for site in sorted(registry.getAllUtilitiesRegisteredFor(ISharedSite),
       
    53                            key=lambda x: II18n(x).query_attribute('title', request=self.request) or ''):
       
    54             menu = TopLinksMenu(self.context, self.request, self.__parent__, self)
       
    55             menu.label = II18n(site).query_attribute('title', request=self.request) or site.__name__
       
    56             menu.url = absolute_url(site, self.request, 'admin#dashboard.html')
       
    57             self.viewlets.append(menu)
       
    58 
       
    59 
       
    60 @viewlet_config(name='shared-tools.menu', layer=IAdminLayer, manager=ITopLinksViewletManager, weight=30)
       
    61 class SharedToolsMenu(TopLinksViewlet):
       
    62     """Shared tools menu"""
       
    63 
       
    64     label = ''
       
    65     css_class = 'top-menu bordered margin-top-10'
       
    66     dropdown_label = _("Shared contents")
       
    67 
       
    68     def update(self):
       
    69         super(SharedToolsMenu, self).update()
       
    70         registry = get_local_registry()
       
    71         for tool in sorted(registry.getAllUtilitiesRegisteredFor(IBaseSharedTool),
       
    72                            key=lambda x: II18n(x).query_attribute('title', request=self.request) or ''):
       
    73             if ISharedSite.providedBy(tool):
       
    74                 continue
       
    75             menu = TopLinksMenu(self.context, self.request, self.__parent__, self)
       
    76             menu.label = II18n(tool).query_attribute('title', request=self.request) or tool.__name__
       
    77             menu.url = absolute_url(tool, self.request, 'admin#dashboard.html')
       
    78             self.viewlets.append(menu)
       
    79 
       
    80 
       
    81 @viewlet_config(name='user-roles.menu', layer=IAdminLayer, manager=ITopLinksViewletManager, weight=90)
       
    82 class UserRolesMenu(TopLinksViewlet):
       
    83     """User roles menu"""
       
    84 
       
    85     label = ''
       
    86     css_class = 'top-menu bordered margin-top-10'
       
    87     dropdown_label = _("My roles")
       
    88 
       
    89     def update(self):
       
    90         super(UserRolesMenu, self).update()
       
    91         catalog = get_utility(ICatalog)
       
    92         params = And(Or(Any(catalog['role:contributor'], {self.request.principal.id}),
       
    93                         Any(catalog['role:manager'], {self.request.principal.id}),
       
    94                         Any(catalog['role:pilot'], {self.request.principal.id})),
       
    95                      NotEq(catalog['content_type'], None))
       
    96         for tool in sorted(unique(CatalogResultSet(CatalogQuery(catalog).query(params))),
       
    97                            key=lambda x: II18n(x).query_attribute('title', request=self.request) or ''):
       
    98             menu = TopLinksMenu(self.context, self.request, self.__parent__, self)
       
    99             menu.label = II18n(tool).query_attribute('title', request=self.request) or tool.__name__
       
   100             menu.url = absolute_url(tool, self.request, 'admin#dashboard.html')
       
   101             self.viewlets.append(menu)
       
   102 
       
   103 
       
   104 @viewlet_config(name='user-addings.menu', layer=IAdminLayer, manager=ITopLinksViewletManager, weight=95)
       
   105 @template_config(template='templates/user-addings.pt', layer=IAdminLayer)
       
   106 class UserAddingsMenu(TopLinksViewlet):
       
   107     """User addings menu"""
       
   108 
       
   109     label = ''
       
   110     css_class = 'top-menu margin-top-5-neg'
       
   111     dropdown_label = ''
       
   112 
       
   113     def update(self):
       
   114         super(UserAddingsMenu, self).update()
       
   115         catalog = get_utility(ICatalog)
       
   116         registry = get_current_registry()
       
   117         params = And(Any(catalog['role:contributor'], {self.request.principal.id}),
       
   118                      NotEq(catalog['content_type'], None))
       
   119         for tool in sorted(unique(CatalogResultSet(CatalogQuery(catalog).query(params))),
       
   120                            key=lambda x: II18n(x).query_attribute('title', request=self.request) or ''):
       
   121             menu = TopLinksMenu(self.context, self.request, self.__parent__, self)
       
   122             adapter = registry.queryMultiAdapter((tool, self.request), IUserAddingsMenuLabel)
       
   123             if adapter is None:
       
   124                 menu.label = self.request.localizer.translate(tool.shared_content_factory.content_class.content_name)
       
   125             else:
       
   126                 menu.label = adapter.label
       
   127             menu.url = absolute_url(tool, self.request, 'add-shared-content.html')
       
   128             menu.data = {'data-toggle': 'modal'}
       
   129             self.viewlets.append(menu)