annotate Modules/_elementtree.c @ 36027:cd8b5b31e02a legacy-trunk

copied cElementTree 1.0.4 to Modules
author Fredrik Lundh <fredrik@pythonware.com>
date Wed, 14 Dec 2005 21:45:58 +0000
parents
children 9301a7c0312a
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
36027
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1 /*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2 * ElementTree
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
3 * $Id: /work/modules/celementtree/cElementTree.c 1089 2005-12-14T21:22:30.257830Z Fredrik $
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
4 *
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
5 * elementtree accelerator
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
6 *
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
7 * History:
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
8 * 1999-06-20 fl created (as part of sgmlop)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
9 * 2001-05-29 fl effdom edition
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
10 * 2001-06-05 fl backported to unix; fixed bogus free in clear
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
11 * 2001-07-10 fl added findall helper
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
12 * 2003-02-27 fl elementtree edition (alpha)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
13 * 2004-06-03 fl updates for elementtree 1.2
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
14 * 2005-01-05 fl added universal name cache, Element/SubElement factories
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
15 * 2005-01-06 fl moved python helpers into C module; removed 1.5.2 support
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
16 * 2005-01-07 fl added 2.1 support; work around broken __copy__ in 2.3
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
17 * 2005-01-08 fl added makeelement method; fixed path support
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
18 * 2005-01-10 fl optimized memory usage
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
19 * 2005-01-11 fl first public release (cElementTree 0.8)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
20 * 2005-01-12 fl split element object into base and extras
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
21 * 2005-01-13 fl use tagged pointers for tail/text (cElementTree 0.9)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
22 * 2005-01-17 fl added treebuilder close method
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
23 * 2005-01-17 fl fixed crash in getchildren
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
24 * 2005-01-18 fl removed observer api, added iterparse (cElementTree 0.9.3)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
25 * 2005-01-23 fl revised iterparse api; added namespace event support (0.9.8)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
26 * 2005-01-26 fl added VERSION module property (cElementTree 1.0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
27 * 2005-01-28 fl added remove method (1.0.1)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
28 * 2005-03-01 fl added iselement function; fixed makeelement aliasing (1.0.2)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
29 * 2005-03-13 fl export Comment and ProcessingInstruction/PI helpers
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
30 * 2005-03-26 fl added Comment and PI support to XMLParser
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
31 * 2005-03-27 fl event optimizations; complain about bogus events
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
32 * 2005-08-08 fl fixed read error handling in parse
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
33 * 2005-08-11 fl added runtime test for copy workaround (1.0.3)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
34 * 2005-12-13 fl added expat_capi support (for xml.etree)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
35 *
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
36 * Copyright (c) 1999-2005 by Secret Labs AB. All rights reserved.
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
37 * Copyright (c) 1999-2005 by Fredrik Lundh.
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
38 *
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
39 * info@pythonware.com
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
40 * http://www.pythonware.com
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
41 */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
42
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
43 #include "Python.h"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
44
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
45 #define VERSION "1.0.4"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
46
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
47 /* -------------------------------------------------------------------- */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
48 /* configuration */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
49
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
50 /* Leave defined to include the expat-based XMLParser type */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
51 #define USE_EXPAT
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
52
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
53 /* Define to to all expat calls via pyexpat's embedded expat library */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
54 /* #define USE_PYEXPAT_CAPI */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
55
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
56 /* An element can hold this many children without extra memory
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
57 allocations. */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
58 #define STATIC_CHILDREN 4
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
59
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
60 /* For best performance, chose a value so that 80-90% of all nodes
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
61 have no more than the given number of children. Set this to zero
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
62 to minimize the size of the element structure itself (this only
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
63 helps if you have lots of leaf nodes with attributes). */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
64
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
65 /* Also note that pymalloc always allocates blocks in multiples of
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
66 eight bytes. For the current version of cElementTree, this means
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
67 that the number of children should be an even number, at least on
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
68 32-bit platforms. */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
69
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
70 /* -------------------------------------------------------------------- */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
71
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
72 #if 0
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
73 static int memory = 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
74 #define ALLOC(size, comment)\
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
75 do { memory += size; printf("%8d - %s\n", memory, comment); } while (0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
76 #define RELEASE(size, comment)\
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
77 do { memory -= size; printf("%8d - %s\n", memory, comment); } while (0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
78 #else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
79 #define ALLOC(size, comment)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
80 #define RELEASE(size, comment)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
81 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
82
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
83 /* compiler tweaks */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
84 #if defined(_MSC_VER)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
85 #define LOCAL(type) static __inline type __fastcall
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
86 #else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
87 #define LOCAL(type) static type
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
88 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
89
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
90 /* compatibility macros */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
91 #if (PY_VERSION_HEX < 0x02040000)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
92 #define PyDict_CheckExact PyDict_Check
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
93 #if (PY_VERSION_HEX < 0x02020000)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
94 #define PyList_CheckExact PyList_Check
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
95 #define PyString_CheckExact PyString_Check
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
96 #if (PY_VERSION_HEX >= 0x01060000)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
97 #define Py_USING_UNICODE /* always enabled for 2.0 and 2.1 */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
98 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
99 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
100 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
101
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
102 #if (PY_VERSION_HEX >= 0x02050000)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
103 #define PY_CONST const /* 2.5 adds const to some API:s */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
104 #else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
105 #define PY_CONST
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
106 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
107
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
108 #if !defined(Py_RETURN_NONE)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
109 #define Py_RETURN_NONE return Py_INCREF(Py_None), Py_None
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
110 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
111
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
112 /* macros used to store 'join' flags in string object pointers. note
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
113 that all use of text and tail as object pointers must be wrapped in
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
114 JOIN_OBJ. see comments in the ElementObject definition for more
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
115 info. */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
116 #define JOIN_GET(p) ((Py_uintptr_t) (p) & 1)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
117 #define JOIN_SET(p, flag) ((void*) ((Py_uintptr_t) (JOIN_OBJ(p)) | (flag)))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
118 #define JOIN_OBJ(p) ((PyObject*) ((Py_uintptr_t) (p) & ~1))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
119
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
120 /* glue functions (see the init function for details) */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
121 static PyObject* elementtree_copyelement_obj;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
122 static PyObject* elementtree_deepcopy_obj;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
123 static PyObject* elementtree_getiterator_obj;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
124 static PyObject* elementpath_obj;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
125
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
126 /* helpers */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
127
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
128 LOCAL(PyObject*)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
129 deepcopy(PyObject* object, PyObject* memo)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
130 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
131 /* do a deep copy of the given object */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
132
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
133 PyObject* args;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
134 PyObject* result;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
135
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
136 if (!elementtree_deepcopy_obj) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
137 PyErr_SetString(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
138 PyExc_RuntimeError,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
139 "deepcopy helper not found"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
140 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
141 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
142 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
143
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
144 args = PyTuple_New(2);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
145 Py_INCREF(object); PyTuple_SET_ITEM(args, 0, (PyObject*) object);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
146 Py_INCREF(memo); PyTuple_SET_ITEM(args, 1, (PyObject*) memo);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
147
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
148 result = PyObject_CallObject(elementtree_deepcopy_obj, args);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
149
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
150 Py_DECREF(args);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
151
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
152 return result;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
153 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
154
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
155 LOCAL(PyObject*)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
156 list_join(PyObject* list)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
157 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
158 /* join list elements (destroying the list in the process) */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
159
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
160 PyObject* joiner;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
161 PyObject* function;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
162 PyObject* args;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
163 PyObject* result;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
164
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
165 switch (PyList_GET_SIZE(list)) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
166 case 0:
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
167 Py_DECREF(list);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
168 return PyString_FromString("");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
169 case 1:
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
170 result = PyList_GET_ITEM(list, 0);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
171 Py_INCREF(result);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
172 Py_DECREF(list);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
173 return result;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
174 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
175
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
176 /* two or more elements: slice out a suitable separator from the
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
177 first member, and use that to join the entire list */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
178
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
179 joiner = PySequence_GetSlice(PyList_GET_ITEM(list, 0), 0, 0);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
180 if (!joiner)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
181 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
182
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
183 function = PyObject_GetAttrString(joiner, "join");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
184 if (!function) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
185 Py_DECREF(joiner);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
186 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
187 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
188
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
189 args = PyTuple_New(1);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
190 PyTuple_SET_ITEM(args, 0, list);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
191
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
192 result = PyObject_CallObject(function, args);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
193
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
194 Py_DECREF(args); /* also removes list */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
195 Py_DECREF(function);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
196 Py_DECREF(joiner);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
197
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
198 return result;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
199 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
200
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
201 #if (PY_VERSION_HEX < 0x02020000)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
202 LOCAL(int)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
203 PyDict_Update(PyObject* dict, PyObject* other)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
204 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
205 /* PyDict_Update emulation for 2.1 and earlier */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
206
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
207 PyObject* res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
208
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
209 res = PyObject_CallMethod(dict, "update", "O", other);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
210 if (!res)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
211 return -1;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
212
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
213 Py_DECREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
214 return 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
215 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
216 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
217
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
218 /* -------------------------------------------------------------------- */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
219 /* the element type */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
220
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
221 typedef struct {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
222
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
223 /* attributes (a dictionary object), or None if no attributes */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
224 PyObject* attrib;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
225
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
226 /* child elements */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
227 int length; /* actual number of items */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
228 int allocated; /* allocated items */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
229
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
230 /* this either points to _children or to a malloced buffer */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
231 PyObject* *children;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
232
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
233 PyObject* _children[STATIC_CHILDREN];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
234
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
235 } ElementObjectExtra;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
236
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
237 typedef struct {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
238 PyObject_HEAD
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
239
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
240 /* element tag (a string). */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
241 PyObject* tag;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
242
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
243 /* text before first child. note that this is a tagged pointer;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
244 use JOIN_OBJ to get the object pointer. the join flag is used
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
245 to distinguish lists created by the tree builder from lists
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
246 assigned to the attribute by application code; the former
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
247 should be joined before being returned to the user, the latter
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
248 should be left intact. */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
249 PyObject* text;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
250
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
251 /* text after this element, in parent. note that this is a tagged
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
252 pointer; use JOIN_OBJ to get the object pointer. */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
253 PyObject* tail;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
254
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
255 ElementObjectExtra* extra;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
256
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
257 } ElementObject;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
258
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
259 staticforward PyTypeObject Element_Type;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
260
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
261 #define Element_CheckExact(op) ((op)->ob_type == &Element_Type)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
262
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
263 /* -------------------------------------------------------------------- */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
264 /* element constructor and destructor */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
265
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
266 LOCAL(int)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
267 element_new_extra(ElementObject* self, PyObject* attrib)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
268 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
269 self->extra = PyObject_Malloc(sizeof(ElementObjectExtra));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
270 if (!self->extra)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
271 return -1;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
272
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
273 if (!attrib)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
274 attrib = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
275
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
276 Py_INCREF(attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
277 self->extra->attrib = attrib;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
278
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
279 self->extra->length = 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
280 self->extra->allocated = STATIC_CHILDREN;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
281 self->extra->children = self->extra->_children;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
282
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
283 return 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
284 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
285
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
286 LOCAL(void)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
287 element_dealloc_extra(ElementObject* self)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
288 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
289 int i;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
290
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
291 Py_DECREF(self->extra->attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
292
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
293 for (i = 0; i < self->extra->length; i++)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
294 Py_DECREF(self->extra->children[i]);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
295
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
296 if (self->extra->children != self->extra->_children)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
297 PyObject_Free(self->extra->children);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
298
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
299 PyObject_Free(self->extra);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
300 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
301
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
302 LOCAL(PyObject*)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
303 element_new(PyObject* tag, PyObject* attrib)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
304 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
305 ElementObject* self;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
306
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
307 self = PyObject_New(ElementObject, &Element_Type);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
308 if (self == NULL)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
309 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
310
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
311 /* use None for empty dictionaries */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
312 if (PyDict_CheckExact(attrib) && !PyDict_Size(attrib))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
313 attrib = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
314
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
315 self->extra = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
316
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
317 if (attrib != Py_None) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
318
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
319 if (element_new_extra(self, attrib) < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
320 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
321
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
322 self->extra->length = 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
323 self->extra->allocated = STATIC_CHILDREN;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
324 self->extra->children = self->extra->_children;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
325
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
326 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
327
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
328 Py_INCREF(tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
329 self->tag = tag;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
330
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
331 Py_INCREF(Py_None);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
332 self->text = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
333
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
334 Py_INCREF(Py_None);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
335 self->tail = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
336
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
337 ALLOC(sizeof(ElementObject), "create element");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
338
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
339 return (PyObject*) self;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
340 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
341
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
342 LOCAL(int)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
343 element_resize(ElementObject* self, int extra)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
344 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
345 int size;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
346 PyObject* *children;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
347
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
348 /* make sure self->children can hold the given number of extra
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
349 elements. set an exception and return -1 if allocation failed */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
350
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
351 if (!self->extra)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
352 element_new_extra(self, NULL);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
353
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
354 size = self->extra->length + extra;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
355
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
356 if (size > self->extra->allocated) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
357 /* use Python 2.4's list growth strategy */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
358 size = (size >> 3) + (size < 9 ? 3 : 6) + size;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
359 if (self->extra->children != self->extra->_children) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
360 children = PyObject_Realloc(self->extra->children,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
361 size * sizeof(PyObject*));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
362 if (!children)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
363 goto nomemory;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
364 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
365 children = PyObject_Malloc(size * sizeof(PyObject*));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
366 if (!children)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
367 goto nomemory;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
368 /* copy existing children from static area to malloc buffer */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
369 memcpy(children, self->extra->children,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
370 self->extra->length * sizeof(PyObject*));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
371 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
372 self->extra->children = children;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
373 self->extra->allocated = size;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
374 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
375
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
376 return 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
377
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
378 nomemory:
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
379 PyErr_NoMemory();
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
380 return -1;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
381 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
382
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
383 LOCAL(int)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
384 element_add_subelement(ElementObject* self, PyObject* element)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
385 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
386 /* add a child element to a parent */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
387
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
388 if (element_resize(self, 1) < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
389 return -1;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
390
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
391 Py_INCREF(element);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
392 self->extra->children[self->extra->length] = element;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
393
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
394 self->extra->length++;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
395
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
396 return 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
397 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
398
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
399 LOCAL(PyObject*)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
400 element_get_attrib(ElementObject* self)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
401 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
402 /* return borrowed reference to attrib dictionary */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
403 /* note: this function assumes that the extra section exists */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
404
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
405 PyObject* res = self->extra->attrib;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
406
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
407 if (res == Py_None) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
408 /* create missing dictionary */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
409 res = PyDict_New();
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
410 if (!res)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
411 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
412 self->extra->attrib = res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
413 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
414
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
415 return res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
416 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
417
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
418 LOCAL(PyObject*)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
419 element_get_text(ElementObject* self)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
420 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
421 /* return borrowed reference to text attribute */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
422
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
423 PyObject* res = self->text;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
424
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
425 if (JOIN_GET(res)) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
426 res = JOIN_OBJ(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
427 if (PyList_CheckExact(res)) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
428 res = list_join(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
429 if (!res)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
430 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
431 self->text = res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
432 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
433 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
434
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
435 return res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
436 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
437
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
438 LOCAL(PyObject*)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
439 element_get_tail(ElementObject* self)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
440 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
441 /* return borrowed reference to text attribute */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
442
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
443 PyObject* res = self->tail;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
444
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
445 if (JOIN_GET(res)) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
446 res = JOIN_OBJ(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
447 if (PyList_CheckExact(res)) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
448 res = list_join(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
449 if (!res)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
450 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
451 self->tail = res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
452 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
453 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
454
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
455 return res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
456 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
457
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
458 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
459 element(PyObject* self, PyObject* args, PyObject* kw)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
460 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
461 PyObject* elem;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
462
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
463 PyObject* tag;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
464 PyObject* attrib = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
465 if (!PyArg_ParseTuple(args, "O|O!:Element", &tag,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
466 &PyDict_Type, &attrib))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
467 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
468
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
469 if (attrib || kw) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
470 attrib = (attrib) ? PyDict_Copy(attrib) : PyDict_New();
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
471 if (!attrib)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
472 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
473 if (kw)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
474 PyDict_Update(attrib, kw);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
475 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
476 Py_INCREF(Py_None);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
477 attrib = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
478 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
479
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
480 elem = element_new(tag, attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
481
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
482 Py_DECREF(attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
483
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
484 return elem;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
485 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
486
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
487 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
488 subelement(PyObject* self, PyObject* args, PyObject* kw)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
489 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
490 PyObject* elem;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
491
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
492 ElementObject* parent;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
493 PyObject* tag;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
494 PyObject* attrib = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
495 if (!PyArg_ParseTuple(args, "O!O|O!:SubElement",
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
496 &Element_Type, &parent, &tag,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
497 &PyDict_Type, &attrib))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
498 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
499
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
500 if (attrib || kw) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
501 attrib = (attrib) ? PyDict_Copy(attrib) : PyDict_New();
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
502 if (!attrib)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
503 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
504 if (kw)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
505 PyDict_Update(attrib, kw);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
506 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
507 Py_INCREF(Py_None);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
508 attrib = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
509 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
510
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
511 elem = element_new(tag, attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
512
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
513 Py_DECREF(attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
514
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
515 if (element_add_subelement(parent, elem) < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
516 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
517
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
518 return elem;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
519 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
520
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
521 static void
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
522 element_dealloc(ElementObject* self)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
523 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
524 if (self->extra)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
525 element_dealloc_extra(self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
526
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
527 /* discard attributes */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
528 Py_DECREF(self->tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
529 Py_DECREF(JOIN_OBJ(self->text));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
530 Py_DECREF(JOIN_OBJ(self->tail));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
531
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
532 RELEASE(sizeof(ElementObject), "destroy element");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
533
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
534 PyObject_Del(self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
535 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
536
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
537 /* -------------------------------------------------------------------- */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
538 /* methods (in alphabetical order) */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
539
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
540 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
541 element_append(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
542 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
543 PyObject* element;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
544 if (!PyArg_ParseTuple(args, "O!:append", &Element_Type, &element))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
545 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
546
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
547 if (element_add_subelement(self, element) < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
548 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
549
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
550 Py_RETURN_NONE;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
551 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
552
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
553 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
554 element_clear(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
555 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
556 if (!PyArg_ParseTuple(args, ":clear"))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
557 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
558
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
559 if (self->extra) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
560 element_dealloc_extra(self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
561 self->extra = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
562 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
563
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
564 Py_INCREF(Py_None);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
565 Py_DECREF(JOIN_OBJ(self->text));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
566 self->text = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
567
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
568 Py_INCREF(Py_None);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
569 Py_DECREF(JOIN_OBJ(self->tail));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
570 self->tail = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
571
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
572 Py_RETURN_NONE;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
573 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
574
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
575 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
576 element_copy(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
577 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
578 int i;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
579 ElementObject* element;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
580
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
581 if (!PyArg_ParseTuple(args, ":__copy__"))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
582 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
583
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
584 element = (ElementObject*) element_new(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
585 self->tag, (self->extra) ? self->extra->attrib : Py_None
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
586 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
587 if (!element)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
588 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
589
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
590 Py_DECREF(JOIN_OBJ(element->text));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
591 element->text = self->text;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
592 Py_INCREF(JOIN_OBJ(element->text));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
593
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
594 Py_DECREF(JOIN_OBJ(element->tail));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
595 element->tail = self->tail;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
596 Py_INCREF(JOIN_OBJ(element->tail));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
597
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
598 if (self->extra) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
599
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
600 if (element_resize(element, self->extra->length) < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
601 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
602
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
603 for (i = 0; i < self->extra->length; i++) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
604 Py_INCREF(self->extra->children[i]);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
605 element->extra->children[i] = self->extra->children[i];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
606 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
607
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
608 element->extra->length = self->extra->length;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
609
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
610 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
611
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
612 return (PyObject*) element;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
613 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
614
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
615 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
616 element_deepcopy(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
617 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
618 int i;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
619 ElementObject* element;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
620 PyObject* tag;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
621 PyObject* attrib;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
622 PyObject* text;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
623 PyObject* tail;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
624 PyObject* id;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
625
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
626 PyObject* memo;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
627 if (!PyArg_ParseTuple(args, "O:__deepcopy__", &memo))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
628 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
629
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
630 tag = deepcopy(self->tag, memo);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
631 if (!tag)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
632 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
633
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
634 if (self->extra) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
635 attrib = deepcopy(self->extra->attrib, memo);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
636 if (!attrib) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
637 Py_DECREF(tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
638 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
639 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
640 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
641 Py_INCREF(Py_None);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
642 attrib = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
643 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
644
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
645 element = (ElementObject*) element_new(tag, attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
646
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
647 Py_DECREF(tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
648 Py_DECREF(attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
649
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
650 if (!element)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
651 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
652
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
653 text = deepcopy(JOIN_OBJ(self->text), memo);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
654 if (!text)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
655 goto error;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
656 Py_DECREF(element->text);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
657 element->text = JOIN_SET(text, JOIN_GET(self->text));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
658
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
659 tail = deepcopy(JOIN_OBJ(self->tail), memo);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
660 if (!tail)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
661 goto error;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
662 Py_DECREF(element->tail);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
663 element->tail = JOIN_SET(tail, JOIN_GET(self->tail));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
664
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
665 if (self->extra) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
666
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
667 if (element_resize(element, self->extra->length) < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
668 goto error;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
669
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
670 for (i = 0; i < self->extra->length; i++) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
671 PyObject* child = deepcopy(self->extra->children[i], memo);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
672 if (!child) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
673 element->extra->length = i;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
674 goto error;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
675 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
676 element->extra->children[i] = child;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
677 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
678
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
679 element->extra->length = self->extra->length;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
680
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
681 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
682
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
683 /* add object to memo dictionary (so deepcopy won't visit it again) */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
684 id = PyInt_FromLong((Py_uintptr_t) self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
685
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
686 i = PyDict_SetItem(memo, id, (PyObject*) element);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
687
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
688 Py_DECREF(id);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
689
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
690 if (i < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
691 goto error;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
692
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
693 return (PyObject*) element;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
694
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
695 error:
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
696 Py_DECREF(element);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
697 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
698 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
699
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
700 LOCAL(int)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
701 checkpath(PyObject* tag)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
702 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
703 int i, check = 1;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
704
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
705 /* check if a tag contains an xpath character */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
706
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
707 #define PATHCHAR(ch) (ch == '/' || ch == '*' || ch == '[' || ch == '@')
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
708
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
709 #if defined(Py_USING_UNICODE)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
710 if (PyUnicode_Check(tag)) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
711 Py_UNICODE *p = PyUnicode_AS_UNICODE(tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
712 for (i = 0; i < PyUnicode_GET_SIZE(tag); i++) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
713 if (p[i] == '{')
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
714 check = 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
715 else if (p[i] == '}')
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
716 check = 1;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
717 else if (check && PATHCHAR(p[i]))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
718 return 1;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
719 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
720 return 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
721 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
722 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
723 if (PyString_Check(tag)) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
724 char *p = PyString_AS_STRING(tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
725 for (i = 0; i < PyString_GET_SIZE(tag); i++) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
726 if (p[i] == '{')
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
727 check = 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
728 else if (p[i] == '}')
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
729 check = 1;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
730 else if (check && PATHCHAR(p[i]))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
731 return 1;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
732 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
733 return 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
734 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
735
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
736 return 1; /* unknown type; might be path expression */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
737 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
738
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
739 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
740 element_find(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
741 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
742 int i;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
743
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
744 PyObject* tag;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
745 if (!PyArg_ParseTuple(args, "O:find", &tag))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
746 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
747
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
748 if (checkpath(tag))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
749 return PyObject_CallMethod(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
750 elementpath_obj, "find", "OO", self, tag
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
751 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
752
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
753 if (!self->extra)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
754 Py_RETURN_NONE;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
755
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
756 for (i = 0; i < self->extra->length; i++) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
757 PyObject* item = self->extra->children[i];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
758 if (Element_CheckExact(item) &&
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
759 PyObject_Compare(((ElementObject*)item)->tag, tag) == 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
760 Py_INCREF(item);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
761 return item;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
762 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
763 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
764
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
765 Py_RETURN_NONE;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
766 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
767
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
768 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
769 element_findtext(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
770 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
771 int i;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
772
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
773 PyObject* tag;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
774 PyObject* default_value = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
775 if (!PyArg_ParseTuple(args, "O|O:findtext", &tag, &default_value))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
776 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
777
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
778 if (checkpath(tag))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
779 return PyObject_CallMethod(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
780 elementpath_obj, "findtext", "OOO", self, tag, default_value
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
781 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
782
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
783 if (!self->extra) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
784 Py_INCREF(default_value);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
785 return default_value;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
786 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
787
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
788 for (i = 0; i < self->extra->length; i++) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
789 ElementObject* item = (ElementObject*) self->extra->children[i];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
790 if (Element_CheckExact(item) && !PyObject_Compare(item->tag, tag)) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
791 PyObject* text = element_get_text(item);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
792 if (text == Py_None)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
793 return PyString_FromString("");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
794 Py_INCREF(text);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
795 return text;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
796 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
797 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
798
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
799 Py_INCREF(default_value);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
800 return default_value;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
801 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
802
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
803 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
804 element_findall(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
805 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
806 int i;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
807 PyObject* out;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
808
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
809 PyObject* tag;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
810 if (!PyArg_ParseTuple(args, "O:findall", &tag))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
811 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
812
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
813 if (checkpath(tag))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
814 return PyObject_CallMethod(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
815 elementpath_obj, "findall", "OO", self, tag
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
816 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
817
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
818 out = PyList_New(0);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
819 if (!out)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
820 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
821
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
822 if (!self->extra)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
823 return out;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
824
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
825 for (i = 0; i < self->extra->length; i++) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
826 PyObject* item = self->extra->children[i];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
827 if (Element_CheckExact(item) &&
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
828 PyObject_Compare(((ElementObject*)item)->tag, tag) == 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
829 if (PyList_Append(out, item) < 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
830 Py_DECREF(out);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
831 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
832 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
833 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
834 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
835
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
836 return out;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
837 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
838
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
839 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
840 element_get(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
841 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
842 PyObject* value;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
843
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
844 PyObject* key;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
845 PyObject* default_value = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
846 if (!PyArg_ParseTuple(args, "O|O:get", &key, &default_value))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
847 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
848
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
849 if (!self->extra || self->extra->attrib == Py_None)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
850 value = default_value;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
851 else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
852 value = PyDict_GetItem(self->extra->attrib, key);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
853 if (!value)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
854 value = default_value;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
855 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
856
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
857 Py_INCREF(value);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
858 return value;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
859 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
860
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
861 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
862 element_getchildren(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
863 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
864 int i;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
865 PyObject* list;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
866
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
867 if (!PyArg_ParseTuple(args, ":getchildren"))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
868 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
869
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
870 if (!self->extra)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
871 return PyList_New(0);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
872
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
873 list = PyList_New(self->extra->length);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
874 if (!list)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
875 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
876
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
877 for (i = 0; i < self->extra->length; i++) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
878 PyObject* item = self->extra->children[i];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
879 Py_INCREF(item);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
880 PyList_SET_ITEM(list, i, item);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
881 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
882
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
883 return list;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
884 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
885
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
886 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
887 element_getiterator(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
888 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
889 PyObject* result;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
890
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
891 PyObject* tag = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
892 if (!PyArg_ParseTuple(args, "|O:getiterator", &tag))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
893 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
894
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
895 if (!elementtree_getiterator_obj) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
896 PyErr_SetString(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
897 PyExc_RuntimeError,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
898 "getiterator helper not found"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
899 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
900 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
901 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
902
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
903 args = PyTuple_New(2);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
904 Py_INCREF(self); PyTuple_SET_ITEM(args, 0, (PyObject*) self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
905 Py_INCREF(tag); PyTuple_SET_ITEM(args, 1, (PyObject*) tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
906
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
907 result = PyObject_CallObject(elementtree_getiterator_obj, args);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
908
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
909 Py_DECREF(args);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
910
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
911 return result;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
912 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
913
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
914 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
915 element_getitem(ElementObject* self, int index)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
916 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
917 if (!self->extra || index < 0 || index >= self->extra->length) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
918 PyErr_SetString(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
919 PyExc_IndexError,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
920 "child index out of range"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
921 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
922 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
923 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
924
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
925 Py_INCREF(self->extra->children[index]);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
926 return self->extra->children[index];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
927 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
928
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
929 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
930 element_getslice(ElementObject* self, int start, int end)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
931 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
932 int i;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
933 PyObject* list;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
934
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
935 if (!self->extra)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
936 return PyList_New(0);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
937
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
938 /* standard clamping */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
939 if (start < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
940 start = 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
941 if (end < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
942 end = 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
943 if (end > self->extra->length)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
944 end = self->extra->length;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
945 if (start > end)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
946 start = end;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
947
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
948 list = PyList_New(end - start);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
949 if (!list)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
950 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
951
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
952 for (i = start; i < end; i++) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
953 PyObject* item = self->extra->children[i];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
954 Py_INCREF(item);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
955 PyList_SET_ITEM(list, i - start, item);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
956 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
957
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
958 return list;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
959 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
960
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
961 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
962 element_insert(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
963 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
964 int i;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
965
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
966 int index;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
967 PyObject* element;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
968 if (!PyArg_ParseTuple(args, "iO!:insert", &index,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
969 &Element_Type, &element))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
970 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
971
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
972 if (!self->extra)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
973 element_new_extra(self, NULL);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
974
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
975 if (index < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
976 index = 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
977 if (index > self->extra->length)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
978 index = self->extra->length;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
979
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
980 if (element_resize(self, 1) < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
981 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
982
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
983 for (i = self->extra->length; i > index; i--)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
984 self->extra->children[i] = self->extra->children[i-1];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
985
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
986 Py_INCREF(element);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
987 self->extra->children[index] = element;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
988
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
989 self->extra->length++;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
990
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
991 Py_RETURN_NONE;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
992 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
993
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
994 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
995 element_items(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
996 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
997 if (!PyArg_ParseTuple(args, ":items"))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
998 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
999
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1000 if (!self->extra || self->extra->attrib == Py_None)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1001 return PyList_New(0);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1002
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1003 return PyDict_Items(self->extra->attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1004 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1005
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1006 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1007 element_keys(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1008 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1009 if (!PyArg_ParseTuple(args, ":keys"))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1010 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1011
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1012 if (!self->extra || self->extra->attrib == Py_None)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1013 return PyList_New(0);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1014
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1015 return PyDict_Keys(self->extra->attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1016 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1017
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1018 static int
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1019 element_length(ElementObject* self)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1020 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1021 if (!self->extra)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1022 return 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1023
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1024 return self->extra->length;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1025 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1026
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1027 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1028 element_makeelement(PyObject* self, PyObject* args, PyObject* kw)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1029 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1030 PyObject* elem;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1031
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1032 PyObject* tag;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1033 PyObject* attrib;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1034 if (!PyArg_ParseTuple(args, "OO:makeelement", &tag, &attrib))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1035 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1036
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1037 attrib = PyDict_Copy(attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1038 if (!attrib)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1039 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1040
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1041 elem = element_new(tag, attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1042
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1043 Py_DECREF(attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1044
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1045 return elem;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1046 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1047
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1048 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1049 element_reduce(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1050 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1051 if (!PyArg_ParseTuple(args, ":__reduce__"))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1052 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1053
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1054 /* Hack alert: This method is used to work around a __copy__
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1055 problem on certain 2.3 and 2.4 versions. To save time and
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1056 simplify the code, we create the copy in here, and use a dummy
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1057 copyelement helper to trick the copy module into doing the
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1058 right thing. */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1059
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1060 if (!elementtree_copyelement_obj) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1061 PyErr_SetString(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1062 PyExc_RuntimeError,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1063 "copyelement helper not found"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1064 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1065 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1066 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1067
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1068 return Py_BuildValue(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1069 "O(N)", elementtree_copyelement_obj, element_copy(self, args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1070 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1071 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1072
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1073 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1074 element_remove(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1075 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1076 int i;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1077
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1078 PyObject* element;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1079 if (!PyArg_ParseTuple(args, "O!:remove", &Element_Type, &element))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1080 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1081
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1082 if (!self->extra) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1083 /* element has no children, so raise exception */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1084 PyErr_SetString(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1085 PyExc_ValueError,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1086 "list.remove(x): x not in list"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1087 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1088 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1089 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1090
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1091 for (i = 0; i < self->extra->length; i++) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1092 if (self->extra->children[i] == element)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1093 break;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1094 if (PyObject_Compare(self->extra->children[i], element) == 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1095 break;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1096 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1097
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1098 if (i == self->extra->length) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1099 /* element is not in children, so raise exception */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1100 PyErr_SetString(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1101 PyExc_ValueError,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1102 "list.remove(x): x not in list"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1103 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1104 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1105 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1106
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1107 Py_DECREF(self->extra->children[i]);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1108
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1109 self->extra->length--;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1110
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1111 for (; i < self->extra->length; i++)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1112 self->extra->children[i] = self->extra->children[i+1];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1113
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1114 Py_RETURN_NONE;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1115 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1116
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1117 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1118 element_repr(ElementObject* self)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1119 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1120 PyObject* repr;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1121 char buffer[100];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1122
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1123 repr = PyString_FromString("<Element ");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1124
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1125 PyString_ConcatAndDel(&repr, PyObject_Repr(self->tag));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1126
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1127 sprintf(buffer, " at %p>", self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1128 PyString_ConcatAndDel(&repr, PyString_FromString(buffer));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1129
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1130 return repr;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1131 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1132
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1133 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1134 element_set(ElementObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1135 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1136 PyObject* attrib;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1137
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1138 PyObject* key;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1139 PyObject* value;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1140 if (!PyArg_ParseTuple(args, "OO:set", &key, &value))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1141 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1142
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1143 if (!self->extra)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1144 element_new_extra(self, NULL);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1145
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1146 attrib = element_get_attrib(self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1147 if (!attrib)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1148 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1149
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1150 if (PyDict_SetItem(attrib, key, value) < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1151 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1152
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1153 Py_RETURN_NONE;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1154 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1155
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1156 static int
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1157 element_setslice(ElementObject* self, int start, int end, PyObject* item)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1158 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1159 int i, new, old;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1160 PyObject* recycle = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1161
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1162 if (!self->extra)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1163 element_new_extra(self, NULL);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1164
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1165 /* standard clamping */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1166 if (start < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1167 start = 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1168 if (end < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1169 end = 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1170 if (end > self->extra->length)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1171 end = self->extra->length;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1172 if (start > end)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1173 start = end;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1174
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1175 old = end - start;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1176
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1177 if (item == NULL)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1178 new = 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1179 else if (PyList_CheckExact(item)) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1180 new = PyList_GET_SIZE(item);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1181 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1182 /* FIXME: support arbitrary sequences? */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1183 PyErr_Format(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1184 PyExc_TypeError,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1185 "expected list, not \"%.200s\"", item->ob_type->tp_name
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1186 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1187 return -1;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1188 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1189
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1190 if (old > 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1191 /* to avoid recursive calls to this method (via decref), move
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1192 old items to the recycle bin here, and get rid of them when
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1193 we're done modifying the element */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1194 recycle = PyList_New(old);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1195 for (i = 0; i < old; i++)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1196 PyList_SET_ITEM(recycle, i, self->extra->children[i + start]);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1197 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1198
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1199 if (new < old) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1200 /* delete slice */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1201 for (i = end; i < self->extra->length; i++)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1202 self->extra->children[i + new - old] = self->extra->children[i];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1203 } else if (new > old) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1204 /* insert slice */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1205 if (element_resize(self, new - old) < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1206 return -1;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1207 for (i = self->extra->length-1; i >= end; i--)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1208 self->extra->children[i + new - old] = self->extra->children[i];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1209 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1210
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1211 /* replace the slice */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1212 for (i = 0; i < new; i++) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1213 PyObject* element = PyList_GET_ITEM(item, i);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1214 Py_INCREF(element);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1215 self->extra->children[i + start] = element;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1216 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1217
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1218 self->extra->length += new - old;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1219
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1220 /* discard the recycle bin, and everything in it */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1221 Py_XDECREF(recycle);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1222
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1223 return 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1224 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1225
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1226 static int
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1227 element_setitem(ElementObject* self, int index, PyObject* item)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1228 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1229 int i;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1230 PyObject* old;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1231
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1232 if (!self->extra || index < 0 || index >= self->extra->length) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1233 PyErr_SetString(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1234 PyExc_IndexError,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1235 "child assignment index out of range");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1236 return -1;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1237 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1238
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1239 old = self->extra->children[index];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1240
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1241 if (item) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1242 Py_INCREF(item);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1243 self->extra->children[index] = item;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1244 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1245 self->extra->length--;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1246 for (i = index; i < self->extra->length; i++)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1247 self->extra->children[i] = self->extra->children[i+1];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1248 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1249
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1250 Py_DECREF(old);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1251
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1252 return 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1253 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1254
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1255 static PyMethodDef element_methods[] = {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1256
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1257 {"clear", (PyCFunction) element_clear, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1258
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1259 {"get", (PyCFunction) element_get, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1260 {"set", (PyCFunction) element_set, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1261
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1262 {"find", (PyCFunction) element_find, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1263 {"findtext", (PyCFunction) element_findtext, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1264 {"findall", (PyCFunction) element_findall, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1265
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1266 {"append", (PyCFunction) element_append, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1267 {"insert", (PyCFunction) element_insert, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1268 {"remove", (PyCFunction) element_remove, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1269
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1270 {"getiterator", (PyCFunction) element_getiterator, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1271 {"getchildren", (PyCFunction) element_getchildren, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1272
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1273 {"items", (PyCFunction) element_items, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1274 {"keys", (PyCFunction) element_keys, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1275
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1276 {"makeelement", (PyCFunction) element_makeelement, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1277
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1278 {"__copy__", (PyCFunction) element_copy, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1279 {"__deepcopy__", (PyCFunction) element_deepcopy, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1280
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1281 /* Some 2.3 and 2.4 versions do not handle the __copy__ method on
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1282 C objects correctly, so we have to fake it using a __reduce__-
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1283 based hack (see the element_reduce implementation above for
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1284 details). */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1285
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1286 /* The behaviour has been changed in 2.3.5 and 2.4.1, so we're
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1287 using a runtime test to figure out if we need to fake things
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1288 or now (see the init code below). The following entry is
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1289 enabled only if the hack is needed. */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1290
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1291 {"!__reduce__", (PyCFunction) element_reduce, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1292
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1293 {NULL, NULL}
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1294 };
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1295
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1296 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1297 element_getattr(ElementObject* self, char* name)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1298 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1299 PyObject* res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1300
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1301 res = Py_FindMethod(element_methods, (PyObject*) self, name);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1302 if (res)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1303 return res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1304
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1305 PyErr_Clear();
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1306
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1307 if (strcmp(name, "tag") == 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1308 res = self->tag;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1309 else if (strcmp(name, "text") == 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1310 res = element_get_text(self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1311 else if (strcmp(name, "tail") == 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1312 res = element_get_tail(self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1313 } else if (strcmp(name, "attrib") == 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1314 if (!self->extra)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1315 element_new_extra(self, NULL);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1316 res = element_get_attrib(self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1317 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1318 PyErr_SetString(PyExc_AttributeError, name);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1319 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1320 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1321
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1322 if (!res)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1323 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1324
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1325 Py_INCREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1326 return res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1327 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1328
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1329 static int
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1330 element_setattr(ElementObject* self, const char* name, PyObject* value)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1331 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1332 if (value == NULL) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1333 PyErr_SetString(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1334 PyExc_AttributeError,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1335 "can't delete element attributes"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1336 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1337 return -1;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1338 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1339
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1340 if (strcmp(name, "tag") == 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1341 Py_DECREF(self->tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1342 self->tag = value;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1343 Py_INCREF(self->tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1344 } else if (strcmp(name, "text") == 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1345 Py_DECREF(JOIN_OBJ(self->text));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1346 self->text = value;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1347 Py_INCREF(self->text);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1348 } else if (strcmp(name, "tail") == 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1349 Py_DECREF(JOIN_OBJ(self->tail));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1350 self->tail = value;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1351 Py_INCREF(self->tail);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1352 } else if (strcmp(name, "attrib") == 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1353 if (!self->extra)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1354 element_new_extra(self, NULL);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1355 Py_DECREF(self->extra->attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1356 self->extra->attrib = value;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1357 Py_INCREF(self->extra->attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1358 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1359 PyErr_SetString(PyExc_AttributeError, name);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1360 return -1;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1361 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1362
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1363 return 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1364 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1365
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1366 static PySequenceMethods element_as_sequence = {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1367 (inquiry) element_length,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1368 0, /* sq_concat */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1369 0, /* sq_repeat */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1370 (intargfunc) element_getitem,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1371 (intintargfunc) element_getslice,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1372 (intobjargproc) element_setitem,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1373 (intintobjargproc) element_setslice,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1374 };
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1375
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1376 statichere PyTypeObject Element_Type = {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1377 PyObject_HEAD_INIT(NULL)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1378 0, "Element", sizeof(ElementObject), 0,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1379 /* methods */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1380 (destructor)element_dealloc, /* tp_dealloc */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1381 0, /* tp_print */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1382 (getattrfunc)element_getattr, /* tp_getattr */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1383 (setattrfunc)element_setattr, /* tp_setattr */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1384 0, /* tp_compare */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1385 (reprfunc)element_repr, /* tp_repr */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1386 0, /* tp_as_number */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1387 &element_as_sequence, /* tp_as_sequence */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1388 };
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1389
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1390 /* ==================================================================== */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1391 /* the tree builder type */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1392
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1393 typedef struct {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1394 PyObject_HEAD
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1395
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1396 PyObject* root; /* root node (first created node) */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1397
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1398 ElementObject* this; /* current node */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1399 ElementObject* last; /* most recently created node */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1400
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1401 PyObject* data; /* data collector (string or list), or NULL */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1402
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1403 PyObject* stack; /* element stack */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1404 int index; /* current stack size (0=empty) */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1405
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1406 /* element tracing */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1407 PyObject* events; /* list of events, or NULL if not collecting */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1408 PyObject* start_event_obj; /* event objects (NULL to ignore) */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1409 PyObject* end_event_obj;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1410 PyObject* start_ns_event_obj;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1411 PyObject* end_ns_event_obj;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1412
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1413 } TreeBuilderObject;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1414
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1415 staticforward PyTypeObject TreeBuilder_Type;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1416
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1417 #define TreeBuilder_CheckExact(op) ((op)->ob_type == &TreeBuilder_Type)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1418
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1419 /* -------------------------------------------------------------------- */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1420 /* constructor and destructor */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1421
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1422 LOCAL(PyObject*)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1423 treebuilder_new(void)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1424 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1425 TreeBuilderObject* self;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1426
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1427 self = PyObject_New(TreeBuilderObject, &TreeBuilder_Type);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1428 if (self == NULL)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1429 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1430
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1431 self->root = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1432
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1433 Py_INCREF(Py_None);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1434 self->this = (ElementObject*) Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1435
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1436 Py_INCREF(Py_None);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1437 self->last = (ElementObject*) Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1438
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1439 self->data = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1440
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1441 self->stack = PyList_New(20);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1442 self->index = 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1443
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1444 self->events = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1445 self->start_event_obj = self->end_event_obj = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1446 self->start_ns_event_obj = self->end_ns_event_obj = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1447
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1448 ALLOC(sizeof(TreeBuilderObject), "create treebuilder");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1449
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1450 return (PyObject*) self;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1451 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1452
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1453 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1454 treebuilder(PyObject* _self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1455 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1456 if (!PyArg_ParseTuple(args, ":TreeBuilder"))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1457 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1458
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1459 return treebuilder_new();
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1460 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1461
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1462 static void
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1463 treebuilder_dealloc(TreeBuilderObject* self)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1464 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1465 Py_XDECREF(self->end_ns_event_obj);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1466 Py_XDECREF(self->start_ns_event_obj);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1467 Py_XDECREF(self->end_event_obj);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1468 Py_XDECREF(self->start_event_obj);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1469 Py_XDECREF(self->events);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1470 Py_DECREF(self->stack);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1471 Py_XDECREF(self->data);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1472 Py_DECREF(self->last);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1473 Py_DECREF(self->this);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1474 Py_XDECREF(self->root);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1475
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1476 RELEASE(sizeof(TreeBuilderObject), "destroy treebuilder");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1477
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1478 PyObject_Del(self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1479 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1480
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1481 /* -------------------------------------------------------------------- */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1482 /* handlers */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1483
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1484 LOCAL(PyObject*)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1485 treebuilder_handle_xml(TreeBuilderObject* self, PyObject* encoding,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1486 PyObject* standalone)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1487 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1488 Py_RETURN_NONE;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1489 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1490
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1491 LOCAL(PyObject*)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1492 treebuilder_handle_start(TreeBuilderObject* self, PyObject* tag,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1493 PyObject* attrib)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1494 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1495 PyObject* node;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1496 PyObject* this;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1497
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1498 if (self->data) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1499 if (self->this == self->last) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1500 self->last->text = JOIN_SET(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1501 self->data, PyList_CheckExact(self->data)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1502 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1503 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1504 self->last->tail = JOIN_SET(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1505 self->data, PyList_CheckExact(self->data)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1506 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1507 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1508 self->data = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1509 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1510
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1511 node = element_new(tag, attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1512 if (!node)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1513 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1514
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1515 this = (PyObject*) self->this;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1516
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1517 if (this != Py_None) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1518 if (element_add_subelement((ElementObject*) this, node) < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1519 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1520 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1521 if (self->root) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1522 PyErr_SetString(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1523 PyExc_SyntaxError,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1524 "multiple elements on top level"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1525 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1526 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1527 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1528 Py_INCREF(node);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1529 self->root = node;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1530 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1531
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1532 if (self->index < PyList_GET_SIZE(self->stack)) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1533 if (PyList_SetItem(self->stack, self->index, this) < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1534 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1535 Py_INCREF(this);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1536 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1537 if (PyList_Append(self->stack, this) < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1538 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1539 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1540 self->index++;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1541
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1542 Py_DECREF(this);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1543 Py_INCREF(node);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1544 self->this = (ElementObject*) node;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1545
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1546 Py_DECREF(self->last);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1547 Py_INCREF(node);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1548 self->last = (ElementObject*) node;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1549
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1550 if (self->start_event_obj) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1551 PyObject* res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1552 PyObject* action = self->start_event_obj;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1553 res = PyTuple_New(2);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1554 if (res) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1555 Py_INCREF(action); PyTuple_SET_ITEM(res, 0, (PyObject*) action);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1556 Py_INCREF(node); PyTuple_SET_ITEM(res, 1, (PyObject*) node);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1557 PyList_Append(self->events, res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1558 Py_DECREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1559 } else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1560 PyErr_Clear(); /* FIXME: propagate error */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1561 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1562
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1563 return node;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1564 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1565
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1566 LOCAL(PyObject*)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1567 treebuilder_handle_data(TreeBuilderObject* self, PyObject* data)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1568 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1569 if (!self->data) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1570 /* store the first item as is */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1571 Py_INCREF(data); self->data = data;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1572 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1573 /* more than one item; use a list to collect items */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1574 if (PyString_CheckExact(self->data) && self->data->ob_refcnt == 1 &&
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1575 PyString_CheckExact(data) && PyString_GET_SIZE(data) == 1) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1576 /* expat often generates single character data sections; handle
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1577 the most common case by resizing the existing string... */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1578 int size = PyString_GET_SIZE(self->data);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1579 if (_PyString_Resize(&self->data, size + 1) < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1580 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1581 PyString_AS_STRING(self->data)[size] = PyString_AS_STRING(data)[0];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1582 } else if (PyList_CheckExact(self->data)) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1583 if (PyList_Append(self->data, data) < 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1584 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1585 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1586 PyObject* list = PyList_New(2);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1587 if (!list)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1588 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1589 PyList_SET_ITEM(list, 0, self->data);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1590 Py_INCREF(data); PyList_SET_ITEM(list, 1, data);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1591 self->data = list;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1592 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1593 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1594
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1595 Py_RETURN_NONE;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1596 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1597
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1598 LOCAL(PyObject*)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1599 treebuilder_handle_end(TreeBuilderObject* self, PyObject* tag)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1600 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1601 PyObject* item;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1602
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1603 if (self->data) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1604 if (self->this == self->last) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1605 self->last->text = JOIN_SET(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1606 self->data, PyList_CheckExact(self->data)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1607 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1608 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1609 self->last->tail = JOIN_SET(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1610 self->data, PyList_CheckExact(self->data)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1611 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1612 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1613 self->data = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1614 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1615
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1616 if (self->index == 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1617 PyErr_SetString(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1618 PyExc_IndexError,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1619 "pop from empty stack"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1620 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1621 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1622 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1623
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1624 self->index--;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1625
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1626 item = PyList_GET_ITEM(self->stack, self->index);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1627 Py_INCREF(item);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1628
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1629 Py_DECREF(self->last);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1630
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1631 self->last = (ElementObject*) self->this;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1632 self->this = (ElementObject*) item;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1633
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1634 if (self->end_event_obj) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1635 PyObject* res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1636 PyObject* action = self->end_event_obj;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1637 PyObject* node = (PyObject*) self->last;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1638 res = PyTuple_New(2);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1639 if (res) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1640 Py_INCREF(action); PyTuple_SET_ITEM(res, 0, (PyObject*) action);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1641 Py_INCREF(node); PyTuple_SET_ITEM(res, 1, (PyObject*) node);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1642 PyList_Append(self->events, res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1643 Py_DECREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1644 } else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1645 PyErr_Clear(); /* FIXME: propagate error */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1646 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1647
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1648 Py_INCREF(self->last);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1649 return (PyObject*) self->last;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1650 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1651
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1652 LOCAL(void)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1653 treebuilder_handle_namespace(TreeBuilderObject* self, int start,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1654 const char* prefix, const char *uri)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1655 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1656 PyObject* res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1657 PyObject* action;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1658 PyObject* parcel;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1659
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1660 if (!self->events)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1661 return;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1662
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1663 if (start) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1664 if (!self->start_ns_event_obj)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1665 return;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1666 action = self->start_ns_event_obj;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1667 /* FIXME: prefix and uri use utf-8 encoding! */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1668 parcel = Py_BuildValue("ss", (prefix) ? prefix : "", uri);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1669 if (!parcel)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1670 return;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1671 Py_INCREF(action);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1672 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1673 if (!self->end_ns_event_obj)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1674 return;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1675 action = self->end_ns_event_obj;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1676 Py_INCREF(action);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1677 parcel = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1678 Py_INCREF(parcel);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1679 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1680
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1681 res = PyTuple_New(2);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1682
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1683 if (res) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1684 PyTuple_SET_ITEM(res, 0, action);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1685 PyTuple_SET_ITEM(res, 1, parcel);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1686 PyList_Append(self->events, res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1687 Py_DECREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1688 } else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1689 PyErr_Clear(); /* FIXME: propagate error */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1690 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1691
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1692 /* -------------------------------------------------------------------- */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1693 /* methods (in alphabetical order) */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1694
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1695 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1696 treebuilder_data(TreeBuilderObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1697 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1698 PyObject* data;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1699 if (!PyArg_ParseTuple(args, "O:data", &data))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1700 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1701
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1702 return treebuilder_handle_data(self, data);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1703 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1704
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1705 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1706 treebuilder_end(TreeBuilderObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1707 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1708 PyObject* tag;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1709 if (!PyArg_ParseTuple(args, "O:end", &tag))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1710 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1711
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1712 return treebuilder_handle_end(self, tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1713 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1714
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1715 LOCAL(PyObject*)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1716 treebuilder_done(TreeBuilderObject* self)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1717 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1718 PyObject* res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1719
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1720 /* FIXME: check stack size? */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1721
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1722 if (self->root)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1723 res = self->root;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1724 else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1725 res = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1726
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1727 Py_INCREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1728 return res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1729 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1730
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1731 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1732 treebuilder_close(TreeBuilderObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1733 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1734 if (!PyArg_ParseTuple(args, ":close"))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1735 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1736
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1737 return treebuilder_done(self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1738 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1739
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1740 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1741 treebuilder_start(TreeBuilderObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1742 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1743 PyObject* tag;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1744 PyObject* attrib = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1745 if (!PyArg_ParseTuple(args, "O|O:start", &tag, &attrib))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1746 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1747
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1748 return treebuilder_handle_start(self, tag, attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1749 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1750
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1751 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1752 treebuilder_xml(TreeBuilderObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1753 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1754 PyObject* encoding;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1755 PyObject* standalone;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1756 if (!PyArg_ParseTuple(args, "OO:xml", &encoding, &standalone))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1757 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1758
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1759 return treebuilder_handle_xml(self, encoding, standalone);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1760 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1761
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1762 static PyMethodDef treebuilder_methods[] = {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1763 {"data", (PyCFunction) treebuilder_data, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1764 {"start", (PyCFunction) treebuilder_start, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1765 {"end", (PyCFunction) treebuilder_end, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1766 {"xml", (PyCFunction) treebuilder_xml, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1767 {"close", (PyCFunction) treebuilder_close, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1768 {NULL, NULL}
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1769 };
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1770
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1771 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1772 treebuilder_getattr(TreeBuilderObject* self, char* name)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1773 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1774 return Py_FindMethod(treebuilder_methods, (PyObject*) self, name);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1775 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1776
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1777 statichere PyTypeObject TreeBuilder_Type = {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1778 PyObject_HEAD_INIT(NULL)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1779 0, "TreeBuilder", sizeof(TreeBuilderObject), 0,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1780 /* methods */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1781 (destructor)treebuilder_dealloc, /* tp_dealloc */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1782 0, /* tp_print */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1783 (getattrfunc)treebuilder_getattr, /* tp_getattr */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1784 };
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1785
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1786 /* ==================================================================== */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1787 /* the expat interface */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1788
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1789 #if defined(USE_EXPAT)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1790
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1791 #include "expat.h"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1792
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1793 #if defined(USE_PYEXPAT_CAPI)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1794 #include "pyexpat.h"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1795 static struct PyExpat_CAPI* expat_capi;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1796 #define EXPAT(func) (expat_capi->func)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1797 #else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1798 #define EXPAT(func) (XML_##func)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1799 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1800
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1801 typedef struct {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1802 PyObject_HEAD
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1803
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1804 XML_Parser parser;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1805
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1806 PyObject* target;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1807 PyObject* entity;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1808
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1809 PyObject* names;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1810
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1811 PyObject* handle_xml;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1812 PyObject* handle_start;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1813 PyObject* handle_data;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1814 PyObject* handle_end;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1815
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1816 PyObject* handle_comment;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1817 PyObject* handle_pi;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1818
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1819 } XMLParserObject;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1820
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1821 staticforward PyTypeObject XMLParser_Type;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1822
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1823 /* helpers */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1824
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1825 #if defined(Py_USING_UNICODE)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1826 LOCAL(int)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1827 checkstring(const char* string, int size)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1828 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1829 int i;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1830
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1831 /* check if an 8-bit string contains UTF-8 characters */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1832 for (i = 0; i < size; i++)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1833 if (string[i] & 0x80)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1834 return 1;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1835
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1836 return 0;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1837 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1838 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1839
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1840 LOCAL(PyObject*)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1841 makestring(const char* string, int size)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1842 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1843 /* convert a UTF-8 string to either a 7-bit ascii string or a
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1844 Unicode string */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1845
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1846 #if defined(Py_USING_UNICODE)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1847 if (checkstring(string, size))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1848 return PyUnicode_DecodeUTF8(string, size, "strict");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1849 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1850
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1851 return PyString_FromStringAndSize(string, size);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1852 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1853
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1854 LOCAL(PyObject*)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1855 makeuniversal(XMLParserObject* self, const char* string)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1856 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1857 /* convert a UTF-8 tag/attribute name from the expat parser
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1858 to a universal name string */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1859
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1860 int size = strlen(string);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1861 PyObject* key;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1862 PyObject* value;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1863
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1864 /* look the 'raw' name up in the names dictionary */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1865 key = PyString_FromStringAndSize(string, size);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1866 if (!key)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1867 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1868
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1869 value = PyDict_GetItem(self->names, key);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1870
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1871 if (value) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1872 Py_INCREF(value);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1873 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1874 /* new name. convert to universal name, and decode as
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1875 necessary */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1876
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1877 PyObject* tag;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1878 char* p;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1879 int i;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1880
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1881 /* look for namespace separator */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1882 for (i = 0; i < size; i++)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1883 if (string[i] == '}')
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1884 break;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1885 if (i != size) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1886 /* convert to universal name */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1887 tag = PyString_FromStringAndSize(NULL, size+1);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1888 p = PyString_AS_STRING(tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1889 p[0] = '{';
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1890 memcpy(p+1, string, size);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1891 size++;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1892 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1893 /* plain name; use key as tag */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1894 Py_INCREF(key);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1895 tag = key;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1896 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1897
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1898 /* decode universal name */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1899 #if defined(Py_USING_UNICODE)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1900 /* inline makestring, to avoid duplicating the source string if
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1901 it's not an utf-8 string */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1902 p = PyString_AS_STRING(tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1903 if (checkstring(p, size)) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1904 value = PyUnicode_DecodeUTF8(p, size, "strict");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1905 Py_DECREF(tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1906 if (!value) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1907 Py_DECREF(key);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1908 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1909 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1910 } else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1911 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1912 value = tag; /* use tag as is */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1913
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1914 /* add to names dictionary */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1915 if (PyDict_SetItem(self->names, key, value) < 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1916 Py_DECREF(key);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1917 Py_DECREF(value);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1918 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1919 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1920 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1921
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1922 Py_DECREF(key);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1923 return value;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1924 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1925
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1926 /* -------------------------------------------------------------------- */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1927 /* handlers */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1928
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1929 static void
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1930 expat_default_handler(XMLParserObject* self, const XML_Char* data_in,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1931 int data_len)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1932 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1933 PyObject* key;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1934 PyObject* value;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1935 PyObject* res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1936
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1937 if (data_len < 2 || data_in[0] != '&')
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1938 return;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1939
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1940 key = makestring(data_in + 1, data_len - 2);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1941 if (!key)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1942 return;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1943
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1944 value = PyDict_GetItem(self->entity, key);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1945
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1946 if (value) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1947 if (TreeBuilder_CheckExact(self->target))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1948 res = treebuilder_handle_data(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1949 (TreeBuilderObject*) self->target, value
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1950 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1951 else if (self->handle_data)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1952 res = PyObject_CallFunction(self->handle_data, "O", value);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1953 else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1954 res = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1955 Py_DECREF(value);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1956 Py_XDECREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1957 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1958 PyErr_Format(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1959 PyExc_SyntaxError, "undefined entity &%s;: line %d, column %d",
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1960 PyString_AS_STRING(key),
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1961 EXPAT(GetErrorLineNumber)(self->parser),
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1962 EXPAT(GetErrorColumnNumber)(self->parser)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1963 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1964 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1965
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1966 Py_DECREF(key);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1967 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1968
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1969 static void
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1970 expat_start_handler(XMLParserObject* self, const XML_Char* tag_in,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1971 const XML_Char **attrib_in)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1972 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1973 PyObject* res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1974 PyObject* tag;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1975 PyObject* attrib;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1976 int ok;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1977
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1978 /* tag name */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1979 tag = makeuniversal(self, tag_in);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1980 if (!tag)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1981 return; /* parser will look for errors */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1982
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1983 /* attributes */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1984 if (attrib_in[0]) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1985 attrib = PyDict_New();
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1986 if (!attrib)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1987 return;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1988 while (attrib_in[0] && attrib_in[1]) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1989 PyObject* key = makeuniversal(self, attrib_in[0]);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1990 PyObject* value = makestring(attrib_in[1], strlen(attrib_in[1]));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1991 if (!key || !value) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1992 Py_XDECREF(value);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1993 Py_XDECREF(key);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1994 Py_DECREF(attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1995 return;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1996 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1997 ok = PyDict_SetItem(attrib, key, value);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1998 Py_DECREF(value);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
1999 Py_DECREF(key);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2000 if (ok < 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2001 Py_DECREF(attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2002 return;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2003 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2004 attrib_in += 2;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2005 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2006 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2007 Py_INCREF(Py_None);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2008 attrib = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2009 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2010
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2011 if (TreeBuilder_CheckExact(self->target))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2012 /* shortcut */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2013 res = treebuilder_handle_start((TreeBuilderObject*) self->target,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2014 tag, attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2015 else if (self->handle_start)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2016 res = PyObject_CallFunction(self->handle_start, "OO", tag, attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2017 else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2018 res = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2019
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2020 Py_DECREF(tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2021 Py_DECREF(attrib);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2022
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2023 Py_XDECREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2024 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2025
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2026 static void
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2027 expat_data_handler(XMLParserObject* self, const XML_Char* data_in,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2028 int data_len)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2029 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2030 PyObject* data;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2031 PyObject* res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2032
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2033 data = makestring(data_in, data_len);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2034
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2035 if (TreeBuilder_CheckExact(self->target))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2036 /* shortcut */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2037 res = treebuilder_handle_data((TreeBuilderObject*) self->target, data);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2038 else if (self->handle_data)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2039 res = PyObject_CallFunction(self->handle_data, "O", data);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2040 else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2041 res = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2042
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2043 Py_DECREF(data);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2044
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2045 Py_XDECREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2046 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2047
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2048 static void
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2049 expat_end_handler(XMLParserObject* self, const XML_Char* tag_in)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2050 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2051 PyObject* tag;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2052 PyObject* res = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2053
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2054 if (TreeBuilder_CheckExact(self->target))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2055 /* shortcut */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2056 /* the standard tree builder doesn't look at the end tag */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2057 res = treebuilder_handle_end(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2058 (TreeBuilderObject*) self->target, Py_None
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2059 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2060 else if (self->handle_end) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2061 tag = makeuniversal(self, tag_in);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2062 if (tag) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2063 res = PyObject_CallFunction(self->handle_end, "O", tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2064 Py_DECREF(tag);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2065 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2066 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2067
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2068 Py_XDECREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2069 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2070
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2071 static void
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2072 expat_start_ns_handler(XMLParserObject* self, const XML_Char* prefix,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2073 const XML_Char *uri)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2074 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2075 treebuilder_handle_namespace(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2076 (TreeBuilderObject*) self->target, 1, prefix, uri
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2077 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2078 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2079
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2080 static void
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2081 expat_end_ns_handler(XMLParserObject* self, const XML_Char* prefix_in)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2082 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2083 treebuilder_handle_namespace(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2084 (TreeBuilderObject*) self->target, 0, NULL, NULL
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2085 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2086 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2087
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2088 static void
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2089 expat_comment_handler(XMLParserObject* self, const XML_Char* comment_in)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2090 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2091 PyObject* comment;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2092 PyObject* res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2093
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2094 if (self->handle_comment) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2095 comment = makestring(comment_in, strlen(comment_in));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2096 if (comment) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2097 res = PyObject_CallFunction(self->handle_comment, "O", comment);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2098 Py_XDECREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2099 Py_DECREF(comment);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2100 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2101 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2102 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2103
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2104 static void
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2105 expat_pi_handler(XMLParserObject* self, const XML_Char* target_in,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2106 const XML_Char* data_in)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2107 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2108 PyObject* target;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2109 PyObject* data;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2110 PyObject* res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2111
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2112 if (self->handle_pi) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2113 target = makestring(target_in, strlen(target_in));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2114 data = makestring(data_in, strlen(data_in));
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2115 if (target && data) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2116 res = PyObject_CallFunction(self->handle_pi, "OO", target, data);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2117 Py_XDECREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2118 Py_DECREF(data);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2119 Py_DECREF(target);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2120 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2121 Py_XDECREF(data);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2122 Py_XDECREF(target);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2123 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2124 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2125 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2126
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2127 /* -------------------------------------------------------------------- */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2128 /* constructor and destructor */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2129
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2130 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2131 xmlparser(PyObject* _self, PyObject* args, PyObject* kw)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2132 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2133 XMLParserObject* self;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2134 /* FIXME: does this need to be static? */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2135 static XML_Memory_Handling_Suite memory_handler;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2136
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2137 PyObject* target = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2138 char* encoding = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2139 static PY_CONST char* kwlist[] = { "target", "encoding", NULL };
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2140 if (!PyArg_ParseTupleAndKeywords(args, kw, "|Oz:XMLParser", kwlist,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2141 &target, &encoding))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2142 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2143
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2144 #if defined(USE_PYEXPAT_CAPI)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2145 if (!expat_capi) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2146 PyErr_SetString(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2147 PyExc_RuntimeError, "cannot load dispatch table from pyexpat"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2148 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2149 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2150 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2151 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2152
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2153 self = PyObject_New(XMLParserObject, &XMLParser_Type);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2154 if (self == NULL)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2155 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2156
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2157 self->entity = PyDict_New();
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2158 if (!self->entity) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2159 PyObject_Del(self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2160 return NULL; /* FIXME: cleanup on error */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2161 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2162
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2163 self->names = PyDict_New();
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2164 if (!self->names) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2165 PyObject_Del(self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2166 return NULL; /* FIXME: cleanup on error */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2167 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2168
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2169 memory_handler.malloc_fcn = PyObject_Malloc;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2170 memory_handler.realloc_fcn = PyObject_Realloc;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2171 memory_handler.free_fcn = PyObject_Free;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2172
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2173 self->parser = EXPAT(ParserCreate_MM)(encoding, &memory_handler, "}");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2174 if (!self->parser) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2175 PyErr_NoMemory();
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2176 return NULL; /* FIXME: cleanup on error */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2177 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2178
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2179 /* setup target handlers */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2180 if (!target) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2181 target = treebuilder_new();
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2182 if (!target) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2183 PyObject_Del(self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2184 return NULL; /* FIXME: cleanup on error */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2185 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2186 } else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2187 Py_INCREF(target);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2188 self->target = target;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2189
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2190 self->handle_xml = PyObject_GetAttrString(target, "xml");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2191 self->handle_start = PyObject_GetAttrString(target, "start");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2192 self->handle_data = PyObject_GetAttrString(target, "data");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2193 self->handle_end = PyObject_GetAttrString(target, "end");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2194 self->handle_comment = PyObject_GetAttrString(target, "comment");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2195 self->handle_pi = PyObject_GetAttrString(target, "pi");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2196
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2197 PyErr_Clear();
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2198
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2199 /* configure parser */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2200 EXPAT(SetUserData)(self->parser, self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2201 EXPAT(SetElementHandler)(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2202 self->parser,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2203 (XML_StartElementHandler) expat_start_handler,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2204 (XML_EndElementHandler) expat_end_handler
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2205 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2206 EXPAT(SetDefaultHandlerExpand)(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2207 self->parser,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2208 (XML_DefaultHandler) expat_default_handler
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2209 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2210 EXPAT(SetCharacterDataHandler)(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2211 self->parser,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2212 (XML_CharacterDataHandler) expat_data_handler
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2213 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2214 if (self->handle_comment)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2215 EXPAT(SetCommentHandler)(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2216 self->parser,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2217 (XML_CommentHandler) expat_comment_handler
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2218 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2219 if (self->handle_pi)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2220 EXPAT(SetProcessingInstructionHandler)(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2221 self->parser,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2222 (XML_ProcessingInstructionHandler) expat_pi_handler
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2223 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2224
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2225 ALLOC(sizeof(XMLParserObject), "create expatparser");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2226
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2227 return (PyObject*) self;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2228 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2229
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2230 static void
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2231 xmlparser_dealloc(XMLParserObject* self)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2232 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2233 EXPAT(ParserFree)(self->parser);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2234
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2235 Py_XDECREF(self->handle_pi);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2236 Py_XDECREF(self->handle_comment);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2237 Py_XDECREF(self->handle_end);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2238 Py_XDECREF(self->handle_data);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2239 Py_XDECREF(self->handle_start);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2240 Py_XDECREF(self->handle_xml);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2241
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2242 Py_DECREF(self->target);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2243 Py_DECREF(self->entity);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2244 Py_DECREF(self->names);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2245
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2246 RELEASE(sizeof(XMLParserObject), "destroy expatparser");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2247
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2248 PyObject_Del(self);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2249 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2250
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2251 /* -------------------------------------------------------------------- */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2252 /* methods (in alphabetical order) */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2253
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2254 LOCAL(PyObject*)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2255 expat_parse(XMLParserObject* self, char* data, int data_len, int final)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2256 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2257 int ok;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2258
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2259 ok = EXPAT(Parse)(self->parser, data, data_len, final);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2260
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2261 if (PyErr_Occurred())
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2262 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2263
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2264 if (!ok) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2265 PyErr_Format(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2266 PyExc_SyntaxError, "%s: line %d, column %d",
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2267 EXPAT(ErrorString)(EXPAT(GetErrorCode)(self->parser)),
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2268 EXPAT(GetErrorLineNumber)(self->parser),
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2269 EXPAT(GetErrorColumnNumber)(self->parser)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2270 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2271 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2272 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2273
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2274 Py_RETURN_NONE;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2275 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2276
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2277 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2278 xmlparser_close(XMLParserObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2279 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2280 /* end feeding data to parser */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2281
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2282 PyObject* res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2283 if (!PyArg_ParseTuple(args, ":close"))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2284 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2285
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2286 res = expat_parse(self, "", 0, 1);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2287
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2288 if (res && TreeBuilder_CheckExact(self->target)) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2289 Py_DECREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2290 return treebuilder_done((TreeBuilderObject*) self->target);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2291 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2292
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2293 return res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2294 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2295
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2296 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2297 xmlparser_feed(XMLParserObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2298 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2299 /* feed data to parser */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2300
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2301 char* data;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2302 int data_len;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2303 if (!PyArg_ParseTuple(args, "s#:feed", &data, &data_len))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2304 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2305
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2306 return expat_parse(self, data, data_len, 0);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2307 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2308
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2309 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2310 xmlparser_parse(XMLParserObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2311 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2312 /* (internal) parse until end of input stream */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2313
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2314 PyObject* reader;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2315 PyObject* buffer;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2316 PyObject* res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2317
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2318 PyObject* fileobj;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2319 if (!PyArg_ParseTuple(args, "O:_parse", &fileobj))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2320 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2321
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2322 reader = PyObject_GetAttrString(fileobj, "read");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2323 if (!reader)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2324 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2325
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2326 /* read from open file object */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2327 for (;;) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2328
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2329 buffer = PyObject_CallFunction(reader, "i", 64*1024);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2330
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2331 if (!buffer) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2332 /* read failed (e.g. due to KeyboardInterrupt) */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2333 Py_DECREF(reader);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2334 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2335 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2336
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2337 if (!PyString_CheckExact(buffer) || PyString_GET_SIZE(buffer) == 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2338 Py_DECREF(buffer);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2339 break;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2340 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2341
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2342 res = expat_parse(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2343 self, PyString_AS_STRING(buffer), PyString_GET_SIZE(buffer), 0
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2344 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2345
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2346 Py_DECREF(buffer);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2347
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2348 if (!res) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2349 Py_DECREF(reader);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2350 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2351 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2352 Py_DECREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2353
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2354 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2355
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2356 Py_DECREF(reader);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2357
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2358 res = expat_parse(self, "", 0, 1);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2359
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2360 if (res && TreeBuilder_CheckExact(self->target)) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2361 Py_DECREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2362 return treebuilder_done((TreeBuilderObject*) self->target);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2363 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2364
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2365 return res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2366 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2367
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2368 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2369 xmlparser_setevents(XMLParserObject* self, PyObject* args)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2370 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2371 /* activate element event reporting */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2372
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2373 int i;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2374 TreeBuilderObject* target;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2375
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2376 PyObject* events; /* event collector */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2377 PyObject* event_set = Py_None;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2378 if (!PyArg_ParseTuple(args, "O!|O:_setevents", &PyList_Type, &events,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2379 &event_set))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2380 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2381
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2382 if (!TreeBuilder_CheckExact(self->target)) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2383 PyErr_SetString(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2384 PyExc_TypeError,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2385 "event handling only supported for cElementTree.Treebuilder "
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2386 "targets"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2387 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2388 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2389 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2390
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2391 target = (TreeBuilderObject*) self->target;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2392
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2393 Py_INCREF(events);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2394 Py_XDECREF(target->events);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2395 target->events = events;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2396
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2397 /* clear out existing events */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2398 Py_XDECREF(target->start_event_obj); target->start_event_obj = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2399 Py_XDECREF(target->end_event_obj); target->end_event_obj = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2400 Py_XDECREF(target->start_ns_event_obj); target->start_ns_event_obj = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2401 Py_XDECREF(target->end_ns_event_obj); target->end_ns_event_obj = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2402
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2403 if (event_set == Py_None) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2404 /* default is "end" only */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2405 target->end_event_obj = PyString_FromString("end");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2406 Py_RETURN_NONE;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2407 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2408
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2409 if (!PyTuple_Check(event_set)) /* FIXME: handle arbitrary sequences */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2410 goto error;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2411
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2412 for (i = 0; i < PyTuple_GET_SIZE(event_set); i++) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2413 PyObject* item = PyTuple_GET_ITEM(event_set, i);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2414 char* event;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2415 if (!PyString_Check(item))
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2416 goto error;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2417 event = PyString_AS_STRING(item);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2418 if (strcmp(event, "start") == 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2419 Py_INCREF(item);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2420 target->start_event_obj = item;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2421 } else if (strcmp(event, "end") == 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2422 Py_INCREF(item);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2423 Py_XDECREF(target->end_event_obj);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2424 target->end_event_obj = item;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2425 } else if (strcmp(event, "start-ns") == 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2426 Py_INCREF(item);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2427 Py_XDECREF(target->start_ns_event_obj);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2428 target->start_ns_event_obj = item;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2429 EXPAT(SetNamespaceDeclHandler)(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2430 self->parser,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2431 (XML_StartNamespaceDeclHandler) expat_start_ns_handler,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2432 (XML_EndNamespaceDeclHandler) expat_end_ns_handler
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2433 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2434 } else if (strcmp(event, "end-ns") == 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2435 Py_INCREF(item);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2436 Py_XDECREF(target->end_ns_event_obj);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2437 target->end_ns_event_obj = item;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2438 EXPAT(SetNamespaceDeclHandler)(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2439 self->parser,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2440 (XML_StartNamespaceDeclHandler) expat_start_ns_handler,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2441 (XML_EndNamespaceDeclHandler) expat_end_ns_handler
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2442 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2443 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2444 PyErr_Format(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2445 PyExc_ValueError,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2446 "unknown event '%s'", event
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2447 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2448 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2449 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2450 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2451
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2452 Py_RETURN_NONE;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2453
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2454 error:
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2455 PyErr_SetString(
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2456 PyExc_TypeError,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2457 "invalid event tuple"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2458 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2459 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2460 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2461
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2462 static PyMethodDef xmlparser_methods[] = {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2463 {"feed", (PyCFunction) xmlparser_feed, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2464 {"close", (PyCFunction) xmlparser_close, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2465 {"_parse", (PyCFunction) xmlparser_parse, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2466 {"_setevents", (PyCFunction) xmlparser_setevents, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2467 {NULL, NULL}
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2468 };
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2469
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2470 static PyObject*
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2471 xmlparser_getattr(XMLParserObject* self, char* name)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2472 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2473 PyObject* res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2474
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2475 res = Py_FindMethod(xmlparser_methods, (PyObject*) self, name);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2476 if (res)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2477 return res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2478
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2479 PyErr_Clear();
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2480
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2481 if (strcmp(name, "entity") == 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2482 res = self->entity;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2483 else if (strcmp(name, "target") == 0)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2484 res = self->target;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2485 else if (strcmp(name, "version") == 0) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2486 char buffer[100];
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2487 sprintf(buffer, "Expat %d.%d.%d", XML_MAJOR_VERSION,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2488 XML_MINOR_VERSION, XML_MICRO_VERSION);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2489 return PyString_FromString(buffer);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2490 } else {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2491 PyErr_SetString(PyExc_AttributeError, name);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2492 return NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2493 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2494
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2495 Py_INCREF(res);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2496 return res;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2497 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2498
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2499 statichere PyTypeObject XMLParser_Type = {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2500 PyObject_HEAD_INIT(NULL)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2501 0, "XMLParser", sizeof(XMLParserObject), 0,
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2502 /* methods */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2503 (destructor)xmlparser_dealloc, /* tp_dealloc */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2504 0, /* tp_print */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2505 (getattrfunc)xmlparser_getattr, /* tp_getattr */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2506 };
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2507
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2508 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2509
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2510 /* ==================================================================== */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2511 /* python module interface */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2512
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2513 static PyMethodDef _functions[] = {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2514 {"Element", (PyCFunction) element, METH_VARARGS|METH_KEYWORDS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2515 {"SubElement", (PyCFunction) subelement, METH_VARARGS|METH_KEYWORDS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2516 {"TreeBuilder", (PyCFunction) treebuilder, METH_VARARGS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2517 #if defined(USE_EXPAT)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2518 {"XMLParser", (PyCFunction) xmlparser, METH_VARARGS|METH_KEYWORDS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2519 {"XMLTreeBuilder", (PyCFunction) xmlparser, METH_VARARGS|METH_KEYWORDS},
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2520 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2521 {NULL, NULL}
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2522 };
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2523
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2524 DL_EXPORT(void)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2525 initcElementTree(void)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2526 {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2527 PyObject* m;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2528 PyObject* g;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2529 char* bootstrap;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2530 #if defined(USE_PYEXPAT_CAPI)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2531 struct PyExpat_CAPI* capi;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2532 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2533
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2534 /* Patch object type */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2535 Element_Type.ob_type = TreeBuilder_Type.ob_type = &PyType_Type;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2536 #if defined(USE_EXPAT)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2537 XMLParser_Type.ob_type = &PyType_Type;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2538 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2539
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2540 m = Py_InitModule("cElementTree", _functions);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2541
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2542 /* python glue code */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2543
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2544 g = PyDict_New();
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2545
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2546 PyDict_SetItemString(g, "__builtins__", PyEval_GetBuiltins());
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2547
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2548 bootstrap = (
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2549
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2550 #if (PY_VERSION_HEX >= 0x02020000 && PY_VERSION_HEX < 0x02030000)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2551 "from __future__ import generators\n" /* enable yield under 2.2 */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2552 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2553
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2554 "from copy import copy, deepcopy\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2555
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2556 "try:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2557 " from elementtree import ElementTree\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2558 "except ImportError:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2559 " import ElementTree\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2560 "ET = ElementTree\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2561 "del ElementTree\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2562
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2563 "import cElementTree\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2564
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2565 "try:\n" /* check if copy works as is */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2566 " copy(cElementTree.Element('x'))\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2567 "except:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2568 " def copyelement(elem):\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2569 " return elem\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2570
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2571 "def Comment(text=None):\n" /* public */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2572 " element = cElementTree.Element(ET.Comment)\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2573 " element.text = text\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2574 " return element\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2575 "cElementTree.Comment = Comment\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2576
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2577 "class ElementTree(ET.ElementTree):\n" /* public */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2578 " def parse(self, source, parser=None):\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2579 " if not hasattr(source, 'read'):\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2580 " source = open(source, 'rb')\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2581 " if parser is not None:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2582 " while 1:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2583 " data = source.read(65536)\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2584 " if not data:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2585 " break\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2586 " parser.feed(data)\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2587 " self._root = parser.close()\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2588 " else:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2589 " parser = cElementTree.XMLParser()\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2590 " self._root = parser._parse(source)\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2591 " return self._root\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2592 "cElementTree.ElementTree = ElementTree\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2593
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2594 "def getiterator(node, tag=None):\n" /* helper */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2595 " if tag == '*':\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2596 " tag = None\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2597 #if (PY_VERSION_HEX < 0x02020000)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2598 " nodes = []\n" /* 2.1 doesn't have yield */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2599 " if tag is None or node.tag == tag:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2600 " nodes.append(node)\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2601 " for node in node:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2602 " nodes.extend(getiterator(node, tag))\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2603 " return nodes\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2604 #else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2605 " if tag is None or node.tag == tag:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2606 " yield node\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2607 " for node in node:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2608 " for node in getiterator(node, tag):\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2609 " yield node\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2610 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2611
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2612 "def parse(source, parser=None):\n" /* public */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2613 " tree = ElementTree()\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2614 " tree.parse(source, parser)\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2615 " return tree\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2616 "cElementTree.parse = parse\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2617
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2618 #if (PY_VERSION_HEX < 0x02020000)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2619 "if hasattr(ET, 'iterparse'):\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2620 " cElementTree.iterparse = ET.iterparse\n" /* delegate on 2.1 */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2621 #else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2622 "class iterparse(object):\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2623 " root = None\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2624 " def __init__(self, file, events=None):\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2625 " if not hasattr(file, 'read'):\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2626 " file = open(file, 'rb')\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2627 " self._file = file\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2628 " self._events = events\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2629 " def __iter__(self):\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2630 " events = []\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2631 " b = cElementTree.TreeBuilder()\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2632 " p = cElementTree.XMLParser(b)\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2633 " p._setevents(events, self._events)\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2634 " while 1:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2635 " data = self._file.read(16384)\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2636 " if not data:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2637 " break\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2638 " p.feed(data)\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2639 " for event in events:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2640 " yield event\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2641 " del events[:]\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2642 " root = p.close()\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2643 " for event in events:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2644 " yield event\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2645 " self.root = root\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2646 "cElementTree.iterparse = iterparse\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2647 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2648
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2649 "def PI(target, text=None):\n" /* public */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2650 " element = cElementTree.Element(ET.ProcessingInstruction)\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2651 " element.text = target\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2652 " if text:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2653 " element.text = element.text + ' ' + text\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2654 " return element\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2655
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2656 " elem = cElementTree.Element(ET.PI)\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2657 " elem.text = text\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2658 " return elem\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2659 "cElementTree.PI = cElementTree.ProcessingInstruction = PI\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2660
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2661 "def XML(text):\n" /* public */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2662 " parser = cElementTree.XMLParser()\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2663 " parser.feed(text)\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2664 " return parser.close()\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2665 "cElementTree.XML = cElementTree.fromstring = XML\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2666
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2667 "def XMLID(text):\n" /* public */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2668 " tree = XML(text)\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2669 " ids = {}\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2670 " for elem in tree.getiterator():\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2671 " id = elem.get('id')\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2672 " if id:\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2673 " ids[id] = elem\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2674 " return tree, ids\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2675 "cElementTree.XMLID = XMLID\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2676
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2677 "cElementTree.dump = ET.dump\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2678 "cElementTree.ElementPath = ElementPath = ET.ElementPath\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2679 "cElementTree.iselement = ET.iselement\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2680 "cElementTree.QName = ET.QName\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2681 "cElementTree.tostring = ET.tostring\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2682 "cElementTree.VERSION = '" VERSION "'\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2683 "cElementTree.__version__ = '" VERSION "'\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2684 "cElementTree.XMLParserError = SyntaxError\n"
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2685
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2686 );
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2687
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2688 PyRun_String(bootstrap, Py_file_input, g, NULL);
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2689
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2690 elementpath_obj = PyDict_GetItemString(g, "ElementPath");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2691
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2692 elementtree_copyelement_obj = PyDict_GetItemString(g, "copyelement");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2693 if (elementtree_copyelement_obj) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2694 /* reduce hack needed; enable reduce method */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2695 PyMethodDef* mp;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2696 for (mp = element_methods; mp->ml_name; mp++)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2697 if (mp->ml_meth == (PyCFunction) element_reduce) {
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2698 mp->ml_name = "__reduce__";
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2699 break;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2700 }
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2701 } else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2702 PyErr_Clear();
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2703 elementtree_deepcopy_obj = PyDict_GetItemString(g, "deepcopy");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2704 elementtree_getiterator_obj = PyDict_GetItemString(g, "getiterator");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2705
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2706 #if defined(USE_PYEXPAT_CAPI)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2707 /* link against pyexpat, if possible */
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2708 capi = PyCObject_Import("pyexpat", "expat_CAPI");
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2709 if (capi &&
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2710 strcmp(capi->magic, PyExpat_CAPI_MAGIC) == 0 &&
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2711 capi->size <= sizeof(*expat_capi) &&
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2712 capi->MAJOR_VERSION == XML_MAJOR_VERSION &&
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2713 capi->MINOR_VERSION == XML_MINOR_VERSION &&
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2714 capi->MICRO_VERSION == XML_MICRO_VERSION)
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2715 expat_capi = capi;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2716 else
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2717 expat_capi = NULL;
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2718 #endif
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2719
cd8b5b31e02a copied cElementTree 1.0.4 to Modules
Fredrik Lundh <fredrik@pythonware.com>
parents:
diff changeset
2720 }