56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
import io
|
|
import itertools
|
|
|
|
from cpl_query.exceptions import IndexOutOfRangeException
|
|
|
|
|
|
class SequenceEnd:
|
|
|
|
def __init__(self):
|
|
self.is_ended = False
|
|
|
|
def set_end(self, value: bool) -> 'SequenceEnd':
|
|
self.is_ended = value
|
|
return self
|
|
|
|
|
|
class SequenceValues:
|
|
def __init__(self, data, _t: type):
|
|
if data is None:
|
|
data = []
|
|
|
|
if not hasattr(data, '__iter__'):
|
|
raise TypeError(f'{type(self).__name__} must be instantiated with an iterable object')
|
|
|
|
self._data = []
|
|
for element in data:
|
|
if _t is not None and type(element) != _t and not isinstance(type(element), _t) and not issubclass(type(element), _t):
|
|
raise Exception(f'Unexpected type: {type(element)}\nExpected type: {_t}')
|
|
self._data.append(element)
|
|
self._index = 0
|
|
self._len = sum(1 for item in self._data)
|
|
self._cycle = itertools.cycle(self._data)
|
|
|
|
def __len__(self):
|
|
return self._len
|
|
|
|
def __iter__(self):
|
|
i = 0
|
|
while i < len(self):
|
|
yield next(self._cycle)
|
|
i += 1
|
|
|
|
def __next__(self):
|
|
if self._index >= len(self):
|
|
raise IndexOutOfRangeException()
|
|
self._index += 1
|
|
|
|
return self.next()
|
|
|
|
def next(self):
|
|
return next(self._cycle)
|
|
|
|
def reset(self):
|
|
self._index = 0
|
|
self._cycle = itertools.cycle(self._data)
|