Python函数管道实现：如何统一处理带参与无参函数？

方案1：统一使用「函数+参数字典」元组格式 #

不管函数是否需要额外参数，都统一存储为(func, kwargs)的元组，无参函数的kwargs传空字典即可。这种方式格式完全一致，遍历处理逻辑极简，无需复杂解析。

示例代码：

def add_one(n):
    return n + 1

def multiply_by(n, factor):
    return n * factor

# 统一定义管道步骤
pipeline = [
    (add_one, {}),
    (multiply_by, {"factor": 3})
]

# 执行管道逻辑
def run_pipeline(initial_num, pipeline):
    result = initial_num
    for func, kwargs in pipeline:
        result = func(result, **kwargs)
    return result

print(run_pipeline(2, pipeline))  # 输出：9

方案2：用dataclass封装步骤（可读性更强） #

如果希望结构更清晰，可借助dataclasses.dataclass封装每个管道步骤，自动处理无参场景的默认值，用户使用时更直观。

示例代码：

from dataclasses import dataclass

@dataclass
class PipelineStep:
    func: callable
    kwargs: dict = None

    def __post_init__(self):
        # 无参时自动设为空字典
        if self.kwargs is None:
            self.kwargs = {}

def add_one(n):
    return n + 1

def multiply_by(n, factor):
    return n * factor

# 定义管道
pipeline = [
    PipelineStep(add_one),
    PipelineStep(multiply_by, {"factor": 3})
]

# 执行管道
def run_pipeline(initial_num, pipeline):
    result = initial_num
    for step in pipeline:
        result = step.func(result, **step.kwargs)
    return result

print(run_pipeline(2, pipeline))  # 输出：9

方案3：封装partial的简易接口（隐藏底层细节） #

如果偏好使用偏函数，但不想让用户直接接触functools.partial，可以自己写一个轻量封装函数，降低用户学习成本。

示例代码：

import functools

def bind_args(func, **kwargs):
    return functools.partial(func, **kwargs)

def add_one(n):
    return n + 1

def multiply_by(n, factor):
    return n * factor

# 定义管道
pipeline = [
    add_one,  # 无参函数直接传入
    bind_args(multiply_by, factor=3)
]

# 执行管道
def run_pipeline(initial_num, pipeline):
    result = initial_num
    for func in pipeline:
        result = func(result)
    return result

print(run_pipeline(2, pipeline))  # 输出：9

内容的提问来源于stack exchange，提问作者JachymDvorak