Data pipelining in Python

pipe.py

import itertools

class Pipe:
    def __init__(self, seq):
        self._seq = seq

    def __iter__(self):
        return iter(self._seq)
    
    def map(self, func):
        return Pipe(map(func, self))

    def flat_map(self, func=None):
        maped_iter = self if func is None else map(func, self)
        return Pipe(itertools.chain.from_iterable(maped_iter))

    def filter(self, func=None):
        return Pipe(filter(func, self))
    
    def before(self, func):
        return Pipe(before_it(func, self))

    def after(self, func):
        return Pipe(after_it(func, self))
    
    @classmethod
    def from_repeated(cls, value):
        return Pipe(repeat(value))

def repeat(value):
    func = value if callable else lambda: value
    while True:
        yield func()

def before_it(func, iter1):
    for it in iter1:
        func()
        yield it

def after_it(func, iter1):
    for it in iter1:
        yield it
        func()