yieldwhile

yieldwhile.py

def yieldwhile(predicate, iterable):
    """Yield every element from the iterable as long as the predicate is True

    This is a counterpart to itertools.dropwhile.

    Args:
        predicate (function): a function taking a single item from iterable as an argument,
            and returning a boolean
        iterable (iterable): an iterable

    Returns:
        (generator) all elements from iterable as long as the predicate is True / until the predicate becomes False
    """
    iterable = iter(iterable)
    for item in iterable:
        if predicate(item):
            yield item
        else:
            break