Pipe - Python Infix Programming Toolkit

本篇將介紹一個有趣實用的 Python 套件 － pipe 。

這套件的用途可以讓 Python 使用像 Linux 指令常用的 | (pipe) 來進行資料的處理。

例如以下的 Python 程式：

>>> x = [1, 2, 3]
>>> ', '.join([str(i) for i in x[::-1]])

可以用 pipe 改寫成：

>>> [1, 2, 3] | reverse | concat
'3, 2, 1'

整個看起來就簡潔很多，而且易讀性也提高很多。

而 pipe 這個套件是怎麼做到這件事的呢？

簡單來說， pipe 實作了 __ror__ 這個物件方法，這個方法就代表 Python 中的 | (Bitwise Or) 運算子，所以可以利用 | 對物件進行操作。

然後再實作物件裝飾子(class decorator) ，就可以將各種函數轉為方便的 pipe 函數。

所以 pipe 類別最簡單的樣子會長這樣：

class Pipe(object):

    def __init__(self, function):
        # 讓裝飾子可以傳一個函數進來，作為 `__ror__` 可以使用的 pipe 函數
        self.function = function

    def __ror__(self, other):
        # 讓 pipe 函數處理 pipe 進來的參數
        return self.function(other)

    def __call__(self, *args, **kwargs):
        # 實作 class decorator 的介面
        return Pipe(lambda x: self.function(x, *args, **kwargs))


@Pipe
def take_the_first_one(iterable):
    return iterable[0:1]

實作完之後就可以用以下的方式來取第一個值了！

>>> a = 'Hello' | take_the_first_one
>>> print a
'H'

以上就是 Pipe 套件的原理與實作，更多方便的 pipe 函數可以參閱 pipe 。

python pipe infix programming