六行代码实现 Python 管道

[abersheeran]

不多说，直接上代码。

from functools import partial

class F(partial):
    def __ror__(self, other):
        if isinstance(other, tuple):
            return self(*other)
        return self(other)

使用样例。

range(10) | F(filter, lambda x: x % 2) | F(sum)

更详细的前因后果介绍，可以看 https://aber.sh/articles/Python-Pipe/ ，实在是懒得复制到 V2EX 调样式了。有兴趣的就看看，没兴趣的就复制代码去用就完事。

小小的得瑟一下，这是最近几天写的最满意、最有用的代码。妙手偶得之~

[abersheeran]

@bruce00 谢谢~

[abersheeran]

@Wincer 要实现你这个，首先要让这些 callable 支持自动柯里化……不是支持一个管道运算符就行了的

自动柯里化的工作量就太大了，因为 Py 里存在 *args 和 **kwargs 。我是想不到有什么好办法能实现的。所以我选择让程序员自己柯里化一次（也就是使用 F 包裹一次）。

[Wincer]

@abersheeran 嗯是的，这确实是目前比较合适的方法了

[iqxd]

请问下楼主 if isinstance(other, tuple): return self(*other) 是为了处理哪种情况呢?

[abersheeran]

@frostming
@Wincer

就比如说明希想要的那个，可以用
def fp(func):
def _(*args, **kwargs):
return F(func, *args, **kwargs)
return _

折中实现一下，强制 double call 才能实际调用。

而你要做的这个，既要支持 direct call 又要支持
double call……那就得第一次调用之后延迟计算，判断如果是走管道，调第二次，判断直接赋值的话再去计算。CPython 应该做不到，可以考虑做个超集语言，在编译期做。

[NeezerGu]

@abersheeran
不是这个吗？ https://github.com/python/cpython/blob/master/Lib/functools.py
哦，这是说， 同时提供了 python 版本和 c 版本？ 如果 c 版本的导入失败就用 python 的？

[frostming]

@Wincer 如果只是 patch function 大可直接把 builtin 的 map, filter 换掉，我说的那种可以用来给 list 加 chaining call:

[1, 2, 3].map(lambda x: x**2)

[mckelvin]

`sum(i for i in range(10) if i % 2)` 比 `range(10) | F(filter, lambda x: x % 2) | F(sum)` 更难读吗？

[yuruizhe]

@Leigg 同+1，我还纳闷 shell 咋能调用 python 函数呢…老实说，我还真没把 python shell 当成过主力 shell，都是 bash…

[muzuiget]

这种用法容易走火入魔。

[Merlini]

之前做文本处理的时候看到网上一个 pipeline 写法感觉也挺不错。
```python
def pipeline(
value: T,
function_pipeline: Sequence[Callable[[T], T]],
) -> T:
"""A generic Unix-like pipeline
:param value: the value you want to pass through a pipeline
:param function_pipeline: an ordered list of functions that
comprise your pipeline
"""
return reduce(lambda v, f: f(v), function_pipeline, value)
```

[zouzou0208]

好玩，原来是 index.py 的作者。好棒。

[chaleaoch]

在这么玩, 会被玩坏的喂~~~~~

[shyling]

🤪f#自带语法

[Wincer]

@frostming 是的，这样也可以，但是适用性会窄一些，需要预先针对某一种类型 patch

[shyrock]

赞！学习了。

[aldslvda]

学习了

[iintothewind]

你这个管道实现是不是可以再优化一下，
毕竟每次都得带 F() 包住后面的操作符， 这个看起来有点多余。

[admirez]

pandas 也可以这样搞么？

[sapocaly]

个人觉得 chain funciton call 可能更符合 python 一些至少更接近 django 一些。比如 range(10).filter(lambda x: x % 2).sum()。毕竟每次加个 F 还是有点难受。当然，去掉 F 容易，我是没想到怎么实现 range(10) | filter, lambada x:x |sum 这样的 syntax