add everything_between

[CodyKochmann]

this is really handy for things like scope parsing, should have been added a long time ago.

from generators import window
from collections import deque

def everything_between(pipe, before, after):
    if type(before)==type(after):
        output_type = ''.join if isinstance(before, str) else type(before)
    else:
        output_type = list
    
    before = tuple(before) if hasattr(before, '__len__') else (before,)
    after = tuple(after) if hasattr(after, '__len__') else (after,)
    
    pipe = window(pipe, max(len(before), len(after)))
    pipe = ((i[0], i[:len(before)]==before, i[:len(after)]==after) for i in pipe)
    
    output = deque()
    
    for i, is_before, is_after in pipe:
        if is_before:
            output.clear()
            for _ in range(len(before)-1):
                next(pipe)
        elif is_after and output:
                yield output_type(output)
                output.clear()
        else:
            output.append(i)
    
if __name__ == '__main__':
    s = 'hello world hello world'
    print(
        list(
            everything_between(
                s, 'o', 'l'
            )
        )
    )
    print(
        list(
            everything_between(
                s, 'he', 'wo'
            )
        )
    )
    print(list(
        everything_between(b'hello world hello world', b'he', b'wo')
    ))