poezio/src/atom_parser.py
2011-07-10 18:15:52 +02:00

48 lines
1.8 KiB
Python

# Copyright 2010-2011 Le Coz Florent <louiz@louiz.org>
#
# This file is part of Poezio.
#
# Poezio is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 3 of the License.
#
# Poezio is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Poezio. If not, see <http://www.gnu.org/licenses/>.
"""
Defines a function returning a dict containing the values from an
atom entry contained in a pubsub entry
"""
ATOM_XMLNS = "http://www.w3.org/2005/Atom"
def parse_atom_entry(pubsub_item):
"""
Takes a pubsub ET.Element item and returns a dict containing
all needed values from the atom entry element.
Returns None if the item does not contain an atom entry.
"""
entry_elem = pubsub_item.find('{%s}entry' % (ATOM_XMLNS,))
if entry_elem is None:
return None
res = {'author':{}}
author_elem = entry_elem.find('{%s}author' % (ATOM_XMLNS,))
if author_elem is not None:
for sub in ('name', 'uri'):
sub_elem = author_elem.find('{%s}%s' % (ATOM_XMLNS, sub,))
if sub_elem is not None:
res['author'][sub] = sub_elem.text
for elem_name in {'title':'text', 'updated':'date', 'published': 'date',
'summary':'text'}:
elem = entry_elem.find('{%s}%s' % (ATOM_XMLNS, elem_name,))
if elem is not None:
res[elem_name] = elem.text
link_elem = entry_elem.find('{%s}link' % (ATOM_XMLNS,))
if link_elem is not None:
res['link_href'] = link_elem.attrib.get('href') or ''
return res