Mercurial > p > roundup > code

annotate roundup/cgi/ZTUtils/Iterator.py @ 1233:69bf0d381fd7

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Zope Collector fixes. - merge Zope Collector #538 fix from ZPT CVS trunk (path expressions with a non-path final alternate no longer try to call a value returned by that alternate) - merge Zope Collector #573 fix from ZPT CVS trunk - added 1000-user/issue benchmark test
author Richard Jones <richard@users.sourceforge.net>
date Thu, 26 Sep 2002 21:54:18 +0000
parents b9988e118055
children fc52d57c6c3e
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1049
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
1 ##############################################################################
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
2 #
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
3 # Copyright (c) 2001 Zope Corporation and Contributors. All Rights Reserved.
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
4 #
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
5 # This software is subject to the provisions of the Zope Public License,
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
6 # Version 2.0 (ZPL). A copy of the ZPL should accompany this distribution.
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
7 # THIS SOFTWARE IS PROVIDED "AS IS" AND ANY AND ALL EXPRESS OR IMPLIED
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
8 # WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
9 # WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
10 # FOR A PARTICULAR PURPOSE
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
11 #
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
12 ##############################################################################
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
13 __doc__='''Iterator class
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
14
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
15 Unlike the builtin iterators of Python 2.2+, these classes are
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
16 designed to maintain information about the state of an iteration.
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
17 The Iterator() function accepts either a sequence or a Python
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
18 iterator. The next() method fetches the next item, and returns
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
19 true if it succeeds.
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
20
1233
69bf0d381fd7 Zope Collector fixes.
Richard Jones <richard@users.sourceforge.net>
parents: 1049
diff changeset
21 $Id: Iterator.py,v 1.2 2002-09-26 21:54:17 richard Exp $'''
69bf0d381fd7 Zope Collector fixes.
Richard Jones <richard@users.sourceforge.net>
parents: 1049
diff changeset
22 __version__='$Revision: 1.2 $'[11:-2]
1049
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
23
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
24 import string
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
25
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
26 class Iterator:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
27 '''Simple Iterator class'''
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
28
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
29 __allow_access_to_unprotected_subobjects__ = 1
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
30
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
31 nextIndex = 0
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
32 def __init__(self, seq):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
33 self.seq = seq
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
34 for inner in seqInner, iterInner:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
35 if inner._supports(seq):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
36 self._inner = inner
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
37 self._prep_next = inner.prep_next
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
38 return
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
39 raise TypeError, "Iterator does not support %s" % `seq`
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
40
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
41 def __getattr__(self, name):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
42 try:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
43 inner = getattr(self._inner, 'it_' + name)
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
44 except AttributeError:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
45 raise AttributeError, name
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
46 return inner(self)
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
47
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
48 def next(self):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
49 if not (hasattr(self, '_next') or self._prep_next(self)):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
50 return 0
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
51 self.index = i = self.nextIndex
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
52 self.nextIndex = i+1
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
53 self._advance(self)
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
54 return 1
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
55
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
56 def _advance(self, it):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
57 self.item = self._next
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
58 del self._next
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
59 del self.end
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
60 self._advance = self._inner.advance
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
61 self.start = 1
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
62
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
63 def number(self): return self.nextIndex
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
64
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
65 def even(self): return not self.index % 2
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
66
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
67 def odd(self): return self.index % 2
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
68
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
69 def letter(self, base=ord('a'), radix=26):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
70 index = self.index
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
71 s = ''
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
72 while 1:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
73 index, off = divmod(index, radix)
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
74 s = chr(base + off) + s
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
75 if not index: return s
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
76
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
77 def Letter(self):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
78 return self.letter(base=ord('A'))
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
79
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
80 def Roman(self, rnvalues=(
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
81 (1000,'M'),(900,'CM'),(500,'D'),(400,'CD'),
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
82 (100,'C'),(90,'XC'),(50,'L'),(40,'XL'),
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
83 (10,'X'),(9,'IX'),(5,'V'),(4,'IV'),(1,'I')) ):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
84 n = self.index + 1
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
85 s = ''
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
86 for v, r in rnvalues:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
87 rct, n = divmod(n, v)
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
88 s = s + r * rct
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
89 return s
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
90
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
91 def roman(self, lower=string.lower):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
92 return lower(self.Roman())
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
93
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
94 def first(self, name=None):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
95 if self.start: return 1
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
96 return not self.same_part(name, self._last, self.item)
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
97
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
98 def last(self, name=None):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
99 if self.end: return 1
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
100 return not self.same_part(name, self.item, self._next)
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
101
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
102 def same_part(self, name, ob1, ob2):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
103 if name is None:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
104 return ob1 == ob2
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
105 no = []
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
106 return getattr(ob1, name, no) == getattr(ob2, name, no) is not no
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
107
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
108 def __iter__(self):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
109 return IterIter(self)
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
110
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
111 class InnerBase:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
112 '''Base Inner class for Iterators'''
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
113 # Prep sets up ._next and .end
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
114 def prep_next(self, it):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
115 it.next = self.no_next
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
116 it.end = 1
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
117 return 0
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
118
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
119 # Advance knocks them down
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
120 def advance(self, it):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
121 it._last = it.item
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
122 it.item = it._next
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
123 del it._next
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
124 del it.end
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
125 it.start = 0
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
126
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
127 def no_next(self, it):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
128 return 0
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
129
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
130 def it_end(self, it):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
131 if hasattr(it, '_next'):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
132 return 0
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
133 return not self.prep_next(it)
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
134
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
135 class SeqInner(InnerBase):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
136 '''Inner class for sequence Iterators'''
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
137
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
138 def _supports(self, ob):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
139 try: ob[0]
1233
69bf0d381fd7 Zope Collector fixes.
Richard Jones <richard@users.sourceforge.net>
parents: 1049
diff changeset
140 except (TypeError, AttributeError): return 0
1049
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
141 except: pass
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
142 return 1
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
143
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
144 def prep_next(self, it):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
145 i = it.nextIndex
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
146 try:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
147 it._next = it.seq[i]
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
148 except IndexError:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
149 it._prep_next = self.no_next
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
150 it.end = 1
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
151 return 0
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
152 it.end = 0
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
153 return 1
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
154
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
155 def it_length(self, it):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
156 it.length = l = len(it.seq)
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
157 return l
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
158
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
159 try:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
160 StopIteration=StopIteration
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
161 except NameError:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
162 StopIteration="StopIteration"
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
163
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
164 class IterInner(InnerBase):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
165 '''Iterator inner class for Python iterators'''
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
166
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
167 def _supports(self, ob):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
168 try:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
169 if hasattr(ob, 'next') and (ob is iter(ob)):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
170 return 1
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
171 except:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
172 return 0
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
173
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
174 def prep_next(self, it):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
175 try:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
176 it._next = it.seq.next()
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
177 except StopIteration:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
178 it._prep_next = self.no_next
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
179 it.end = 1
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
180 return 0
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
181 it.end = 0
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
182 return 1
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
183
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
184 class IterIter:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
185 def __init__(self, it):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
186 self.it = it
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
187 self.skip = it.nextIndex > 0 and not it.end
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
188 def next(self):
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
189 it = self.it
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
190 if self.skip:
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
191 self.skip = 0
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
192 return it.item
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
193 if it.next():
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
194 return it.item
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
195 raise StopIteration
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
196
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
197 seqInner = SeqInner()
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
198 iterInner = IterInner()
Roundup Issue Tracker: http://roundup-tracker.org/