xml_helper.py

   1 #  -*- Mode: python; coding: utf-8; indent-tabs-mode: nil -*- */
   2 #
   3 #  This file is part of systemd.
   4 #
   5 #  Copyright 2012-2013 Zbigniew Jędrzejewski-Szmek
   6 #
   7 #  systemd is free software; you can redistribute it and/or modify it
   8 #  under the terms of the GNU Lesser General Public License as published by
   9 #  the Free Software Foundation; either version 2.1 of the License, or
  10 #  (at your option) any later version.
  11 #
  12 #  systemd is distributed in the hope that it will be useful, but
  13 #  WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15 #  Lesser General Public License for more details.
  16 #
  17 #  You should have received a copy of the GNU Lesser General Public License
  18 #  along with systemd; If not, see <http://www.gnu.org/licenses/>.
  19
  20 try:
  21     from lxml import etree as tree
  22
  23     class CustomResolver(tree.Resolver):
  24         def resolve(self, url, id, context):
  25             if 'custom-entities.ent' in url:
  26                 return self.resolve_filename('man/custom-entities.ent', context)
  27
  28     _parser = tree.XMLParser()
  29     _parser.resolvers.add(CustomResolver())
  30     xml_parse = lambda page: tree.parse(page, _parser)
  31     xml_print = lambda xml: tree.tostring(xml, pretty_print=True,
  32                                           encoding='utf-8')
  33 except ImportError:
  34     import xml.etree.ElementTree as tree
  35     import re as _re
  36     import io as _io
  37
  38     def xml_parse(page):
  39         s = _re.sub(b'&[a-zA-Z0-9_]+;', b'', open(page, 'rb').read())
  40         return tree.parse(_io.BytesIO(s))
  41     xml_print = lambda xml: tree.tostring(xml, encoding='utf-8')