def generator():
    i = 1
while True:
    yield i ** 2
i += 1

def fill(pool, gen):
    pool += itertools.islice(gen, poolSize - len(pool))
gen = generator()
pool = []
Как чисто захватить из бесконечного генератора?