~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to elementtree/ElementTree.py

• browse files at revision 225
• compare with another revision
• download diff
• download tarball
• view history from revision 225

• reverse the comparison (225 to 802)
• stop comparing with revision 802

Show diffs side-by-side

added

removed

elementtree/ElementTree.py

1

1

#

2

2

# ElementTree

3

 

# $Id: ElementTree.py 2326 2005-03-17 07:45:21Z fredrik $

 

3

# $Id: ElementTree.py 1862 2004-06-18 07:31:02Z Fredrik $

4

4

#

5

5

# light-weight XML support for Python 1.5.2 and later.

6

6

#

 

7

# this is a stripped-down version of Secret Labs' effDOM library (part

 

8

# of xmlToolkit).  compared to effDOM, this implementation has:

 

9

#

 

10

# - no support for observers

 

11

# - no html-specific extensions (e.g. entity preload)

 

12

# - no custom entities, doctypes, etc

 

13

# - no accelerator module

 

14

#

7

15

# history:

8

16

# 2001-10-20 fl   created (from various sources)

9

17

# 2001-11-01 fl   return root from parse method

30

38

# 2004-03-28 fl   added XMLID helper

31

39

# 2004-06-02 fl   added default support to findtext

32

40

# 2004-06-08 fl   fixed encoding of non-ascii element/attribute names

33

 

# 2004-08-23 fl   take advantage of post-2.1 expat features

34

 

# 2005-02-01 fl   added iterparse implementation

35

 

# 2005-03-02 fl   fixed iterparse support for pre-2.2 versions

36

41

#

37

 

# Copyright (c) 1999-2005 by Fredrik Lundh.  All rights reserved.

 

42

# Copyright (c) 1999-2004 by Fredrik Lundh.  All rights reserved.

38

43

#

39

44

# fredrik@pythonware.com

40

45

# http://www.pythonware.com

42

47

# --------------------------------------------------------------------

43

48

# The ElementTree toolkit is

44

49

#

45

 

# Copyright (c) 1999-2005 by Fredrik Lundh

 

50

# Copyright (c) 1999-2004 by Fredrik Lundh

46

51

#

47

52

# By obtaining, using, and/or copying this software and/or its

48

53

# associated documentation, you agree that you have read, understood,

73

78

    "dump",

74

79

    "Element", "ElementTree",

75

80

    "fromstring",

76

 

    "iselement", "iterparse",

 

81

    "iselement",

77

82

    "parse",

78

83

    "PI", "ProcessingInstruction",

79

84

    "QName",

138

143

# TODO: add support for custom namespace resolvers/default namespaces

139

144

# TODO: add improved support for incremental parsing

140

145

 

141

 

VERSION = "1.2.6"

 

146

VERSION = "1.2"

142

147

 

143

148

##

144

149

# Internal element class.  This class defines the Element interface,

696

701

                for k, v in xmlns_items:

697

702

                    file.write(" %s=\"%s\"" % (_encode(k, encoding),

698

703

                                               _escape_attrib(v, encoding)))

699

 

            if node.text or len(node):

 

704

            if node.text or node:

700

705

                file.write(">")

701

706

                if node.text:

702

707

                    file.write(_escape_cdata(node.text, encoding))

860

865

    return tree

861

866

 

862

867

##

863

 

# Parses an XML document into an element tree incrementally, and reports

864

 

# what's going on to the user.

865

 

#

866

 

# @param source A filename or file object containing XML data.

867

 

# @param events A list of events to report back.  If omitted, only "end"

868

 

#     events are reported.

869

 

# @return A (event, elem) iterator.

870

 

 

871

 

class iterparse:

872

 

 

873

 

    def __init__(self, source, events=None):

874

 

        if not hasattr(source, "read"):

875

 

            source = open(source, "rb")

876

 

        self._file = source

877

 

        self._events = []

878

 

        self._index = 0

879

 

        self.root = self._root = None

880

 

        self._parser = XMLTreeBuilder()

881

 

        # wire up the parser for event reporting

882

 

        parser = self._parser._parser

883

 

        append = self._events.append

884

 

        if events is None:

885

 

            events = ["end"]

886

 

        for event in events:

887

 

            if event == "start":

888

 

                try:

889

 

                    parser.ordered_attributes = 1

890

 

                    parser.specified_attributes = 1

891

 

                    def handler(tag, attrib_in, event=event, append=append,

892

 

                                start=self._parser._start_list):

893

 

                        append((event, start(tag, attrib_in)))

894

 

                    parser.StartElementHandler = handler

895

 

                except AttributeError:

896

 

                    def handler(tag, attrib_in, event=event, append=append,

897

 

                                start=self._parser._start):

898

 

                        append((event, start(tag, attrib_in)))

899

 

                    parser.StartElementHandler = handler

900

 

            elif event == "end":

901

 

                def handler(tag, event=event, append=append,

902

 

                            end=self._parser._end):

903

 

                    append((event, end(tag)))

904

 

                parser.EndElementHandler = handler

905

 

            elif event == "start-ns":

906

 

                def handler(prefix, uri, event=event, append=append):

907

 

                    try:

908

 

                        uri = _encode(uri, "ascii")

909

 

                    except UnicodeError:

910

 

                        pass

911

 

                    append((event, (prefix or "", uri)))

912

 

                parser.StartNamespaceDeclHandler = handler

913

 

            elif event == "end-ns":

914

 

                def handler(prefix, event=event, append=append):

915

 

                    append((event, None))

916

 

                parser.EndNamespaceDeclHandler = handler

917

 

 

918

 

    def next(self):

919

 

        while 1:

920

 

            try:

921

 

                item = self._events[self._index]

922

 

            except IndexError:

923

 

                if self._parser is None:

924

 

                    self.root = self._root

925

 

                    try:

926

 

                        raise StopIteration

927

 

                    except NameError:

928

 

                        raise IndexError

929

 

                # load event buffer

930

 

                del self._events[:]

931

 

                self._index = 0

932

 

                data = self._file.read(16384)

933

 

                if data:

934

 

                    self._parser.feed(data)

935

 

                else:

936

 

                    self._root = self._parser.close()

937

 

                    self._parser = None

938

 

            else:

939

 

                self._index = self._index + 1

940

 

                return item

941

 

 

942

 

    try:

943

 

        iter

944

 

        def __iter__(self):

945

 

            return self

946

 

    except NameError:

947

 

        def __getitem__(self, index):

948

 

            return self.next()

949

 

 

950

 

##

951

868

# Parses an XML document from a string constant.  This function can

952

869

# be used to embed "XML literals" in Python code.

953

870

#

1108

1025

class XMLTreeBuilder:

1109

1026

 

1110

1027

    def __init__(self, html=0, target=None):

1111

 

        try:

1112

 

            from xml.parsers import expat

1113

 

        except ImportError:

1114

 

            raise ImportError(

1115

 

                "No module named expat; use SimpleXMLTreeBuilder instead"

1116

 

                )

 

1028

        from xml.parsers import expat

1117

1029

        self._parser = parser = expat.ParserCreate(None, "}")

1118

1030

        if target is None:

1119

1031

            target = TreeBuilder()

1120

1032

        self._target = target

1121

1033

        self._names = {} # name memo cache

1122

 

        # callbacks

1123

 

        parser.DefaultHandlerExpand = self._default

 

1034

        parser.DefaultHandler = self._default

1124

1035

        parser.StartElementHandler = self._start

1125

1036

        parser.EndElementHandler = self._end

1126

1037

        parser.CharacterDataHandler = self._data

1127

 

        # let expat do the buffering, if supported

1128

 

        try:

1129

 

            self._parser.buffer_text = 1

1130

 

        except AttributeError:

1131

 

            pass

1132

 

        # use new-style attribute handling, if supported

1133

 

        try:

1134

 

            self._parser.ordered_attributes = 1

1135

 

            self._parser.specified_attributes = 1

1136

 

            parser.StartElementHandler = self._start_list

1137

 

        except AttributeError:

1138

 

            pass

1139

1038

        encoding = None

1140

1039

        if not parser.returns_unicode:

1141

1040

            encoding = "utf-8"

1146

1045

    def _fixtext(self, text):

1147

1046

        # convert text string to ascii, if possible

1148

1047

        try:

1149

 

            return _encode(text, "ascii")

 

1048

            return str(text) # what if the default encoding is changed?

1150

1049

        except UnicodeError:

1151

1050

            return text

1152

1051

 

1169

1068

            attrib[fixname(key)] = self._fixtext(value)

1170

1069

        return self._target.start(tag, attrib)

1171

1070

 

1172

 

    def _start_list(self, tag, attrib_in):

1173

 

        fixname = self._fixname

1174

 

        tag = fixname(tag)

1175

 

        attrib = {}

1176

 

        if attrib_in:

1177

 

            for i in range(0, len(attrib_in), 2):

1178

 

                attrib[fixname(attrib_in[i])] = self._fixtext(attrib_in[i+1])

1179

 

        return self._target.start(tag, attrib)

1180

 

 

1181

1071

    def _data(self, text):

1182

1072

        return self._target.data(self._fixtext(text))

1183

1073

 

• Older »

Loggerhead is a web-based interface for Breezy
Version: 2.0.1