def __init__(self, iterable, total_size):
    self.iterator = iter(iterable)
    self.total_size = total_size
def __init__(self, iterable, total_size):
    self.iterator = iter(iterable)
    self.total_size = total_size