Clojure's -> and ->> operator in python

Thread-end.py

"""
This mimic the behavior of clojure's Thread-end ->>

"""

from functools import reduce, wraps
from typing import Callable


def double_rightarrow_decorator(f: Callable):
    """
    f(..., a,b,c) -> f(..., a,b)(c)
    """

    @wraps(f)
    def decorated(*args):
        @wraps(f)
        def after(first):
            return f(*args, first)

        return after

    return decorated


def apply(x, f: Callable):
    return f(x)


def rightarrow(var, *funcs: Callable):
    return reduce(apply, funcs, var)


@double_rightarrow_decorator
def f1(x, y):
    """
    this accepts two arguments and returns a list

    """
    print(x, y)
    return [x, y]


rightarrow(1, f1(2), f1(3), f1(4), f1(5))

Thread-first.py

"""
This mimic the behavior of Clojure's Thread-first ->

"""

from functools import reduce, wraps
from typing import Callable


def rightarrow_decorator(f: Callable):
    """
    f(a,b,c,...) -> f(b,c,...)(a)

    """

    @wraps(f)
    def decorated(*args, **kwargs):
        @wraps(f)
        def after(first):
            return f(first, *args, **kwargs)

        return after

    return decorated


def apply(x, f: Callable):
    return f(x)


def rightarrow(var, *funcs: Callable):
    return reduce(apply, funcs, var)


@rightarrow_decorator
def f1(x, y):
    """
    this accepts two arguments and returns a list

    """
    print(x, y)
    return [x, y]


rightarrow(1, f1(2), f1(3), f1(4), f1(5))