1 | # Author: Melnychuk Taras |
---|
2 | # Contact: fenix@quintagroup.com |
---|
3 | # Date: $Date: 2006-08-11 |
---|
4 | # Copyright: quintagroup.com |
---|
5 | |
---|
6 | import re |
---|
7 | from Products.CMFCore.utils import getToolByName |
---|
8 | from Products.PortalTransforms.interfaces import itransform |
---|
9 | |
---|
10 | import string |
---|
11 | letters = string.letters |
---|
12 | _ABSOLUTE_REL_URL=r'(?P<url>(?P<protocol>http|https|ftp|mailto|file|about[:/]+?)?[%s0-9_\@\.\,\?\!\/\:\;\-\#\~\=\&\%%\+\\]+)' % letters |
---|
13 | |
---|
14 | |
---|
15 | class ruid_to_url: |
---|
16 | """Transform which replaces resolve uid into urls""" |
---|
17 | |
---|
18 | __implements__ = itransform |
---|
19 | |
---|
20 | __name__ = "ruid_to_url" |
---|
21 | inputs = ('text/html',) |
---|
22 | output = 'text/html' |
---|
23 | |
---|
24 | def __init__(self, name=None): |
---|
25 | if name: |
---|
26 | self.__name__ = name |
---|
27 | self.ruid_regexp = re.compile("(?P<tag>\<(?P<tag_name>a|img)[^>]*(?P<url_attr>href|src)\s*=\s*[\'|\"]?%s[\'|\"]?[\s|\>|\/])" \ |
---|
28 | % _ABSOLUTE_REL_URL,re.I|re.S) |
---|
29 | |
---|
30 | def name(self): |
---|
31 | return self.__name__ |
---|
32 | |
---|
33 | def find_ruid(self, data): |
---|
34 | tags_url = [{m.group('tag'):m.group('url').replace('\\','/')}\ |
---|
35 | for m in self.ruid_regexp.finditer(data)] |
---|
36 | tags_ruid = [tu for tu in tags_url if tu.values()[0].startswith('resolveuid')] |
---|
37 | unique_ruid = [] |
---|
38 | [unique_ruid.append(tu.values()[0]) for tu in tags_ruid if tu.values()[0] not in unique_ruid] |
---|
39 | return tags_ruid, unique_ruid |
---|
40 | |
---|
41 | def mapRUID_URL(self, unique_ruid, portal): |
---|
42 | ruid_url = {} |
---|
43 | rc = getToolByName(portal, 'reference_catalog') |
---|
44 | pu = getToolByName(portal, 'portal_url') |
---|
45 | for ruid in unique_ruid: |
---|
46 | ruid_url[ruid] = pu.getRelativeUrl(rc.lookupObject(ruid.replace('resolveuid/', ''))) |
---|
47 | return ruid_url |
---|
48 | |
---|
49 | def convert(self, orig, data, **kwargs): |
---|
50 | text = orig |
---|
51 | tags_ruid, unique_ruid = self.find_ruid(text) |
---|
52 | if unique_ruid: |
---|
53 | ruid_url = self.mapRUID_URL(unique_ruid, kwargs['context']) |
---|
54 | for tag_ruid in tags_ruid: |
---|
55 | tag, ruid = tag_ruid.items()[0] |
---|
56 | text = text.replace(tag, tag.replace(ruid, ruid_url[ruid])) |
---|
57 | |
---|
58 | data.setData(text) |
---|
59 | return data |
---|
60 | |
---|
61 | |
---|
62 | def register(): |
---|
63 | return ruid_to_url() |
---|
64 | |
---|