Mercurial > libervia-backend
annotate src/plugins/plugin_xep_0071.py @ 1961:c73e08094a95
memory (sqlite): better handling of IntegrityError
| author | Goffi <goffi@goffi.org> |
|---|---|
| date | Sun, 19 Jun 2016 22:22:12 +0200 |
| parents | 633b5c21aefd |
| children | a2bc5089c2eb |
| rev | line source |
|---|---|
|
1934
2daf7b4c6756
use of /usr/bin/env instead of /usr/bin/python in shebang
Goffi <goffi@goffi.org>
parents:
1813
diff
changeset
|
1 #!/usr/bin/env python2 |
| 668 | 2 # -*- coding: utf-8 -*- |
| 3 | |
| 4 # SAT plugin for Publish-Subscribe (xep-0071) | |
| 1766 | 5 # Copyright (C) 2009-2016 Jérôme Poisson (goffi@goffi.org) |
| 668 | 6 |
| 7 # This program is free software: you can redistribute it and/or modify | |
| 8 # it under the terms of the GNU Affero General Public License as published by | |
| 9 # the Free Software Foundation, either version 3 of the License, or | |
| 10 # (at your option) any later version. | |
| 11 | |
| 12 # This program is distributed in the hope that it will be useful, | |
| 13 # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 15 # GNU Affero General Public License for more details. | |
| 16 | |
| 17 # You should have received a copy of the GNU Affero General Public License | |
| 18 # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
| 19 | |
| 771 | 20 from sat.core.i18n import _ |
|
832
c4b22aedb7d7
plugin groupblog, XEP-0071, XEP-0277, text_syntaxes: manage raw/rich/xhtml data for content/title:
souliane <souliane@mailoo.org>
parents:
811
diff
changeset
|
21 from sat.core import exceptions |
|
993
301b342c697a
core: use of the new core.log module:
Goffi <goffi@goffi.org>
parents:
922
diff
changeset
|
22 from sat.core.log import getLogger |
|
301b342c697a
core: use of the new core.log module:
Goffi <goffi@goffi.org>
parents:
922
diff
changeset
|
23 log = getLogger(__name__) |
| 668 | 24 |
|
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
25 from twisted.internet import defer |
|
993
301b342c697a
core: use of the new core.log module:
Goffi <goffi@goffi.org>
parents:
922
diff
changeset
|
26 from wokkel import disco, iwokkel |
| 668 | 27 from zope.interface import implements |
| 28 # from lxml import etree | |
|
1542
94901070478e
plugins: added new MissingModule exceptions to plugins using third party modules
Goffi <goffi@goffi.org>
parents:
1396
diff
changeset
|
29 try: |
|
94901070478e
plugins: added new MissingModule exceptions to plugins using third party modules
Goffi <goffi@goffi.org>
parents:
1396
diff
changeset
|
30 from lxml import html |
|
94901070478e
plugins: added new MissingModule exceptions to plugins using third party modules
Goffi <goffi@goffi.org>
parents:
1396
diff
changeset
|
31 except ImportError: |
|
94901070478e
plugins: added new MissingModule exceptions to plugins using third party modules
Goffi <goffi@goffi.org>
parents:
1396
diff
changeset
|
32 raise exceptions.MissingModule(u"Missing module lxml, please download/install it from http://lxml.de/") |
| 668 | 33 try: |
| 34 from twisted.words.protocols.xmlstream import XMPPHandler | |
| 35 except ImportError: | |
| 36 from wokkel.subprotocols import XMPPHandler | |
| 37 | |
| 38 NS_XHTML_IM = 'http://jabber.org/protocol/xhtml-im' | |
| 39 NS_XHTML = 'http://www.w3.org/1999/xhtml' | |
| 40 | |
| 41 PLUGIN_INFO = { | |
| 42 "name": "XHTML-IM Plugin", | |
| 43 "import_name": "XEP-0071", | |
| 44 "type": "XEP", | |
| 45 "protocols": ["XEP-0071"], | |
| 46 "dependencies": ["TEXT-SYNTAXES"], | |
| 47 "main": "XEP_0071", | |
| 48 "handler": "yes", | |
| 49 "description": _("""Implementation of XHTML-IM""") | |
| 50 } | |
| 51 | |
| 52 allowed = { | |
| 53 "a": set(["href", "style", "type"]), | |
| 54 "blockquote": set(["style"]), | |
| 55 "body": set(["style"]), | |
| 56 "br": set([]), | |
| 57 "cite": set(["style"]), | |
| 58 "em": set([]), | |
| 59 "img": set(["alt", "height", "src", "style", "width"]), | |
| 60 "li": set(["style"]), | |
| 61 "ol": set(["style"]), | |
| 62 "p": set(["style"]), | |
| 63 "span": set(["style"]), | |
| 64 "strong": set([]), | |
| 65 "ul": set(["style"]), | |
| 66 } | |
| 67 | |
| 68 styles_allowed = ["background-color", "color", "font-family", "font-size", "font-style", "font-weight", "margin-left", "margin-right", "text-align", "text-decoration"] | |
| 69 | |
| 70 blacklist = ['script'] # tag that we have to kill (we don't keep content) | |
| 71 | |
| 72 | |
| 73 class XEP_0071(object): | |
| 74 SYNTAX_XHTML_IM = "XHTML-IM" | |
| 75 | |
| 76 def __init__(self, host): | |
|
993
301b342c697a
core: use of the new core.log module:
Goffi <goffi@goffi.org>
parents:
922
diff
changeset
|
77 log.info(_("XHTML-IM plugin initialization")) |
| 668 | 78 self.host = host |
|
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
79 self._s = self.host.plugins["TEXT-SYNTAXES"] |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
80 self._s.addSyntax(self.SYNTAX_XHTML_IM, lambda xhtml: xhtml, self.XHTML2XHTML_IM, [self._s.OPT_HIDDEN]) |
| 668 | 81 host.trigger.add("MessageReceived", self.messageReceivedTrigger) |
|
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
82 host.trigger.add("messageSend", self.messageSendTrigger) |
| 668 | 83 |
| 84 def getHandler(self, profile): | |
| 85 return XEP_0071_handler(self) | |
| 86 | |
|
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
87 def _messagePostTreat(self, data, body_elts): |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
88 """Callback which manage the post treatment of the message in case of XHTML-IM found |
| 668 | 89 @param data: data send by MessageReceived trigger through post_treat deferred |
|
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
90 @param body_elts: XHTML-IM body elements found |
| 668 | 91 @return: the data with the extra parameter updated |
| 92 """ | |
|
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
93 # TODO: check if text only body is empty, then try to convert XHTML-IM to pure text and show a warning message |
| 668 | 94 def converted(xhtml): |
|
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
95 if lang: |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
96 data['extra']['xhtml_{}'.format(lang)] = xhtml |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
97 else: |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
98 data['extra']['xhtml'] = xhtml |
| 668 | 99 |
|
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
100 defers = [] |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
101 for body_elt in body_elts: |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
102 lang = body_elt.getAttribute('xml:lang', '') |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
103 d = self._s.convert(body_elt.toXml(), self.SYNTAX_XHTML_IM, safe=True) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
104 d.addCallback(converted, lang) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
105 defers.append(d) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
106 |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
107 d_list = defer.DeferredList(defers) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
108 d_list.addCallback(lambda dummy: data) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
109 return d_list |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
110 |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
111 def _messageSendAddRich(self, data, client): |
|
702
a25db3fe3959
plugin XEP-0071: rich messages management for sendMessage
Goffi <goffi@goffi.org>
parents:
701
diff
changeset
|
112 """ Construct XHTML-IM node and add it XML element |
|
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
113 |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
114 @param data: message data as sended by messageSend callback |
|
702
a25db3fe3959
plugin XEP-0071: rich messages management for sendMessage
Goffi <goffi@goffi.org>
parents:
701
diff
changeset
|
115 """ |
|
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
116 # at this point, either ['extra']['rich'] or ['extra']['xhtml'] exists |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
117 # but both can't exist at the same time |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
118 message_elt = data['xml'] |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
119 html_elt = message_elt.addElement((NS_XHTML_IM, 'html')) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
120 |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
121 def syntax_converted(xhtml_im, lang): |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
122 body_elt = html_elt.addElement((NS_XHTML, 'body')) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
123 if lang: |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
124 body_elt['xml:lang'] = lang |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
125 data['extra']['xhtml_{}'.format(lang)] = xhtml_im |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
126 else: |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
127 data['extra']['xhtml'] = xhtml_im |
|
702
a25db3fe3959
plugin XEP-0071: rich messages management for sendMessage
Goffi <goffi@goffi.org>
parents:
701
diff
changeset
|
128 body_elt.addRawXml(xhtml_im) |
|
a25db3fe3959
plugin XEP-0071: rich messages management for sendMessage
Goffi <goffi@goffi.org>
parents:
701
diff
changeset
|
129 |
|
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
130 syntax = self._s.getCurrentSyntax(client.profile) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
131 defers = [] |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
132 try: |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
133 rich = data['extra']['rich'] |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
134 except KeyError: |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
135 # we have directly XHTML |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
136 for lang, xhtml in data['extra']['xhtml'].iteritems(): |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
137 d = self._s.convert(xhtml, self._s.SYNTAX_XHTML, self.SYNTAX_XHTML_IM) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
138 d.addCallback(syntax_converted, lang) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
139 defers.append(d) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
140 else: |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
141 # we have rich syntax to convert |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
142 for lang, rich_data in rich.iteritems(): |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
143 d = self._s.convert(rich_data, syntax, self.SYNTAX_XHTML_IM) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
144 d.addCallback(syntax_converted, lang) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
145 defers.append(d) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
146 d_list = defer.DeferredList(defers) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
147 d_list.addCallback(lambda dummy: data) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
148 return d_list |
|
702
a25db3fe3959
plugin XEP-0071: rich messages management for sendMessage
Goffi <goffi@goffi.org>
parents:
701
diff
changeset
|
149 |
|
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
150 def messageReceivedTrigger(self, client, message, post_treat): |
| 668 | 151 """ Check presence of XHTML-IM in message |
| 152 """ | |
| 153 try: | |
| 154 html_elt = message.elements(NS_XHTML_IM, 'html').next() | |
| 155 except StopIteration: | |
| 156 # No XHTML-IM | |
| 157 pass | |
|
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
158 else: |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
159 body_elts = html_elt.elements(NS_XHTML, 'body') |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
160 post_treat.addCallback(self._messagePostTreat, body_elts) |
| 668 | 161 return True |
| 162 | |
|
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
163 def messageSendTrigger(self, client, data, pre_xml_treatments, post_xml_treatments): |
|
702
a25db3fe3959
plugin XEP-0071: rich messages management for sendMessage
Goffi <goffi@goffi.org>
parents:
701
diff
changeset
|
164 """ Check presence of rich text in extra |
|
a25db3fe3959
plugin XEP-0071: rich messages management for sendMessage
Goffi <goffi@goffi.org>
parents:
701
diff
changeset
|
165 """ |
|
1955
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
166 rich = {} |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
167 xhtml = {} |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
168 for key, value in data['extra'].iteritems(): |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
169 if key.startswith('rich'): |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
170 rich[key[5:]] = value |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
171 elif key.startswith('xhtml'): |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
172 xhtml[key[6:]] = value |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
173 if rich and xhtml: |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
174 raise exceptions.DataError(_(u"Can't have XHTML and rich content at the same time")) |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
175 if rich or xhtml: |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
176 if rich: |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
177 data['rich'] = rich |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
178 else: |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
179 data['xhtml'] = xhtml |
|
633b5c21aefd
backend, frontend: messages refactoring (huge commit, not finished):
Goffi <goffi@goffi.org>
parents:
1934
diff
changeset
|
180 post_xml_treatments.addCallback(self._messageSendAddRich, client) |
|
702
a25db3fe3959
plugin XEP-0071: rich messages management for sendMessage
Goffi <goffi@goffi.org>
parents:
701
diff
changeset
|
181 return True |
|
a25db3fe3959
plugin XEP-0071: rich messages management for sendMessage
Goffi <goffi@goffi.org>
parents:
701
diff
changeset
|
182 |
| 668 | 183 def _purgeStyle(self, styles_raw): |
| 184 """ Remove unauthorised styles according to the XEP-0071 | |
| 185 @param styles_raw: raw styles (value of the style attribute) | |
| 186 """ | |
| 187 purged = [] | |
| 188 | |
| 189 styles = [style.strip().split(':') for style in styles_raw.split(';')] | |
| 190 | |
| 191 for style_tuple in styles: | |
| 192 if len(style_tuple) != 2: | |
| 193 continue | |
| 194 name, value = style_tuple | |
| 195 name = name.strip() | |
| 196 if name not in styles_allowed: | |
| 197 continue | |
| 198 purged.append((name, value.strip())) | |
| 199 | |
| 200 return u'; '.join([u"%s: %s" % data for data in purged]) | |
| 201 | |
| 202 def XHTML2XHTML_IM(self, xhtml): | |
| 203 """ Convert XHTML document to XHTML_IM subset | |
| 204 @param xhtml: raw xhtml to convert | |
| 205 """ | |
| 206 # TODO: more clever tag replacement (replace forbidden tags with equivalents when possible) | |
| 207 | |
| 208 parser = html.HTMLParser(remove_comments=True, encoding='utf-8') | |
| 209 root = html.fromstring(xhtml, parser=parser) | |
| 210 body_elt = root.find('body') | |
| 211 if body_elt is None: | |
| 212 # we use the whole XML as body if no body element is found | |
| 213 body_elt = html.Element('body') | |
| 214 body_elt.append(root) | |
| 215 else: | |
| 216 body_elt.attrib.clear() | |
| 217 | |
| 218 allowed_tags = allowed.keys() | |
| 219 to_strip = [] | |
| 220 for elem in body_elt.iter(): | |
| 221 if elem.tag not in allowed_tags: | |
| 222 to_strip.append(elem) | |
| 223 else: | |
| 224 # we remove unallowed attributes | |
| 225 attrib = elem.attrib | |
| 226 att_to_remove = set(attrib).difference(allowed[elem.tag]) | |
| 227 for att in att_to_remove: | |
| 228 del(attrib[att]) | |
| 229 if "style" in attrib: | |
| 230 attrib["style"] = self._purgeStyle(attrib["style"]) | |
| 231 | |
| 232 for elem in to_strip: | |
| 233 if elem.tag in blacklist: | |
| 234 #we need to remove the element and all descendants | |
|
993
301b342c697a
core: use of the new core.log module:
Goffi <goffi@goffi.org>
parents:
922
diff
changeset
|
235 log.debug(u"removing black listed tag: %s" % (elem.tag)) |
| 668 | 236 elem.drop_tree() |
| 237 else: | |
| 238 elem.drop_tag() | |
|
701
98b2400e17d6
plugin XEP-0071: XHTML2XHTML_IM don't return the <body> root tag anymore.
Goffi <goffi@goffi.org>
parents:
668
diff
changeset
|
239 if len(body_elt) !=1: |
|
98b2400e17d6
plugin XEP-0071: XHTML2XHTML_IM don't return the <body> root tag anymore.
Goffi <goffi@goffi.org>
parents:
668
diff
changeset
|
240 root_elt = body_elt |
|
98b2400e17d6
plugin XEP-0071: XHTML2XHTML_IM don't return the <body> root tag anymore.
Goffi <goffi@goffi.org>
parents:
668
diff
changeset
|
241 body_elt.tag = "p" |
|
98b2400e17d6
plugin XEP-0071: XHTML2XHTML_IM don't return the <body> root tag anymore.
Goffi <goffi@goffi.org>
parents:
668
diff
changeset
|
242 else: |
|
98b2400e17d6
plugin XEP-0071: XHTML2XHTML_IM don't return the <body> root tag anymore.
Goffi <goffi@goffi.org>
parents:
668
diff
changeset
|
243 root_elt = body_elt[0] |
| 668 | 244 |
|
701
98b2400e17d6
plugin XEP-0071: XHTML2XHTML_IM don't return the <body> root tag anymore.
Goffi <goffi@goffi.org>
parents:
668
diff
changeset
|
245 return html.tostring(root_elt, encoding='unicode', method='xml') |
| 668 | 246 |
| 247 class XEP_0071_handler(XMPPHandler): | |
| 248 implements(iwokkel.IDisco) | |
| 249 | |
| 250 def __init__(self, plugin_parent): | |
| 251 self.plugin_parent = plugin_parent | |
| 252 self.host = plugin_parent.host | |
| 253 | |
| 254 def getDiscoInfo(self, requestor, target, nodeIdentifier=''): | |
| 255 return [disco.DiscoFeature(NS_XHTML_IM)] | |
| 256 | |
| 257 def getDiscoItems(self, requestor, target, nodeIdentifier=''): | |
| 258 return [] |