source: products/qPloneDropDownMenu/tags/0.2.1/utils.py

# -*- coding: utf-8 -*-
""" Utility functions """

from zope.component import getMultiAdapter

from OFS.DTMLMethod import addDTMLMethod

from Products.CMFCore.utils import getToolByName
from Products.CMFPlone.migrations.migration_util import safeEditProperty
from Products.CMFCore.Expression import Expression, createExprContext

from config import PROPERTY_FIELD, PROPERTY_SHEET


TAB_HTML_SNIPPET = """<li id="portaltab-%(id)s" class="plain">
  <a href="%(url)s" accesskey="t" title="%(desc)s">%(name)s</a>
</li>
"""


def addCSS(container, sheetId, title, csshovering):
    """ Add DTML Method object to portal root """
    addDTMLMethod(container, sheetId, title, csshovering)

def updateMenu(site):
    pu = getToolByName(site, 'plone_utils')
    pa = getToolByName(site, 'portal_actions')
    portal_props = getToolByName(site, 'portal_properties')

    # collect all portal tabs
    context_state = getMultiAdapter((site, site.REQUEST),
                                    name=u'plone_context_state')
    actions = context_state.actions()
    portal_tabs_view = getMultiAdapter((site, site.REQUEST),
                                       name='portal_tabs_view')
    portal_tabs = portal_tabs_view.topLevelTabs(actions=actions)

    # dump to html
    value = u""
    enc = portal_props.site_properties.getProperty('default_charset', 'utf-8')
    for tab in portal_tabs:
        value += TAB_HTML_SNIPPET % {'id': toUnicode(tab['id'], enc),
                                     'url': toUnicode(tab['url'], enc),
                                     'desc': toUnicode(tab['description'], enc),
                                     'name': toUnicode(tab['name'], enc)}

    if not hasattr(portal_props.aq_base, PROPERTY_SHEET):
        portal_props.addPropertySheet(PROPERTY_SHEET,
                                      'DropDown Menu Properties')
    ap = getattr(portal_props.aq_base, PROPERTY_SHEET)
    safeEditProperty(ap, PROPERTY_FIELD, value, 'text')

def toUnicode(value, enc='utf-8'):
    if isinstance(value, str):
        return value.decode(enc)
    else:
        return value