Haskell-like do notation

[MichaelSchneeberger]

I implemented a do decorator that allows to avoid nesting flat_map function using the Haskell-like do notation.

import rx

from donotation import do
from rx import operators as op


def callback(source, fn):
    return source.pipe(
        op.flat_map(fn),
    )


# define new do decorator applying custom flat_map operation using a callback
rx_do = do(callback=callback)


@rx_do
def rx_action():
    v = yield rx.from_([1, 2, 3, 4])

    if v % 2 == 0:
        return rx.from_(list(range(v)))
    else:
        return rx.empty()

# Output will be 0 1 0 1 2 3
rx_action().subscribe(print)

Here is the link to the project:

https://github.com/MichaelSchneeberger/do-notation