author | Sandrine Ribeau <sandrine.ribeau@logilab.fr> |
Thu, 23 Apr 2009 13:46:57 -0700 | |
changeset 1674 | e4383a510089 |
parent 1157 | 81a383cdda5c |
child 1263 | 01152fffd593 |
permissions | -rw-r--r-- |
0 | 1 |
# -*- coding: utf-8 -*- |
2 |
"""user interface libraries |
|
3 |
||
4 |
contains some functions designed to help implementation of cubicweb user interface |
|
5 |
||
6 |
:organization: Logilab |
|
525
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
7 |
:copyright: 2001-2009 LOGILAB S.A. (Paris, FRANCE), all rights reserved. |
0 | 8 |
:contact: http://www.logilab.fr/ -- mailto:contact@logilab.fr |
9 |
""" |
|
10 |
__docformat__ = "restructuredtext en" |
|
11 |
||
12 |
import csv |
|
13 |
import decimal |
|
14 |
import locale |
|
15 |
import re |
|
16 |
from urllib import quote as urlquote |
|
17 |
from cStringIO import StringIO |
|
164 | 18 |
from copy import deepcopy |
0 | 19 |
|
20 |
import simplejson |
|
21 |
||
22 |
from mx.DateTime import DateTimeType, DateTimeDeltaType |
|
23 |
||
24 |
from logilab.common.textutils import unormalize |
|
362
a6a319f000c3
use mtconverter's html_unescape rather than saxutils' escape to deal with any html entity
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
350
diff
changeset
|
25 |
from logilab.mtconverter import html_escape, html_unescape |
0 | 26 |
|
27 |
def ustrftime(date, fmt='%Y-%m-%d'): |
|
28 |
"""like strftime, but returns a unicode string instead of an encoded |
|
29 |
string which may be problematic with localized date. |
|
30 |
|
|
31 |
encoding is guessed by locale.getpreferredencoding() |
|
32 |
""" |
|
33 |
# date format may depend on the locale |
|
34 |
encoding = locale.getpreferredencoding(do_setlocale=False) or 'UTF-8' |
|
35 |
return unicode(date.strftime(fmt), encoding) |
|
36 |
||
37 |
||
38 |
def rql_for_eid(eid): |
|
39 |
"""return the rql query necessary to fetch entity with the given eid. This |
|
40 |
function should only be used to generate link with rql inside, not to give |
|
41 |
to cursor.execute (in which case you won't benefit from rql cache). |
|
42 |
||
43 |
:Parameters: |
|
44 |
- `eid`: the eid of the entity we should search |
|
45 |
:rtype: str |
|
46 |
:return: the rql query |
|
47 |
""" |
|
48 |
return 'Any X WHERE X eid %s' % eid |
|
49 |
||
50 |
||
51 |
def printable_value(req, attrtype, value, props=None, displaytime=True): |
|
52 |
"""return a displayable value (i.e. unicode string)""" |
|
53 |
if value is None or attrtype == 'Bytes': |
|
54 |
return u'' |
|
55 |
if attrtype == 'String': |
|
56 |
# don't translate empty value if you don't want strange results |
|
57 |
if props is not None and value and props.get('internationalizable'): |
|
58 |
return req._(value) |
|
59 |
||
60 |
return value |
|
61 |
if attrtype == 'Date': |
|
62 |
return ustrftime(value, req.property_value('ui.date-format')) |
|
63 |
if attrtype == 'Time': |
|
64 |
return ustrftime(value, req.property_value('ui.time-format')) |
|
65 |
if attrtype == 'Datetime': |
|
66 |
if not displaytime: |
|
67 |
return ustrftime(value, req.property_value('ui.date-format')) |
|
68 |
return ustrftime(value, req.property_value('ui.datetime-format')) |
|
69 |
if attrtype == 'Boolean': |
|
70 |
if value: |
|
71 |
return req._('yes') |
|
72 |
return req._('no') |
|
73 |
if attrtype == 'Float': |
|
74 |
value = req.property_value('ui.float-format') % value |
|
75 |
return unicode(value) |
|
76 |
||
77 |
||
78 |
# text publishing ############################################################# |
|
79 |
||
80 |
try: |
|
81 |
from cubicweb.common.rest import rest_publish # pylint: disable-msg=W0611 |
|
82 |
except ImportError: |
|
83 |
def rest_publish(entity, data): |
|
84 |
"""default behaviour if docutils was not found""" |
|
85 |
return data |
|
86 |
||
87 |
TAG_PROG = re.compile(r'</?.*?>', re.U) |
|
88 |
def remove_html_tags(text): |
|
89 |
"""Removes HTML tags from text |
|
90 |
||
91 |
>>> remove_html_tags('<td>hi <a href="http://www.google.fr">world</a></td>') |
|
92 |
'hi world' |
|
93 |
>>> |
|
94 |
""" |
|
95 |
return TAG_PROG.sub('', text) |
|
96 |
||
97 |
||
98 |
REF_PROG = re.compile(r"<ref\s+rql=([\'\"])([^\1]*?)\1\s*>([^<]*)</ref>", re.U) |
|
99 |
def _subst_rql(view, obj): |
|
100 |
delim, rql, descr = obj.groups() |
|
101 |
return u'<a href="%s">%s</a>' % (view.build_url(rql=rql), descr) |
|
102 |
||
103 |
def html_publish(view, text): |
|
104 |
"""replace <ref rql=''> links by <a href="...">""" |
|
105 |
if not text: |
|
106 |
return u'' |
|
107 |
return REF_PROG.sub(lambda obj, view=view:_subst_rql(view, obj), text) |
|
108 |
||
277
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
109 |
# fallback implementation, nicer one defined below if lxml is available |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
110 |
def soup2xhtml(data, encoding): |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
111 |
return data |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
112 |
|
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
113 |
# fallback implementation, nicer one defined below if lxml> 2.0 is available |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
114 |
def safe_cut(text, length): |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
115 |
"""returns a string of length <length> based on <text>, removing any html |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
116 |
tags from given text if cut is necessary.""" |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
117 |
if text is None: |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
118 |
return u'' |
362
a6a319f000c3
use mtconverter's html_unescape rather than saxutils' escape to deal with any html entity
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
350
diff
changeset
|
119 |
noenttext = html_unescape(text) |
350
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
120 |
text_nohtml = remove_html_tags(noenttext) |
277
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
121 |
# try to keep html tags if text is short enough |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
122 |
if len(text_nohtml) <= length: |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
123 |
return text |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
124 |
# else if un-tagged text is too long, cut it |
350
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
125 |
return html_escape(text_nohtml[:length] + u'...') |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
126 |
|
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
127 |
fallback_safe_cut = safe_cut |
277
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
128 |
|
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
129 |
|
0 | 130 |
try: |
131 |
from lxml import etree |
|
228
27b958dc72ae
[lxml] lxml version < 2 does not provide an iter method on some elements
Aurelien Campeas <aurelien.campeas@logilab.fr>
parents:
165
diff
changeset
|
132 |
except (ImportError, AttributeError): |
0 | 133 |
# gae environment: lxml not availabel |
277
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
134 |
pass |
0 | 135 |
else: |
136 |
||
137 |
def soup2xhtml(data, encoding): |
|
138 |
"""tidy (at least try) html soup and return the result |
|
139 |
Note: the function considers a string with no surrounding tag as valid |
|
140 |
if <div>`data`</div> can be parsed by an XML parser |
|
141 |
""" |
|
142 |
xmltree = etree.HTML('<div>%s</div>' % data) |
|
143 |
# NOTE: lxml 1.1 (etch platforms) doesn't recognize |
|
144 |
# the encoding=unicode parameter (lxml 2.0 does), this is |
|
145 |
# why we specify an encoding and re-decode to unicode later |
|
146 |
body = etree.tostring(xmltree[0], encoding=encoding) |
|
147 |
# remove <body> and </body> and decode to unicode |
|
148 |
return body[11:-13].decode(encoding) |
|
149 |
||
277
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
150 |
if hasattr(etree.HTML('<div>test</div>'), 'iter'): |
165 | 151 |
|
277
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
152 |
def safe_cut(text, length): |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
153 |
"""returns an html document of length <length> based on <text>, |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
154 |
and cut is necessary. |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
155 |
""" |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
156 |
if text is None: |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
157 |
return u'' |
350
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
158 |
dom = etree.HTML(text) |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
159 |
curlength = 0 |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
160 |
add_ellipsis = False |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
161 |
for element in dom.iter(): |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
162 |
if curlength >= length: |
277
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
163 |
parent = element.getparent() |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
164 |
parent.remove(element) |
350
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
165 |
if curlength == length and (element.text or element.tail): |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
166 |
add_ellipsis = True |
277
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
167 |
else: |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
168 |
if element.text is not None: |
350
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
169 |
element.text = cut(element.text, length - curlength) |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
170 |
curlength += len(element.text) |
277
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
171 |
if element.tail is not None: |
350
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
172 |
if curlength < length: |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
173 |
element.tail = cut(element.tail, length - curlength) |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
174 |
curlength += len(element.tail) |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
175 |
elif curlength == length: |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
176 |
element.tail = '...' |
277
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
177 |
else: |
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
178 |
element.tail = '' |
350
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
179 |
text = etree.tounicode(dom[0])[6:-7] # remove wrapping <body></body> |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
180 |
if add_ellipsis: |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
181 |
return text + u'...' |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
182 |
return text |
1157
81a383cdda5c
text_cut() accepts a gotoperiod parameter
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
927
diff
changeset
|
183 |
|
81a383cdda5c
text_cut() accepts a gotoperiod parameter
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
927
diff
changeset
|
184 |
def text_cut(text, nbwords=30, gotoperiod=True): |
350
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
185 |
"""from the given plain text, return a text with at least <nbwords> words, |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
186 |
trying to go to the end of the current sentence. |
277
a11a3c231050
fix lxml is available, we can have a nicer version of soup2xhtml even if its lxml < 2.0
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
228
diff
changeset
|
187 |
|
1157
81a383cdda5c
text_cut() accepts a gotoperiod parameter
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
927
diff
changeset
|
188 |
:param nbwords: the minimum number of words required |
81a383cdda5c
text_cut() accepts a gotoperiod parameter
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
927
diff
changeset
|
189 |
:param gotoperiod: specifies if the function should try to go to |
81a383cdda5c
text_cut() accepts a gotoperiod parameter
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
927
diff
changeset
|
190 |
the first period after the cut (i.e. finish |
81a383cdda5c
text_cut() accepts a gotoperiod parameter
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
927
diff
changeset
|
191 |
the sentence if possible) |
81a383cdda5c
text_cut() accepts a gotoperiod parameter
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
927
diff
changeset
|
192 |
|
350
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
193 |
Note that spaces are normalized. |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
194 |
""" |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
195 |
if text is None: |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
196 |
return u'' |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
197 |
words = text.split() |
927
bfcc610c3d5e
text_cut must return unicode not string
Stephanie Marcu <stephanie.marcu@logilab.fr>
parents:
525
diff
changeset
|
198 |
text = u' '.join(words) # normalize spaces |
1157
81a383cdda5c
text_cut() accepts a gotoperiod parameter
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
927
diff
changeset
|
199 |
textlength = minlength = len(' '.join(words[:nbwords])) |
81a383cdda5c
text_cut() accepts a gotoperiod parameter
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
927
diff
changeset
|
200 |
if gotoperiod: |
81a383cdda5c
text_cut() accepts a gotoperiod parameter
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
927
diff
changeset
|
201 |
textlength = text.find('.', minlength) + 1 |
81a383cdda5c
text_cut() accepts a gotoperiod parameter
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
927
diff
changeset
|
202 |
if textlength == 0: # no period found |
81a383cdda5c
text_cut() accepts a gotoperiod parameter
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
927
diff
changeset
|
203 |
textlength = minlength |
350
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
204 |
return text[:textlength] |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
205 |
|
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
206 |
def cut(text, length): |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
207 |
"""returns a string of a maximum length <length> based on <text> |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
208 |
(approximatively, since if text has been cut, '...' is added to the end of the string, |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
209 |
resulting in a string of len <length> + 3) |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
210 |
""" |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
211 |
if text is None: |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
212 |
return u'' |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
213 |
if len(text) <= length: |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
214 |
return text |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
215 |
# else if un-tagged text is too long, cut it |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
216 |
return text[:length] + u'...' |
f34ef2c64605
cleanup/fix cut variants
Sylvain Thenault <sylvain.thenault@logilab.fr>
parents:
277
diff
changeset
|
217 |
|
165 | 218 |
|
0 | 219 |
|
220 |
# HTML generation helper functions ############################################ |
|
221 |
||
525
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
222 |
def simple_sgml_tag(tag, content=None, **attrs): |
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
223 |
"""generation of a simple sgml tag (eg without children tags) easier |
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
224 |
|
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
225 |
content and attributes will be escaped |
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
226 |
""" |
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
227 |
value = u'<%s' % tag |
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
228 |
if attrs: |
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
229 |
value += u' ' + u' '.join(u'%s="%s"' % (attr, html_escape(unicode(value))) |
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
230 |
for attr, value in attrs.items()) |
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
231 |
if content: |
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
232 |
value += u'>%s</%s>' % (html_escape(unicode(content)), tag) |
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
233 |
else: |
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
234 |
value += u'/>' |
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
235 |
return value |
bd4e03297cf0
function to make generation of a simple sgml tag
sylvain.thenault@logilab.fr
parents:
362
diff
changeset
|
236 |
|
0 | 237 |
def tooltipize(text, tooltip, url=None): |
238 |
"""make an HTML tooltip""" |
|
239 |
url = url or '#' |
|
240 |
return u'<a href="%s" title="%s">%s</a>' % (url, tooltip, text) |
|
241 |
||
242 |
def toggle_action(nodeid): |
|
243 |
"""builds a HTML link that uses the js toggleVisibility function""" |
|
244 |
return u"javascript: toggleVisibility('%s')" % nodeid |
|
245 |
||
246 |
def toggle_link(nodeid, label): |
|
247 |
"""builds a HTML link that uses the js toggleVisibility function""" |
|
248 |
return u'<a href="%s">%s</a>' % (toggle_action(nodeid), label) |
|
249 |
||
250 |
def ajax_replace_url(nodeid, rql, vid=None, swap=False, **extraparams): |
|
251 |
"""builds a replacePageChunk-like url |
|
252 |
>>> ajax_replace_url('foo', 'Person P') |
|
253 |
"javascript: replacePageChunk('foo', 'Person%20P');" |
|
254 |
>>> ajax_replace_url('foo', 'Person P', 'oneline') |
|
255 |
"javascript: replacePageChunk('foo', 'Person%20P', 'oneline');" |
|
256 |
>>> ajax_replace_url('foo', 'Person P', 'oneline', name='bar', age=12) |
|
257 |
"javascript: replacePageChunk('foo', 'Person%20P', 'oneline', {'age':12, 'name':'bar'});" |
|
258 |
>>> ajax_replace_url('foo', 'Person P', name='bar', age=12) |
|
259 |
"javascript: replacePageChunk('foo', 'Person%20P', 'null', {'age':12, 'name':'bar'});" |
|
260 |
""" |
|
261 |
params = [repr(nodeid), repr(urlquote(rql))] |
|
262 |
if extraparams and not vid: |
|
263 |
params.append("'null'") |
|
264 |
elif vid: |
|
265 |
params.append(repr(vid)) |
|
266 |
if extraparams: |
|
267 |
params.append(simplejson.dumps(extraparams)) |
|
268 |
if swap: |
|
269 |
params.append('true') |
|
270 |
return "javascript: replacePageChunk(%s);" % ', '.join(params) |
|
271 |
||
272 |
||
273 |
from StringIO import StringIO |
|
274 |
||
275 |
def ureport_as_html(layout): |
|
276 |
from logilab.common.ureports import HTMLWriter |
|
277 |
formater = HTMLWriter(True) |
|
278 |
stream = StringIO() #UStringIO() don't want unicode assertion |
|
279 |
formater.format(layout, stream) |
|
280 |
res = stream.getvalue() |
|
281 |
if isinstance(res, str): |
|
282 |
res = unicode(res, 'UTF8') |
|
283 |
return res |
|
284 |
||
285 |
def render_HTML_tree(tree, selected_node=None, render_node=None, caption=None): |
|
286 |
""" |
|
287 |
Generate a pure HTML representation of a tree given as an instance |
|
288 |
of a logilab.common.tree.Node |
|
289 |
||
290 |
selected_node is the currently selected node (if any) which will |
|
291 |
have its surrounding <div> have id="selected" (which default |
|
292 |
to a bold border libe with the default CSS). |
|
293 |
||
294 |
render_node is a function that should take a Node content (Node.id) |
|
295 |
as parameter and should return a string (what will be displayed |
|
296 |
in the cell). |
|
297 |
||
298 |
Warning: proper rendering of the generated html code depends on html_tree.css |
|
299 |
""" |
|
300 |
tree_depth = tree.depth_down() |
|
301 |
if render_node is None: |
|
302 |
render_node = str |
|
303 |
||
304 |
# helper function that build a matrix from the tree, like: |
|
305 |
# +------+-----------+-----------+ |
|
306 |
# | root | child_1_1 | child_2_1 | |
|
307 |
# | root | child_1_1 | child_2_2 | |
|
308 |
# | root | child_1_2 | | |
|
309 |
# | root | child_1_3 | child_2_3 | |
|
310 |
# | root | child_1_3 | child_2_4 | |
|
311 |
# +------+-----------+-----------+ |
|
312 |
# from: |
|
313 |
# root -+- child_1_1 -+- child_2_1 |
|
314 |
# | | |
|
315 |
# | +- child_2_2 |
|
316 |
# +- child_1_2 |
|
317 |
# | |
|
318 |
# +- child1_3 -+- child_2_3 |
|
319 |
# | |
|
320 |
# +- child_2_2 |
|
321 |
def build_matrix(path, matrix): |
|
322 |
if path[-1].is_leaf(): |
|
323 |
matrix.append(path[:]) |
|
324 |
else: |
|
325 |
for child in path[-1].children: |
|
326 |
build_matrix(path[:] + [child], matrix) |
|
327 |
||
328 |
matrix = [] |
|
329 |
build_matrix([tree], matrix) |
|
330 |
||
331 |
# make all lines in the matrix have the same number of columns |
|
332 |
for line in matrix: |
|
333 |
line.extend([None]*(tree_depth-len(line))) |
|
334 |
for i in range(len(matrix)-1, 0, -1): |
|
335 |
prev_line, line = matrix[i-1:i+1] |
|
336 |
for j in range(len(line)): |
|
337 |
if line[j] == prev_line[j]: |
|
338 |
line[j] = None |
|
339 |
||
340 |
# We build the matrix of link types (between 2 cells on a line of the matrix) |
|
341 |
# link types are : |
|
342 |
link_types = {(True, True, True ): 1, # T |
|
343 |
(False, False, True ): 2, # | |
|
344 |
(False, True, True ): 3, # + (actually, vert. bar with horiz. bar on the right) |
|
345 |
(False, True, False): 4, # L |
|
346 |
(True, True, False): 5, # - |
|
347 |
} |
|
348 |
links = [] |
|
349 |
for i, line in enumerate(matrix): |
|
350 |
links.append([]) |
|
351 |
for j in range(tree_depth-1): |
|
352 |
cell_11 = line[j] is not None |
|
353 |
cell_12 = line[j+1] is not None |
|
354 |
cell_21 = line[j+1] is not None and line[j+1].next_sibling() is not None |
|
355 |
link_type = link_types.get((cell_11, cell_12, cell_21), 0) |
|
356 |
if link_type == 0 and i > 0 and links[i-1][j] in (1,2,3): |
|
357 |
link_type = 2 |
|
358 |
links[-1].append(link_type) |
|
359 |
||
360 |
||
361 |
# We can now generate the HTML code for the <table> |
|
362 |
s = u'<table class="tree">\n' |
|
363 |
if caption: |
|
364 |
s += '<caption>%s</caption>\n' % caption |
|
365 |
||
366 |
for i, link_line in enumerate(links): |
|
367 |
line = matrix[i] |
|
368 |
||
369 |
s += '<tr>' |
|
370 |
for j, link_cell in enumerate(link_line): |
|
371 |
cell = line[j] |
|
372 |
if cell: |
|
373 |
if cell.id == selected_node: |
|
374 |
s += '<td class="tree_cell" rowspan="2"><div id="selected" class="tree_cell">%s</div></td>' % (render_node(cell.id)) |
|
375 |
else: |
|
376 |
s += '<td class="tree_cell" rowspan="2"><div class="tree_cell">%s</div></td>' % (render_node(cell.id)) |
|
377 |
else: |
|
378 |
s += '<td rowspan="2"> </td>' |
|
379 |
s += '<td class="tree_cell_%d_1"> </td>' % link_cell |
|
380 |
s += '<td class="tree_cell_%d_2"> </td>' % link_cell |
|
381 |
||
382 |
cell = line[-1] |
|
383 |
if cell: |
|
384 |
if cell.id == selected_node: |
|
385 |
s += '<td class="tree_cell" rowspan="2"><div id="selected" class="tree_cell">%s</div></td>' % (render_node(cell.id)) |
|
386 |
else: |
|
387 |
s += '<td class="tree_cell" rowspan="2"><div class="tree_cell">%s</div></td>' % (render_node(cell.id)) |
|
388 |
else: |
|
389 |
s += '<td rowspan="2"> </td>' |
|
390 |
||
391 |
s += '</tr>\n' |
|
392 |
if link_line: |
|
393 |
s += '<tr>' |
|
394 |
for j, link_cell in enumerate(link_line): |
|
395 |
s += '<td class="tree_cell_%d_3"> </td>' % link_cell |
|
396 |
s += '<td class="tree_cell_%d_4"> </td>' % link_cell |
|
397 |
s += '</tr>\n' |
|
398 |
||
399 |
s += '</table>' |
|
400 |
return s |
|
401 |
||
402 |
||
403 |
||
404 |
# traceback formatting ######################################################## |
|
405 |
||
406 |
import traceback |
|
407 |
||
408 |
def rest_traceback(info, exception): |
|
409 |
"""return a ReST formated traceback""" |
|
410 |
res = [u'Traceback\n---------\n::\n'] |
|
411 |
for stackentry in traceback.extract_tb(info[2]): |
|
412 |
res.append(u'\tFile %s, line %s, function %s' % tuple(stackentry[:3])) |
|
413 |
if stackentry[3]: |
|
414 |
res.append(u'\t %s' % stackentry[3].decode('utf-8', 'replace')) |
|
415 |
res.append(u'\n') |
|
416 |
try: |
|
417 |
res.append(u'\t Error: %s\n' % exception) |
|
418 |
except: |
|
419 |
pass |
|
420 |
return u'\n'.join(res) |
|
421 |
||
422 |
||
423 |
def html_traceback(info, exception, title='', |
|
424 |
encoding='ISO-8859-1', body=''): |
|
425 |
""" return an html formatted traceback from python exception infos. |
|
426 |
""" |
|
427 |
tcbk = info[2] |
|
428 |
stacktb = traceback.extract_tb(tcbk) |
|
429 |
strings = [] |
|
430 |
if body: |
|
431 |
strings.append(u'<div class="error_body">') |
|
432 |
# FIXME |
|
433 |
strings.append(body) |
|
434 |
strings.append(u'</div>') |
|
435 |
if title: |
|
436 |
strings.append(u'<h1 class="error">%s</h1>'% html_escape(title)) |
|
437 |
try: |
|
438 |
strings.append(u'<p class="error">%s</p>' % html_escape(str(exception)).replace("\n","<br />")) |
|
439 |
except UnicodeError: |
|
440 |
pass |
|
441 |
strings.append(u'<div class="error_traceback">') |
|
442 |
for index, stackentry in enumerate(stacktb): |
|
443 |
strings.append(u'<b>File</b> <b class="file">%s</b>, <b>line</b> ' |
|
444 |
u'<b class="line">%s</b>, <b>function</b> ' |
|
445 |
u'<b class="function">%s</b>:<br/>'%( |
|
446 |
html_escape(stackentry[0]), stackentry[1], html_escape(stackentry[2]))) |
|
447 |
if stackentry[3]: |
|
448 |
string = html_escape(stackentry[3]).decode('utf-8', 'replace') |
|
449 |
strings.append(u' %s<br/>\n' % (string)) |
|
450 |
# add locals info for each entry |
|
451 |
try: |
|
452 |
local_context = tcbk.tb_frame.f_locals |
|
453 |
html_info = [] |
|
454 |
chars = 0 |
|
455 |
for name, value in local_context.iteritems(): |
|
456 |
value = html_escape(repr(value)) |
|
457 |
info = u'<span class="name">%s</span>=%s, ' % (name, value) |
|
458 |
line_length = len(name) + len(value) |
|
459 |
chars += line_length |
|
460 |
# 150 is the result of *years* of research ;-) (CSS might be helpful here) |
|
461 |
if chars > 150: |
|
462 |
info = u'<br/>' + info |
|
463 |
chars = line_length |
|
464 |
html_info.append(info) |
|
465 |
boxid = 'ctxlevel%d' % index |
|
466 |
strings.append(u'[%s]' % toggle_link(boxid, '+')) |
|
467 |
strings.append(u'<div id="%s" class="pycontext hidden">%s</div>' % |
|
468 |
(boxid, ''.join(html_info))) |
|
469 |
tcbk = tcbk.tb_next |
|
470 |
except Exception: |
|
471 |
pass # doesn't really matter if we have no context info |
|
472 |
strings.append(u'</div>') |
|
473 |
return '\n'.join(strings) |
|
474 |
||
475 |
# csv files / unicode support ################################################# |
|
476 |
||
477 |
class UnicodeCSVWriter: |
|
478 |
"""proxies calls to csv.writer.writerow to be able to deal with unicode""" |
|
479 |
||
480 |
def __init__(self, wfunc, encoding, **kwargs): |
|
481 |
self.writer = csv.writer(self, **kwargs) |
|
482 |
self.wfunc = wfunc |
|
483 |
self.encoding = encoding |
|
484 |
||
485 |
def write(self, data): |
|
486 |
self.wfunc(data) |
|
487 |
||
488 |
def writerow(self, row): |
|
489 |
csvrow = [] |
|
490 |
for elt in row: |
|
491 |
if isinstance(elt, unicode): |
|
492 |
csvrow.append(elt.encode(self.encoding)) |
|
493 |
else: |
|
494 |
csvrow.append(str(elt)) |
|
495 |
self.writer.writerow(csvrow) |
|
496 |
||
497 |
def writerows(self, rows): |
|
498 |
for row in rows: |
|
499 |
self.writerow(row) |
|
500 |
||
501 |
||
502 |
# some decorators ############################################################# |
|
503 |
||
504 |
class limitsize(object): |
|
505 |
def __init__(self, maxsize): |
|
506 |
self.maxsize = maxsize |
|
507 |
||
508 |
def __call__(self, function): |
|
509 |
def newfunc(*args, **kwargs): |
|
510 |
ret = function(*args, **kwargs) |
|
511 |
if isinstance(ret, basestring): |
|
512 |
return ret[:self.maxsize] |
|
513 |
return ret |
|
514 |
return newfunc |
|
515 |
||
516 |
||
517 |
def jsonize(function): |
|
518 |
import simplejson |
|
519 |
def newfunc(*args, **kwargs): |
|
520 |
ret = function(*args, **kwargs) |
|
521 |
if isinstance(ret, decimal.Decimal): |
|
522 |
ret = float(ret) |
|
523 |
elif isinstance(ret, DateTimeType): |
|
524 |
ret = ret.strftime('%Y-%m-%d %H:%M') |
|
525 |
elif isinstance(ret, DateTimeDeltaType): |
|
526 |
ret = ret.seconds |
|
527 |
try: |
|
528 |
return simplejson.dumps(ret) |
|
529 |
except TypeError: |
|
530 |
return simplejson.dumps(repr(ret)) |
|
531 |
return newfunc |
|
532 |
||
533 |
||
534 |
def htmlescape(function): |
|
535 |
def newfunc(*args, **kwargs): |
|
536 |
ret = function(*args, **kwargs) |
|
537 |
assert isinstance(ret, basestring) |
|
538 |
return html_escape(ret) |
|
539 |
return newfunc |