falan / pipes Goto Github PK

In the Elixir programming language the |> pipe operator allows you to chain together multiple function calls so that this:

c(b(a(1, 2), 3, 4))

can be written more readably as:

1 |> a(2) |> b(3, 4) |> c()

All the pipe operator does is pass its left operand as the first argument of the right operand, so that a |> b(...) becomes b(a, ...).

Various pipe implementations in Python to date allow a list of functions to be applied to an initial value, but do not support the partial, missing first argument syntax of Elixir.

This library provides a function decorator that causes Python >> right shift operators within the function to act exactly like Elixir pipes:

from pipeop import pipes

def add(a, b):
    return a + b
    
def times(a, b):
    return a * b
    
@pipes
def calc()
    print 1 >> add(2) >> times(3)  # prints 9

Functions can have any number of arguments:

def add3(a, b, c):
    return a + b + c
    
@pipes
def calc()
    print 1 >> add3(2, 3)  # prints 6

In Elixir libraries the first argument of a function is chosen with pipes in mind but this is (obviously) not the case in Python - for instance the enumerable args of map and reduce are first in their Elixir equivalents but last in Python. For this reason I've also redefined the left shift operator << to append it's left operand to the list of call arguments of the right operand:

@pipes
def my_pow():
  print 2 >> pow(3)  # prints 8
  print 2 << pow(3)  # prints 9

You can drop the braces for functions or lambdas (enclosed in braces) with a single argument:

@pipes
def sum(self):
    print [1, 2, 3] >> sum  # prints 6
    print 1 >> (lambda x: x + 1)  # prints 2

In Elixir pipes are often laid out one per line. In Python you need brackets to do the same thing without line continuations, but it still looks pretty neat:

@pipes
def pretty_pipe():
    print (
        range(-5, 0)
        << map(lambda x: x + 1)
        << map(abs)
        << map(str)
        >> tuple
    )  # prints ('4', '3', '2', '1', '0')

The decorator can also be applied to a class to decorate all the methods in that class.

Normally there should be a small amount of processing overhead on the first time the function is defined due to the function being recompiled and cached. Otherwise there should be no difference to the performance of the conventionally nested call code.

This is initial alpha code. It has been tested on Python 2.7.14 and 3.6.5 using simple functions. Source line attributes are preserved so debuggers should be able to follow the code as it executes. Pull requests and bug reports gratefully accepted.

Robin Hilliard

PS: Thanks to https://github.com/Stvad for submitting the first issue with some great suggestions.