summaryrefslogtreecommitdiff
path: root/src/zope/pagetemplate/tests/batch.py
blob: 45f3ad9266bfa2ccfea907afba9c452bb249660d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
##############################################################################
#
# Copyright (c) 2001, 2002 Zope Foundation and Contributors.
# All Rights Reserved.
#
# This software is subject to the provisions of the Zope Public License,
# Version 2.1 (ZPL).  A copy of the ZPL should accompany this distribution.
# THIS SOFTWARE IS PROVIDED "AS IS" AND ANY AND ALL EXPRESS OR IMPLIED
# WARRANTIES ARE DISCLAIMED, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
# WARRANTIES OF TITLE, MERCHANTABILITY, AGAINST INFRINGEMENT, AND FITNESS
# FOR A PARTICULAR PURPOSE.
#
##############################################################################
"""Batching support tests
"""

class batch(object):
    """Create a sequence batch"""

    def __init__(self, sequence, size, start=0, end=0,
                 orphan=3, overlap=0):

        start=start+1

        start,end,sz=opt(start,end,size,orphan,sequence)

        self._last=end-1
        self._first=start-1

        self._sequence=sequence
        self._size=size
        self._start=start
        self._end=end
        self._orphan=orphan
        self._overlap=overlap

    def previous_sequence(self): return self._first

    def previous_sequence_end_number(self):
        start,end,spam=opt(0, self._start-1+self._overlap,
                           self._size, self._orphan, self._sequence)
        return end

    def previous_sequence_start_number(self):
        start,end,spam=opt(0, self._start-1+self._overlap,
                           self._size, self._orphan, self._sequence)
        return start

    def previous_sequence_end_item(self):
        start,end,spam=opt(0, self._start-1+self._overlap,
                           self._size, self._orphan, self._sequence)
        return self._sequence[end-1]

    def previous_sequence_start_item(self):
        start,end,spam=opt(0, self._start-1+self._overlap,
                           self._size, self._orphan, self._sequence)
        return self._sequence[start-1]

    def next_sequence_end_number(self):
        start,end,spam=opt(self._end+1-self._overlap, 0,
                           self._size, self._orphan, self._sequence)
        return end

    def next_sequence_start_number(self):
        start,end,spam=opt(self._end+1-self._overlap, 0,
                           self._size, self._orphan, self._sequence)
        return start

    def next_sequence_end_item(self):
        start,end,spam=opt(self._end+1-self._overlap, 0,
                           self._size, self._orphan, self._sequence)
        return self._sequence[end-1]

    def next_sequence_start_item(self):
        start,end,spam=opt(self._end+1-self._overlap, 0,
                           self._size, self._orphan, self._sequence)
        return self._sequence[start-1]


    def next_sequence(self):
        try: self._sequence[self._end]
        except IndexError: return 0
        else: return 1

    def __getitem__(self, index):
        if index > self._last: raise IndexError(index)
        return self._sequence[index+self._first]

def opt(start,end,size,orphan,sequence):
    if size < 1:
        if start > 0 and end > 0 and end >= start:
            size=end+1-start
        else: size=7

    if start > 0:

        try: sequence[start-1]
        except: start=len(sequence)

        if end > 0:
            if end < start: end=start
        else:
            end=start+size-1
            try: sequence[end+orphan-1]
            except: end=len(sequence)
    elif end > 0:
        try: sequence[end-1]
        except: end=len(sequence)
        start=end+1-size
        if start - 1 < orphan: start=1
    else:
        start=1
        end=start+size-1
        try: sequence[end+orphan-1]
        except: end=len(sequence)
    return start,end,size