source: products/quintagroup.plonegooglesitemaps/branches/1.7.1/quintagroup/plonegooglesitemaps/browser/utils.py @ 3399

import re

from Missing import MV as Missing_Value
from Products.CMFCore.utils import getToolByName

from quintagroup.canonicalpath.interfaces import ICanonicalLink

ADD_ZOPE = re.compile('^/')
ADD_PLONE = re.compile('^[^http://|https://|\\\]')
OPERATIONS_PARSE = re.compile(r"(.?[^\\])/(.*[^\\]|)/(.*[^\\]|)/")
_marker = []


def searchAndReplace(string, what, withs):
    """Emulate sed command s/"""
    res = re.sub(what, withs, string)
    return res
OPERATORS = {'s': searchAndReplace, }


def applyOperations(objects, operations):
    """Parse Operations """
    operations = [OPERATIONS_PARSE.match(op).groups() for op in operations]
    result = {}
    for ob in objects:
        url = _marker
        if hasattr(ob, 'canonical_link'):
            url = ob.canonical_link
        if url in [Missing_Value, _marker]:
            url = ICanonicalLink(ob.getObject()).canonical_link
        for operator, what, withs in operations:
            url = OPERATORS[operator](url, what, withs.replace("\\", ""))
        #TODO: Remove or replace following condition
        #it is senseless in the case we need intelligent
        #result set. Better condition would be to place
        #freshest brain into result
        if url in result.keys():
            continue
        #TODO: replace brain with only data necessary to
        #generate sitemap
        result[url] = ob
    return result


def additionalURLs(context):
    """Add URLs to sitemap that arn't objects"""
    res = []

    plone_home = getToolByName(context, 'portal_url')()
    root = context.getPhysicalRoot().absolute_url()
    URLs = context.getUrls()

    for url in URLs:
        if ADD_ZOPE.match(url):
            res.append(root + url)
        elif ADD_PLONE.match(url):
            res.append(plone_home + '/' + url)
        else:
            res.append(url)
    return res