File: XML.py

package info (click to toggle)
ditrack 0.8-1.1
  • links: PTS
  • area: main
  • in suites: squeeze, wheezy
  • size: 516 kB
  • ctags: 645
  • sloc: python: 3,762; makefile: 43
file content (129 lines) | stat: -rw-r--r-- 3,886 bytes parent folder | download | duplicates (3)
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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
#
# XML.py - DITrack XML output support
#
# Copyright (c) 2007 The DITrack Project, www.ditrack.org.
#
# $Id: list.py 2182 2007-10-09 21:23:50Z vss $
# $HeadURL: https://127.0.0.1/ditrack/src/trunk/DITrack/Command/list.py $
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
#  * Redistributions of source code must retain the above copyright notice,
# this list of conditions and the following disclaimer.
#  * Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions and the following disclaimer in the documentation
# and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
#

# XXX: this module is dt-specific, so it should be moved to an appropriate
# place.

import sys
from xml.sax.saxutils import XMLGenerator
from xml.sax.xmlreader import AttributesImpl

import DITrack.dt.globals

# XML format version
XML_FORMAT = "1"

class Writer:
    """
    Low level XML output primitives.
    """
    def __init__(self, out=sys.stdout):
        self._xml = XMLGenerator(out, "utf-8")
        self._xml.startDocument()

    def closetag(self, tag, nl=False, indent=0):
        assert indent >= 0

        self.text(" " * indent)

        self._xml.endElement(tag)
        if nl:
            self.text("\n")

    def finish(self):
        self._xml.endDocument()

    def opentag(self, tag, attrs={}, nl=False, indent=0):
        assert indent >= 0

        self.text(" " * indent)

        self._xml.startElement(tag, AttributesImpl(attrs))
        if nl:
            self.text("\n")

    def tag_enclose(self, tag, attrs, text, nl=False, indent=0):
        assert indent >= 0

        self.text(" " * indent)

        self._xml.startElement(tag, attrs)
        self._xml.characters(text)
        self._xml.endElement(tag)

        if nl:
            self.text("\n")

    def text(self, s):
        self._xml.characters(s)


class Output:
    """
    DITrack-specific XML output producer.
    """

    def __init__(self, command, input):
        """
        Writes the dt XML output header to stdout. COMMAND is a command name
        and INPUT is a list of tuples (TAG, VALUE) to be output as input
        parameters to the command (e.g. enclosed in an '<input>' block).
        """

        self.writer = Writer()
        self.writer.opentag("dt",
            {
                "format":       XML_FORMAT,
                "dt-version":   DITrack.dt.globals.VERSION,
            }
        )
        self.writer.text("\n\n")

        self.writer.opentag("input", { "command": command })

        for k, v in input:
            self.writer.text("\n    ")
            self.writer.tag_enclose(k, {}, v)

        self.writer.text("\n")

        self.writer.closetag("input")
        self.writer.text("\n\n")

        self.writer.opentag("output")
        self.writer.text("\n\n")

    def finish(self):
        self.writer.closetag("output")
        self.writer.text("\n")
        self.writer.closetag("dt")
        self.writer.text("\n")
        self.writer.finish()