annotate markup/path.py @ 43:e93de542929b

Trac example: fixes for alternative links.
author cmlenz
date Mon, 03 Jul 2006 22:03:21 +0000
parents fec9f4897415
children 33c2702cf6da
rev   line source
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
1 # -*- coding: utf-8 -*-
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
2 #
27
b8456279c444 * Fix the boilerplate in the Python source files.
cmlenz
parents: 26
diff changeset
3 # Copyright (C) 2006 Christopher Lenz
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
4 # All rights reserved.
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
5 #
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
6 # This software is licensed as described in the file COPYING, which
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
7 # you should have received as part of this distribution. The terms
27
b8456279c444 * Fix the boilerplate in the Python source files.
cmlenz
parents: 26
diff changeset
8 # are also available at http://markup.cmlenz.net/wiki/License.
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
9 #
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
10 # This software consists of voluntary contributions made by many
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
11 # individuals. For the exact contribution history, see the revision
27
b8456279c444 * Fix the boilerplate in the Python source files.
cmlenz
parents: 26
diff changeset
12 # history and logs, available at http://markup.cmlenz.net/log/.
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
13
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
14 """Basic support for evaluating XPath expressions against streams."""
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
15
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
16 import re
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
17
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
18 from markup.core import QName, Stream
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
19
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
20 __all__ = ['Path']
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
21
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
22
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
23 class Path(object):
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
24 """Implements basic XPath support on streams.
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
25
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
26 Instances of this class represent a "compiled" XPath expression, and provide
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
27 methods for testing the path against a stream, as well as extracting a
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
28 substream matching that path.
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
29 """
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
30 _TOKEN_RE = re.compile('(::|\.\.|\(\)|[/.:\[\]\(\)@=!])|'
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
31 '([^/:\[\]\(\)@=!\s]+)|'
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
32 '\s+')
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
33 _QUOTES = (("'", "'"), ('"', '"'))
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
34
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
35 def __init__(self, text):
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
36 """Create the path object from a string.
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
37
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
38 @param text: the path expression
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
39 """
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
40 self.source = text
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
41
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
42 steps = []
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
43 cur_op = ''
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
44 cur_tag = ''
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
45 in_predicate = False
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
46 for op, tag in self._TOKEN_RE.findall(text):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
47 if op:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
48 if op == '[':
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
49 in_predicate = True
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
50 elif op == ']':
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
51 in_predicate = False
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
52 elif op.startswith('('):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
53 if cur_tag == 'text':
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
54 steps[-1] = (False, self._FunctionText(), [])
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
55 else:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
56 raise NotImplementedError('XPath function "%s" not '
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
57 'supported' % cur_tag)
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
58 elif op == '.':
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
59 steps.append([False, self._CurrentElement(), []])
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
60 else:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
61 cur_op += op
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
62 cur_tag = ''
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
63 else:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
64 closure = cur_op in ('', '//')
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
65 if cur_op == '@':
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
66 if tag == '*':
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
67 node_test = self._AnyAttribute()
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
68 else:
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
69 node_test = self._AttributeByName(tag)
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
70 else:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
71 if tag == '*':
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
72 node_test = self._AnyChildElement()
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
73 elif in_predicate:
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
74 if len(tag) > 1 and (tag[0], tag[-1]) in self._QUOTES:
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
75 node_test = self._LiteralString(tag[1:-1])
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
76 if cur_op == '=':
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
77 node_test = self._OperatorEq(steps[-1][2][-1],
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
78 node_test)
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
79 steps[-1][2].pop()
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
80 elif cur_op == '!=':
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
81 node_test = self._OperatorNeq(steps[-1][2][-1],
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
82 node_test)
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
83 steps[-1][2].pop()
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
84 else:
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
85 node_test = self._ChildElementByName(tag)
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
86 if in_predicate:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
87 steps[-1][2].append(node_test)
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
88 else:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
89 steps.append([closure, node_test, []])
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
90 cur_op = ''
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
91 cur_tag = tag
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
92
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
93 self.steps = []
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
94 for step in steps:
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
95 self.steps.append(tuple(step))
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
96
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
97 def __repr__(self):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
98 return '<%s "%s">' % (self.__class__.__name__, self.source)
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
99
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
100 def select(self, stream):
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
101 """Returns a substream of the given stream that matches the path.
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
102
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
103 If there are no matches, this method returns an empty stream.
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
104
33
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
105 >>> from markup.input import XML
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
106 >>> xml = XML('<root><elem><child>Text</child></elem></root>')
37
224b0b41d1da Moved some of the tests for the strip directive to a new unittest test case to not clutter up the documentation.
cmlenz
parents: 33
diff changeset
107
33
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
108 >>> print Path('child').select(xml)
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
109 <child>Text</child>
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
110
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
111 >>> print Path('child/text()').select(xml)
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
112 Text
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
113
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
114 @param stream: the stream to select from
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
115 @return: the substream matching the path, or an empty stream
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
116 """
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
117 stream = iter(stream)
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
118 def _generate():
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
119 test = self.test()
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
120 for kind, data, pos in stream:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
121 result = test(kind, data, pos)
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
122 if result is True:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
123 yield kind, data, pos
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
124 depth = 1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
125 while depth > 0:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
126 ev = stream.next()
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
127 depth += {Stream.START: 1, Stream.END: -1}.get(ev[0], 0)
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
128 yield ev
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
129 test(*ev)
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
130 elif result:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
131 yield result
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
132 return Stream(_generate())
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
133
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
134 def test(self, ignore_context=False):
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
135 """Returns a function that can be used to track whether the path matches
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
136 a specific stream event.
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
137
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
138 The function returned expects the positional arguments `kind`, `data`,
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
139 and `pos`, i.e. basically an unpacked stream event. If the path matches
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
140 the event, the function returns the match (for example, a `START` or
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
141 `TEXT` event.) Otherwise, it returns `None` or `False`.
33
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
142
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
143 >>> from markup.input import XML
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
144 >>> xml = XML('<root><elem><child id="1"/></elem><child id="2"/></root>')
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
145 >>> test = Path('child').test()
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
146 >>> for kind, data, pos in xml:
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
147 ... if test(kind, data, pos):
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
148 ... print kind, data
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
149 START (u'child', [(u'id', u'1')])
0e1fc0211416 Add doctests for path module.
cmlenz
parents: 27
diff changeset
150 START (u'child', [(u'id', u'2')])
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
151 """
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
152 stack = [0] # stack of cursors into the location path
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
153
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
154 def _test(kind, data, pos):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
155 if not stack:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
156 return False
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
157
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
158 if kind is Stream.END:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
159 stack.pop()
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
160 return None
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
161
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
162 if kind is Stream.START:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
163 stack.append(stack[-1])
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
164
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
165 matched = False
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
166 closure, node_test, predicates = self.steps[stack[-1]]
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
167
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
168 matched = node_test(kind, data, pos)
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
169 if matched and predicates:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
170 for predicate in predicates:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
171 if not predicate(kind, data, pos):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
172 matched = None
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
173 break
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
174
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
175 if matched:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
176 if stack[-1] == len(self.steps) - 1:
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
177 if ignore_context or len(stack) > 2 \
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
178 or node_test.axis != 'child':
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
179 return matched
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
180 else:
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
181 stack[-1] += 1
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
182
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
183 elif kind is Stream.START and not closure:
24
547e36f7ec94 Cosmetic (mostly whitespace) changes.
cmlenz
parents: 23
diff changeset
184 # If this step is not a closure, it cannot be matched until the
547e36f7ec94 Cosmetic (mostly whitespace) changes.
cmlenz
parents: 23
diff changeset
185 # current element is closed... so we need to move the cursor
547e36f7ec94 Cosmetic (mostly whitespace) changes.
cmlenz
parents: 23
diff changeset
186 # back to the last closure and retest that against the current
547e36f7ec94 Cosmetic (mostly whitespace) changes.
cmlenz
parents: 23
diff changeset
187 # element
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
188 closures = [step for step in self.steps[:stack[-1]] if step[0]]
25
c4201b794ab0 Oops. Fix typo in [25].
cmlenz
parents: 24
diff changeset
189 closures.reverse()
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
190 for closure, node_test, predicates in closures:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
191 stack[-1] -= 1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
192 if closure:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
193 matched = node_test(kind, data, pos)
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
194 if matched:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
195 stack[-1] += 1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
196 break
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
197
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
198 return None
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
199
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
200 return _test
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
201
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
202 class _NodeTest(object):
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
203 """Abstract node test."""
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
204 axis = None
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
205 def __repr__(self):
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
206 return '<%s>' % self.__class__.__name__
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
207
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
208 class _CurrentElement(_NodeTest):
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
209 """Node test that matches the context node."""
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
210 axis = 'self'
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
211 def __call__(self, kind, *_):
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
212 if kind is Stream.START:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
213 return True
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
214 return None
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
215
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
216 class _AnyChildElement(_NodeTest):
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
217 """Node test that matches any child element."""
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
218 axis = 'child'
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
219 def __call__(self, kind, *_):
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
220 if kind is Stream.START:
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
221 return True
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
222 return None
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
223
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
224 class _ChildElementByName(_NodeTest):
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
225 """Node test that matches a child element with a specific tag name."""
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
226 axis = 'child'
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
227 def __init__(self, name):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
228 self.name = QName(name)
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
229 def __call__(self, kind, data, _):
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
230 if kind is Stream.START:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
231 return data[0].localname == self.name
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
232 return None
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
233 def __repr__(self):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
234 return '<%s "%s">' % (self.__class__.__name__, self.name)
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
235
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
236 class _AnyAttribute(_NodeTest):
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
237 """Node test that matches any attribute."""
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
238 axis = 'attribute'
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
239 def __call__(self, kind, data, pos):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
240 if kind is Stream.START:
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
241 text = ''.join([val for _, val in data[1]])
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
242 if text:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
243 return Stream.TEXT, text, pos
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
244 return None
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
245 return None
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
246
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
247 class _AttributeByName(_NodeTest):
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
248 """Node test that matches an attribute with a specific name."""
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
249 axis = 'attribute'
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
250 def __init__(self, name):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
251 self.name = QName(name)
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
252 def __call__(self, kind, data, pos):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
253 if kind is Stream.START:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
254 if self.name in data[1]:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
255 return Stream.TEXT, data[1].get(self.name), pos
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
256 return None
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
257 return None
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
258 def __repr__(self):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
259 return '<%s "%s">' % (self.__class__.__name__, self.name)
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
260
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
261 class _Function(_NodeTest):
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
262 """Abstract node test representing a function."""
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
263
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
264 class _FunctionText(_Function):
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
265 """Function that returns text content."""
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
266 def __call__(self, kind, data, pos):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
267 if kind is Stream.TEXT:
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
268 return kind, data, pos
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
269 return None
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
270
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
271 class _LiteralString(_NodeTest):
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
272 """Always returns a literal string."""
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
273 def __init__(self, value):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
274 self.value = value
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
275 def __call__(self, *_):
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
276 return Stream.TEXT, self.value, (-1, -1)
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
277
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
278 class _OperatorEq(_NodeTest):
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
279 """Equality comparison operator."""
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
280 def __init__(self, lval, rval):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
281 self.lval = lval
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
282 self.rval = rval
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
283 def __call__(self, kind, data, pos):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
284 lval = self.lval(kind, data, pos)
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
285 rval = self.rval(kind, data, pos)
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
286 return (lval and lval[1]) == (rval and rval[1])
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
287 def __repr__(self):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
288 return '<%s %r = %r>' % (self.__class__.__name__, self.lval,
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
289 self.rval)
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
290
38
fec9f4897415 Fix for #2 (incorrect context node in path expressions). Still some paths that produce incorrect results, but the common case seems to work now.
cmlenz
parents: 37
diff changeset
291 class _OperatorNeq(_NodeTest):
26
039fc5b87405 * Split out the XPath tests into a separate `unittest`-based file.
cmlenz
parents: 25
diff changeset
292 """Inequality comparison operator."""
1
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
293 def __init__(self, lval, rval):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
294 self.lval = lval
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
295 self.rval = rval
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
296 def __call__(self, kind, data, pos):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
297 lval = self.lval(kind, data, pos)
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
298 rval = self.rval(kind, data, pos)
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
299 return (lval and lval[1]) != (rval and rval[1])
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
300 def __repr__(self):
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
301 return '<%s %r != %r>' % (self.__class__.__name__, self.lval,
821114ec4f69 Initial import.
cmlenz
parents:
diff changeset
302 self.rval)
Copyright (C) 2012-2017 Edgewall Software