xml_helper.py revision 21ac6ff143cc8bebfbd1818af28e8c6f82cd5265
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering# -*- Mode: python; coding: utf-8; indent-tabs-mode: nil -*- */
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering#
12b42c76672a66c2d4ea7212c14f8f1b5a62b78dTom Gundersen# This file is part of systemd.
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering#
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering# Copyright 2012-2013 Zbigniew Jędrzejewski-Szmek
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering#
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering# systemd is free software; you can redistribute it and/or modify it
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering# under the terms of the GNU Lesser General Public License as published by
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering# the Free Software Foundation; either version 2.1 of the License, or
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering# (at your option) any later version.
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering#
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering# systemd is distributed in the hope that it will be useful, but
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering# WITHOUT ANY WARRANTY; without even the implied warranty of
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering# Lesser General Public License for more details.
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering#
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering# You should have received a copy of the GNU Lesser General Public License
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering# along with systemd; If not, see <http://www.gnu.org/licenses/>.
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poetteringfrom lxml import etree as tree
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poetteringclass CustomResolver(tree.Resolver):
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering def resolve(self, url, id, context):
4819ff0358b6317c195fd4b1768e03d09c871070Lennart Poettering if 'custom-entities.ent' in url:
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek return self.resolve_filename('man/custom-entities.ent', context)
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmektry:
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek _parser = tree.XMLParser()
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek _parser.resolvers.add(CustomResolver())
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek def xml_parse(page):
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek doc = tree.parse(page, _parser)
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek doc.xinclude()
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek return doc
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek def xml_print(xml):
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek return tree.tostring(xml, pretty_print=True, encoding='utf-8')
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmekexcept ImportError:
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek import xml.etree.ElementTree as tree
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek import re as _re
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek import io as _io
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek def xml_parse(page):
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek s = _re.sub(b'&[a-zA-Z0-9_]+;', b'', open(page, 'rb').read())
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek return tree.parse(_io.BytesIO(s))
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek xml_print = lambda xml: tree.tostring(xml, encoding='utf-8')
798d3a524ea57aaf40cb53858aaa45ec702f012dZbigniew Jędrzejewski-Szmek