summaryrefslogtreecommitdiffstats
path: root/src/lib/Bcfg2/Client/XML.py
blob: 858479611033dcf2d3347b2133ab3ed0967a182d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
'''XML lib compatibility layer for the Bcfg2 client'''

# library will use lxml, then builtin xml.etree, then ElementTree

try:
    from lxml.etree import Element, SubElement, XML, tostring
    from lxml.etree import XMLSyntaxError as ParseError
    driver = 'lxml'
except ImportError:
    # lxml not available
    from xml.parsers.expat import ExpatError as ParseError
    try:
        import xml.etree.ElementTree
        Element = xml.etree.ElementTree.Element
        SubElement = xml.etree.ElementTree.SubElement
        XML = xml.etree.ElementTree.XML
        def tostring(e, encoding=None, xml_declaration=None):
            return xml.etree.ElementTree.tostring(e, encoding=encoding)
        driver = 'etree-py'
    except ImportError:
        try:
            from elementtree.ElementTree import Element, SubElement, XML, tostring
            driver = 'etree'
            import elementtree.ElementTree
            Element = elementtree.ElementTree.Element
            SubElement = elementtree.ElementTree.SubElement
            XML = elementtree.ElementTree.XML
            def tostring(e, encoding=None, xml_declaration=None):
                return elementtree.ElementTree.tostring(e)

        except ImportError:
            print("Failed to load lxml, xml.etree and elementtree.ElementTree")
            print("Cannot continue")
            raise SystemExit(1)

len([Element, SubElement, XML, tostring, ParseError])