author | Aurelien Campeas <aurelien.campeas@logilab.fr> |
Tue, 20 Oct 2009 14:24:19 +0200 | |
branch | stable |
changeset 3748 | 7143198dd8a4 |
parent 2650 | 18aec79ec3a3 |
child 2798 | 9c650701cb17 |
child 4212 | ab6573088b4a |
permissions | -rw-r--r-- |
0 | 1 |
"""Objects interacting together to provides the external page embeding |
2 |
functionality. |
|
3 |
||
4 |
||
5 |
:organization: Logilab |
|
1977
606923dff11b
big bunch of copyright / docstring update
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
1802
diff
changeset
|
6 |
:copyright: 2001-2009 LOGILAB S.A. (Paris, FRANCE), license is LGPL v2. |
0 | 7 |
:contact: http://www.logilab.fr/ -- mailto:contact@logilab.fr |
1977
606923dff11b
big bunch of copyright / docstring update
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
1802
diff
changeset
|
8 |
:license: GNU Lesser General Public License, v2.1 - http://www.gnu.org/licenses |
0 | 9 |
""" |
10 |
__docformat__ = "restructuredtext en" |
|
11 |
||
12 |
import re |
|
13 |
from urlparse import urljoin |
|
14 |
from urllib2 import urlopen, Request, HTTPError |
|
15 |
||
16 |
from logilab.mtconverter import guess_encoding |
|
17 |
||
18 |
from cubicweb import urlquote # XXX should use view.url_quote method |
|
692
800592b8d39b
replace deprecated cubicweb.common.selectors by its new module path (cubicweb.selectors)
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
631
diff
changeset
|
19 |
from cubicweb.selectors import (one_line_rset, score_entity, |
800592b8d39b
replace deprecated cubicweb.common.selectors by its new module path (cubicweb.selectors)
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
631
diff
changeset
|
20 |
match_search_state, implements) |
0 | 21 |
from cubicweb.interfaces import IEmbedable |
800 | 22 |
from cubicweb.view import NOINDEX, NOFOLLOW |
0 | 23 |
from cubicweb.common.uilib import soup2xhtml |
24 |
from cubicweb.web.controller import Controller |
|
25 |
from cubicweb.web.action import Action |
|
26 |
from cubicweb.web.views import basetemplates |
|
27 |
||
28 |
||
29 |
class ExternalTemplate(basetemplates.TheMainTemplate): |
|
30 |
"""template embeding an external web pages into CubicWeb web interface |
|
31 |
""" |
|
32 |
id = 'external' |
|
1802
d628defebc17
delete-trailing-whitespace + some copyright update
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
1132
diff
changeset
|
33 |
|
0 | 34 |
def call(self, body): |
35 |
# XXX fallback to HTML 4 mode when embeding ? |
|
36 |
self.set_request_content_type() |
|
37 |
self.req.search_state = ('normal',) |
|
38 |
self.template_header(self.content_type, None, self.req._('external page'), |
|
39 |
[NOINDEX, NOFOLLOW]) |
|
40 |
self.content_header() |
|
41 |
self.w(body) |
|
42 |
self.content_footer() |
|
43 |
self.template_footer() |
|
44 |
||
45 |
||
46 |
class EmbedController(Controller): |
|
47 |
id = 'embed' |
|
48 |
template = 'external' |
|
49 |
||
50 |
def publish(self, rset=None): |
|
51 |
req = self.req |
|
52 |
if 'custom_css' in req.form: |
|
53 |
req.add_css(req.form['custom_css']) |
|
54 |
embedded_url = req.form['url'] |
|
55 |
allowed = self.config['embed-allowed'] |
|
56 |
_ = req._ |
|
57 |
if allowed is None or not allowed.match(embedded_url): |
|
58 |
body = '<h2>%s</h2><h3>%s</h3>' % ( |
|
59 |
_('error while embedding page'), |
|
60 |
_('embedding this url is forbidden')) |
|
61 |
else: |
|
62 |
prefix = req.build_url(self.id, url='') |
|
63 |
authorization = req.get_header('Authorization') |
|
64 |
if authorization: |
|
65 |
headers = {'Authorization' : authorization} |
|
66 |
else: |
|
67 |
headers = {} |
|
68 |
try: |
|
69 |
body = embed_external_page(embedded_url, prefix, |
|
70 |
headers, req.form.get('custom_css')) |
|
71 |
body = soup2xhtml(body, self.req.encoding) |
|
72 |
except HTTPError, err: |
|
73 |
body = '<h2>%s</h2><h3>%s</h3>' % ( |
|
74 |
_('error while embedding page'), err) |
|
1092
b8fbb95dc0eb
process_rql now done in the controller
sylvain.thenault@logilab.fr
parents:
800
diff
changeset
|
75 |
self.process_rql(req.form.get('rql')) |
2650
18aec79ec3a3
R [vreg] important refactoring of the vregistry, moving behaviour to end dictionnary (and so leaving room for more flexibility ; keep bw compat ; update api usage in cw
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents:
1977
diff
changeset
|
76 |
return self.vreg['views'].main_template(req, self.template, |
18aec79ec3a3
R [vreg] important refactoring of the vregistry, moving behaviour to end dictionnary (and so leaving room for more flexibility ; keep bw compat ; update api usage in cw
Sylvain Thénault <sylvain.thenault@logilab.fr>
parents:
1977
diff
changeset
|
77 |
rset=self.rset, body=body) |
0 | 78 |
|
79 |
||
631
99f5852f8604
major selector refactoring (mostly to avoid looking for select parameters on the target class), start accept / interface unification)
sylvain.thenault@logilab.fr
parents:
431
diff
changeset
|
80 |
def entity_has_embedable_url(entity): |
99f5852f8604
major selector refactoring (mostly to avoid looking for select parameters on the target class), start accept / interface unification)
sylvain.thenault@logilab.fr
parents:
431
diff
changeset
|
81 |
"""return 1 if the entity provides an allowed embedable url""" |
99f5852f8604
major selector refactoring (mostly to avoid looking for select parameters on the target class), start accept / interface unification)
sylvain.thenault@logilab.fr
parents:
431
diff
changeset
|
82 |
url = entity.embeded_url() |
99f5852f8604
major selector refactoring (mostly to avoid looking for select parameters on the target class), start accept / interface unification)
sylvain.thenault@logilab.fr
parents:
431
diff
changeset
|
83 |
if not url or not url.strip(): |
99f5852f8604
major selector refactoring (mostly to avoid looking for select parameters on the target class), start accept / interface unification)
sylvain.thenault@logilab.fr
parents:
431
diff
changeset
|
84 |
return 0 |
99f5852f8604
major selector refactoring (mostly to avoid looking for select parameters on the target class), start accept / interface unification)
sylvain.thenault@logilab.fr
parents:
431
diff
changeset
|
85 |
allowed = entity.config['embed-allowed'] |
99f5852f8604
major selector refactoring (mostly to avoid looking for select parameters on the target class), start accept / interface unification)
sylvain.thenault@logilab.fr
parents:
431
diff
changeset
|
86 |
if allowed is None or not allowed.match(url): |
99f5852f8604
major selector refactoring (mostly to avoid looking for select parameters on the target class), start accept / interface unification)
sylvain.thenault@logilab.fr
parents:
431
diff
changeset
|
87 |
return 0 |
99f5852f8604
major selector refactoring (mostly to avoid looking for select parameters on the target class), start accept / interface unification)
sylvain.thenault@logilab.fr
parents:
431
diff
changeset
|
88 |
return 1 |
99f5852f8604
major selector refactoring (mostly to avoid looking for select parameters on the target class), start accept / interface unification)
sylvain.thenault@logilab.fr
parents:
431
diff
changeset
|
89 |
|
99f5852f8604
major selector refactoring (mostly to avoid looking for select parameters on the target class), start accept / interface unification)
sylvain.thenault@logilab.fr
parents:
431
diff
changeset
|
90 |
|
0 | 91 |
class EmbedAction(Action): |
92 |
"""display an 'embed' link on entity implementing `embeded_url` method |
|
93 |
if the returned url match embeding configuration |
|
94 |
""" |
|
95 |
id = 'embed' |
|
742
99115e029dca
replaced most of __selectors__ assignments with __select__
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
692
diff
changeset
|
96 |
__select__ = (one_line_rset() & match_search_state('normal') |
1802
d628defebc17
delete-trailing-whitespace + some copyright update
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
1132
diff
changeset
|
97 |
& implements(IEmbedable) |
742
99115e029dca
replaced most of __selectors__ assignments with __select__
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
692
diff
changeset
|
98 |
& score_entity(entity_has_embedable_url)) |
1802
d628defebc17
delete-trailing-whitespace + some copyright update
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
1132
diff
changeset
|
99 |
|
0 | 100 |
title = _('embed') |
631
99f5852f8604
major selector refactoring (mostly to avoid looking for select parameters on the target class), start accept / interface unification)
sylvain.thenault@logilab.fr
parents:
431
diff
changeset
|
101 |
controller = 'embed' |
1802
d628defebc17
delete-trailing-whitespace + some copyright update
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
1132
diff
changeset
|
102 |
|
0 | 103 |
def url(self, row=0): |
104 |
entity = self.rset.get_entity(row, 0) |
|
105 |
url = urljoin(self.req.base_url(), entity.embeded_url()) |
|
106 |
if self.req.form.has_key('rql'): |
|
107 |
return self.build_url(url=url, rql=self.req.form['rql']) |
|
108 |
return self.build_url(url=url) |
|
109 |
||
110 |
||
111 |
||
112 |
# functions doing necessary substitutions to embed an external html page ###### |
|
113 |
||
114 |
||
115 |
BODY_RGX = re.compile('<body.*?>(.*?)</body>', re.I | re.S | re.U) |
|
116 |
HREF_RGX = re.compile('<a\s+href="([^"]*)"', re.I | re.S | re.U) |
|
117 |
SRC_RGX = re.compile('<img\s+src="([^"]*)"', re.I | re.S | re.U) |
|
118 |
||
119 |
||
120 |
class replace_href: |
|
121 |
def __init__(self, prefix, custom_css=None): |
|
122 |
self.prefix = prefix |
|
123 |
self.custom_css = custom_css |
|
1802
d628defebc17
delete-trailing-whitespace + some copyright update
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
1132
diff
changeset
|
124 |
|
0 | 125 |
def __call__(self, match): |
126 |
original_url = match.group(1) |
|
127 |
url = self.prefix + urlquote(original_url, safe='') |
|
128 |
if self.custom_css is not None: |
|
129 |
if '?' in url: |
|
130 |
url = '%s&custom_css=%s' % (url, self.custom_css) |
|
131 |
else: |
|
132 |
url = '%s?custom_css=%s' % (url, self.custom_css) |
|
133 |
return '<a href="%s"' % url |
|
134 |
||
631
99f5852f8604
major selector refactoring (mostly to avoid looking for select parameters on the target class), start accept / interface unification)
sylvain.thenault@logilab.fr
parents:
431
diff
changeset
|
135 |
|
0 | 136 |
class absolutize_links: |
137 |
def __init__(self, embedded_url, tag, custom_css=None): |
|
138 |
self.embedded_url = embedded_url |
|
139 |
self.tag = tag |
|
140 |
self.custom_css = custom_css |
|
1802
d628defebc17
delete-trailing-whitespace + some copyright update
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
1132
diff
changeset
|
141 |
|
0 | 142 |
def __call__(self, match): |
143 |
original_url = match.group(1) |
|
144 |
if '://' in original_url: |
|
145 |
return match.group(0) # leave it unchanged |
|
146 |
return '%s="%s"' % (self.tag, urljoin(self.embedded_url, original_url)) |
|
147 |
||
148 |
||
149 |
def prefix_links(body, prefix, embedded_url, custom_css=None): |
|
150 |
filters = ((HREF_RGX, absolutize_links(embedded_url, '<a href', custom_css)), |
|
151 |
(SRC_RGX, absolutize_links(embedded_url, '<img src')), |
|
152 |
(HREF_RGX, replace_href(prefix, custom_css))) |
|
153 |
for rgx, repl in filters: |
|
154 |
body = rgx.sub(repl, body) |
|
155 |
return body |
|
631
99f5852f8604
major selector refactoring (mostly to avoid looking for select parameters on the target class), start accept / interface unification)
sylvain.thenault@logilab.fr
parents:
431
diff
changeset
|
156 |
|
99f5852f8604
major selector refactoring (mostly to avoid looking for select parameters on the target class), start accept / interface unification)
sylvain.thenault@logilab.fr
parents:
431
diff
changeset
|
157 |
|
0 | 158 |
def embed_external_page(url, prefix, headers=None, custom_css=None): |
159 |
req = Request(url, headers=(headers or {})) |
|
160 |
content = urlopen(req).read() |
|
161 |
page_source = unicode(content, guess_encoding(content), 'replace') |
|
1132 | 162 |
page_source = page_source |
0 | 163 |
match = BODY_RGX.search(page_source) |
164 |
if match is None: |
|
165 |
return page_source |
|
166 |
return prefix_links(match.group(1), prefix, url, custom_css) |