Python中如何复用参数类型分支处理的函数逻辑并校验参数类型？

可以通过封装通用逻辑的方式避免重复代码，同时完成类型校验，这里提供两种简洁的实现方式：

方式1：通用工具函数 #

先定义一个统一处理x参数的工具函数，包含类型校验和取值逻辑，所有需要该逻辑的函数直接调用这个工具函数即可：

def get_x_value(t, x):
    # 类型校验：仅允许float或可调用对象（函数）
    if isinstance(x, float):
        return x
    elif callable(x):
        return x(t)
    else:
        raise TypeError("参数x必须是float类型或可调用函数")

之后你的业务函数就可以简化为：

def foo(t, y, x):
    x_val = get_x_value(t, x)
    return t * y * x_val

def bar(t, y, x):
    x_val = get_x_value(t, x)
    return t + y + x_val

# 测试示例
print(foo(2, 3, 4.0))  # 输出24.0
print(foo(2, 3, lambda t: t**2))  # 输出2*3*(2²)=24

方式2：装饰器 #

如果需要批量处理大量函数，装饰器会更简洁，只需给函数加一个装饰器，即可自动完成x的类型校验和取值转换：

def handle_x_param(func):
    def wrapper(t, y, x):
        # 类型校验与取值转换
        if isinstance(x, float):
            x_val = x
        elif callable(x):
            x_val = x(t)
        else:
            raise TypeError("参数x必须是float类型或可调用函数")
        # 调用原函数，传入处理后的x值
        return func(t, y, x_val)
    return wrapper

使用装饰器修饰业务函数：

@handle_x_param
def foo(t, y, x_val):
    return t * y * x_val

@handle_x_param
def baz(t, y, x_val):
    return t * y / x_val

# 测试示例
print(baz(4, 2, 0.5))  # 输出4*2/0.5=16.0
print(baz(4, 2, lambda t: t/2))  # 输出4*2/(4/2)=4.0

两种方式的区别：

• 工具函数更灵活，适合需要在函数内部多次使用x值，或需要自定义处理时机的场景；
• 装饰器更简洁，一次修饰即可统一处理所有逻辑，适合批量定义同类型函数的场景。

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