本篇將介紹一個有趣實用的 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