utils.py

   1 # -*- coding: utf-8 -*-
   2 #######################################################################################
   3 #   Plinn - http://plinn.org                                                          #
   4 #   Copyright (C) 2005-2007  Benoît PIN <benoit.pin@ensmp.fr>                         #
   5 #                                                                                     #
   6 #   This program is free software; you can redistribute it and/or                     #
   7 #   modify it under the terms of the GNU General Public License                       #
   8 #   as published by the Free Software Foundation; either version 2                    #
   9 #   of the License, or (at your option) any later version.                            #
  10 #                                                                                     #
  11 #   This program is distributed in the hope that it will be useful,                   #
  12 #   but WITHOUT ANY WARRANTY; without even the implied warranty of                    #
  13 #   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the                     #
  14 #   GNU General Public License for more details.                                      #
  15 #                                                                                     #
  16 #   You should have received a copy of the GNU General Public License                 #
  17 #   along with this program; if not, write to the Free Software                       #
  18 #   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.   #
  19 #######################################################################################
  20 """ Plinn public utilities
  21
  22
  23 """
  24
  25 import string
  26 import re
  27 from types import StringType
  28 from random import randrange
  29 from Acquisition import aq_base
  30 from quopri import encodestring
  31 from zope.globalrequest import getRequest
  32 from AccessControl.PermissionRole import rolesForPermissionOn
  33 from AccessControl import ModuleSecurityInfo
  34 from AccessControl import getSecurityManager
  35 from AccessControl.User import UnrestrictedUser
  36 from OFS.CopySupport import _cb_decode, _cb_encode, cookie_path
  37 from Products.CMFCore.utils import getToolByName, getUtilityByInterfaceName
  38 from Products.CMFCore.exceptions import BadRequest
  39 from Products.Utf8Splitter.Utf8Splitter import Utf8Utils
  40 from Globals import REPLACEABLE, NOT_REPLACEABLE, UNIQUE
  41 from zope.i18n import translate as i18ntranslate
  42 from zope.i18n.interfaces import IUserPreferredLanguages
  43 from zope.i18nmessageid import MessageFactory
  44 from zope.component.interfaces import ComponentLookupError
  45 from zope.dottedname.resolve import resolve as resolve_dotted_name
  46 from zope.component import queryAdapter
  47
  48 _marker = []
  49
  50 security = ModuleSecurityInfo( 'Products.Plinn.utils' )
  51
  52 security.declarePublic('thisObjectComeFromPortalSkin')
  53 def thisObjectComeFromPortalSkin(ob, portal=None):
  54     """ check if ob comes from portal_skins """
  55     if not portal :
  56         portal = getToolByName(ob, 'portal_url')
  57         portal = portal.getPortalObject()
  58
  59     if ob.aq_self == portal.aq_self :
  60         return False
  61
  62     obId = ob.id
  63     if callable(obId) :
  64         obId = obId()
  65
  66     sob = getattr(portal, obId, None)
  67
  68     if sob is None :
  69         return False
  70     elif not(sob.aq_inner.aq_self is ob.aq_inner.aq_self) :
  71         return False
  72     else :
  73         try :
  74             portal._checkId(obId)
  75             return True
  76         except BadRequest :
  77             return False
  78
  79 security.declarePublic('listActionProviders_')
  80 def listActionProviders_(context) :
  81     atool = getToolByName(context, 'portal_actions')
  82     return atool.listActionProviders()
  83
  84 def capitalizeCompoundGivenName(givenName) :
  85     givenName = givenName.strip()
  86     givenNames = ' '.join(givenName.split('-')).split()
  87     givenNameCapitalized = '-'.join(map(string.capitalize, givenNames))
  88     return givenNameCapitalized
  89
  90
  91 def formatFullName(memberName, memberGivenName, memberId, nameBefore=1) :
  92     memberName = memberName.decode('utf-8')
  93     memberGivenName = memberGivenName.decode('utf-8')
  94     memberFullName = u''
  95     if memberName and memberGivenName :
  96         if nameBefore :
  97             memberFullName = memberName.upper() + ' ' + capitalizeCompoundGivenName(memberGivenName)
  98         else :
  99             memberFullName = capitalizeCompoundGivenName(memberGivenName) + ' ' + memberName.upper()
 100
 101     elif memberName and not memberGivenName :
 102         memberFullName = memberName.upper()
 103
 104     elif not memberName and memberGivenName :
 105         memberFullName = capitalizeCompoundGivenName(memberGivenName)
 106
 107     else :
 108         memberFullName = memberId
 109
 110     return memberFullName.encode('utf-8')
 111
 112 # from OFS.ObjectManager #63
 113 bad_url_chars = re.compile(r'[^a-zA-Z0-9-_~,.$\(\)@]')
 114
 115 security.declarePublic('makeValidId')
 116 def makeValidId(self, id, allow_dup=0):
 117     id = Utf8Utils.desacc(id)
 118     id = bad_url_chars.sub('-', id)
 119     # If allow_dup is false, an error will be raised if an object
 120     # with the given id already exists. If allow_dup is true,
 121     # only check that the id string contains no illegal chars;
 122     # check_valid_id() will be called again later with allow_dup
 123     # set to false before the object is added.
 124
 125     makeRandomId = False
 126     if id in ('.', '..'):
 127         makeRandomId = True
 128     if id.startswith('_'):
 129         id = id.lstrip('_')
 130     if id.startswith('aq_'):
 131         id = id[3:]
 132
 133     while id.endswith('__') :
 134         id = id[:-1]
 135     if not allow_dup:
 136         obj = getattr(self, id, None)
 137         if obj is not None:
 138             # An object by the given id exists either in this
 139             # ObjectManager or in the acquisition path.
 140             flags = getattr(obj, '__replaceable__', NOT_REPLACEABLE)
 141             if hasattr(aq_base(self), id):
 142                 # The object is located in this ObjectManager.
 143                 if not flags & REPLACEABLE:
 144                     makeRandomId = True
 145                 # else the object is replaceable even if the UNIQUE
 146                 # flag is set.
 147             elif flags & UNIQUE:
 148                 makeRandomId = True
 149     if id == 'REQUEST':
 150         makeRandomId = True
 151
 152     if makeRandomId is True :
 153         id = str(randrange(2,10000)) + id
 154     return id
 155
 156
 157
 158 def _checkMemberPermission(userid, permission, obj, StringType = type('')):
 159     user = obj.aq_inner.acl_users.getUser(userid)
 160     roles = rolesForPermissionOn(permission, obj)
 161     if type(roles) is StringType:
 162         roles=[roles]
 163     if user.allowed( obj, roles ):
 164         return 1
 165     return 0
 166
 167 def getCPInfo(self) :
 168     try: cp = _cb_decode(self.REQUEST['__cp'])
 169     except: return None
 170     return cp
 171
 172
 173 def popCP(self, indexes=None) :
 174     try: cp = _cb_decode(self.REQUEST['__cp'])
 175     except: return
 176
 177     paths = list(cp[1])
 178     if indexes is not None :
 179         indexes = list(indexes)
 180         indexes.sort()
 181         indexes.reverse()
 182         for index in indexes :
 183             paths.pop(index)
 184     else :
 185         paths.pop()
 186
 187     if not paths :
 188         self.REQUEST.RESPONSE.expireCookie('__cp', path=self.REQUEST['BASEPATH1'] or "/")
 189     else :
 190         paths = tuple(paths)
 191         cp = _cb_encode( (cp[0], paths) )
 192         resp = self.REQUEST['RESPONSE']
 193         resp.setCookie('__cp', cp, path='%s' % cookie_path(self.REQUEST))
 194
 195 security.declarePublic('Message')
 196 Message = MessageFactory('plinn')
 197
 198 security.declarePublic('translate')
 199 def translate(message, context=None):
 200     """ Translate i18n message.
 201     """
 202     if isinstance(message, Exception):
 203         try:
 204             message = message[0]
 205         except (TypeError, IndexError):
 206             pass
 207     if not context :
 208         request = getRequest()
 209     else :
 210         request = context.REQUEST
 211     return i18ntranslate(message, domain='plinn', context=request)
 212
 213 security.declarePublic('desacc')
 214 desacc = Utf8Utils.desacc
 215
 216 security.declarePublic('getPreferredLanguages')
 217 def getPreferredLanguages(context):
 218     """ returns browser prefered languages"""
 219     request = getattr(context, 'REQUEST', None)
 220     if request is not None :
 221         adapter = IUserPreferredLanguages(request, None)
 222         if adapter is not None :
 223             return adapter.getPreferredLanguages()
 224     return []
 225
 226 security.declarePublic('getBestTranslationLanguage')
 227 def getBestTranslationLanguage(langs, context):
 228     """ returns best translation language according
 229         availables languages (param langs)
 230         and user preferences (retrieves by context)
 231     """
 232     request = getattr(context, 'REQUEST', None)
 233     if request :
 234         negociator = getUtilityByInterfaceName('zope.i18n.interfaces.INegotiator')
 235         return negociator.getLanguage(langs, request) or langs[0]
 236     else :
 237         return langs[0]
 238
 239 security.declarePublic('getAdapterByInterface')
 240 def getAdapterByInterface(ob, dotted_name, default=_marker) :
 241     """ Get the adapter which provides the interface on the given object.
 242     """
 243     try:
 244         iface = resolve_dotted_name(dotted_name)
 245     except ImportError:
 246         if default is _marker:
 247             raise ComponentLookupError, dotted_name
 248         return default
 249
 250     adapter = queryAdapter(ob, iface, default=default)
 251     if adapter is _marker :
 252         raise ComponentLookupError, "no adapter providing %r found on %r" % (dotted_name, ob)
 253
 254     # the adapter must be wrapped to allow security mahinery to work.
 255     if adapter != default :
 256         return adapter.__of__(ob)
 257     else :
 258         return default
 259
 260 def encodeQuopriEmail(name, email) :
 261     qpName = encodestring(name).replace('=\n', '')
 262     return '''"=?utf-8?q?%s?=" <%s>''' % (qpName, email)
 263
 264 def encodeMailHeader(content) :
 265     s = encodestring(content).replace('=\n', '')
 266     s = s.replace('_', '=5F')
 267     s = s.replace(' ', '_')
 268
 269     lines = []
 270     STEP = 50
 271     start = 0
 272     stop = STEP
 273     part = s[start:stop]
 274     lines.append(part)
 275
 276     while len(part) == STEP:
 277         start = start + STEP
 278         stop = stop + STEP
 279         part = s[start:stop]
 280         lines.append(part)
 281
 282     lines = [' =?utf-8?Q?%s?=' % part for part in lines]
 283     s = '\n'.join(lines)
 284     s = s.strip()
 285     return s
 286
 287
 288 def _sudo(func, userid=None) :
 289     """
 290     execute func or any callable object
 291     without restriction. Used to switch off
 292     security assertions (eg. checkPermission) encountered
 293     during the execution.
 294     """
 295
 296     sm = getSecurityManager()
 297     restrictedUser = sm.getUser()
 298
 299     if not userid :
 300         userid = restrictedUser.getId()
 301
 302     sm._context.user = UnrestrictedUser(userid, '', (), ())
 303
 304     deferedEx = None
 305     try :
 306         ret = func()
 307     except Exception, e :
 308         deferedEx = e
 309
 310     sm._context.user = restrictedUser
 311
 312     if deferedEx is not None :
 313         raise e
 314
 315     return ret
 316