This commit is contained in:
Florent Le Coz 2011-09-24 00:02:09 +02:00
commit c7a0659a1d

View file

@ -174,6 +174,7 @@ colors = {
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
def get_body_from_message_stanza(message): def get_body_from_message_stanza(message):
""" """
Returns a string with xhtml markups converted to Returns a string with xhtml markups converted to
@ -216,6 +217,8 @@ def xhtml_to_poezio_colors(text):
shell = '' shell = ''
rules = css.split(';') rules = css.split(';')
for rule in rules: for rule in rules:
if ':' not in rule:
continue
key, value = rule.split(':', 1) key, value = rule.split(':', 1)
key = key.strip() key = key.strip()
value = value.strip() value = value.strip()
@ -239,13 +242,16 @@ def xhtml_to_poezio_colors(text):
shell += '\x19a' shell += '\x19a'
return shell return shell
def trim(string):
return re.sub(whitespace_re, ' ', string)
log.debug(text) log.debug(text)
xml = ET.fromstring(text) xml = ET.fromstring(text)
message = '' message = ''
for elem in xml.iter(): for elem in xml.iter():
if elem.tag == '{http://www.w3.org/1999/xhtml}a': if elem.tag == '{http://www.w3.org/1999/xhtml}a':
if 'href' in elem.attrib and elem.attrib['href'] != elem.text: if 'href' in elem.attrib and elem.attrib['href'] != elem.text:
message += '\x19u%s\x19o (%s)' % (elem.attrib['href'].strip(), elem.text.strip()) message += '\x19u%s\x19o (%s)' % (trim(elem.attrib['href']), trim(elem.text))
else: else:
message += '\x19u' + elem.text + '\x19o' message += '\x19u' + elem.text + '\x19o'
elif elem.tag == '{http://www.w3.org/1999/xhtml}blockquote': elif elem.tag == '{http://www.w3.org/1999/xhtml}blockquote':
@ -260,7 +266,7 @@ def xhtml_to_poezio_colors(text):
message += '\x19i' message += '\x19i'
elif elem.tag == '{http://www.w3.org/1999/xhtml}img' and 'src' in elem.attrib: elif elem.tag == '{http://www.w3.org/1999/xhtml}img' and 'src' in elem.attrib:
if 'alt' in elem.attrib: if 'alt' in elem.attrib:
message += '%s (%s)' % (elem.attrib['src'].strip(), elem.attrib['alt'].strip()) message += '%s (%s)' % (trim(elem.attrib['src']), trim(elem.attrib['alt']))
else: else:
message += elem.attrib['src'] message += elem.attrib['src']
elif elem.tag == '{http://www.w3.org/1999/xhtml}li': elif elem.tag == '{http://www.w3.org/1999/xhtml}li':
@ -284,7 +290,7 @@ def xhtml_to_poezio_colors(text):
if (elem.text and elem.tag != '{http://www.w3.org/1999/xhtml}a' if (elem.text and elem.tag != '{http://www.w3.org/1999/xhtml}a'
and elem.tag != '{http://www.w3.org/1999/xhtml}br' and elem.tag != '{http://www.w3.org/1999/xhtml}br'
and elem.tag != '{http://www.w3.org/1999/xhtml}img'): and elem.tag != '{http://www.w3.org/1999/xhtml}img'):
message += elem.text.replace('\n', '') message += trim(elem.text)
if ('style' in elem.attrib and elem.tag != '{http://www.w3.org/1999/xhtml}br' if ('style' in elem.attrib and elem.tag != '{http://www.w3.org/1999/xhtml}br'
and elem.tag != '{http://www.w3.org/1999/xhtml}em' and elem.tag != '{http://www.w3.org/1999/xhtml}em'
@ -306,7 +312,7 @@ def xhtml_to_poezio_colors(text):
message += ' [' + elem.attrib['title'] + ']' message += ' [' + elem.attrib['title'] + ']'
if elem.tail: if elem.tail:
message += elem.tail.strip() message += trim(elem.tail)
return message return message