如何复用Python函数签名以实现智能提示与低维护封装？

要让A.func复用B.send的函数签名，同时让MyPy和IDE智能提示生效，可以通过类型注解+函数签名绑定的方式实现，以下是两种可行方案：

方案一：用ParamSpec实现静态类型对齐（适配MyPy） #

Python 3.10+支持ParamSpec，可以精准捕获函数的参数结构，结合TypeVar定义返回类型，让静态检查工具识别A.func的参数与B.send完全一致：

from typing import TypeVar, ParamSpec, Callable
import functools
import inspect

# 定义参数规格和返回类型变量
P = ParamSpec("P")
R = TypeVar("R")

class B:
    """ 外部包中的原始类 """
    def __enter__(self):
        return self
    def __exit__(self, *_):
        pass
    def send(self, msg: str, timeout: float = 5.0) -> None:
        """ 需要复用的目标方法签名 """
        pass
    def get(self) -> str:
        return "response"

class A:
    def __init__(self):
        self.other = B()

    def func(self, *args: P.args, **kwargs: P.kwargs) -> R:
        with self.other as connection:
            connection.send(*args, **kwargs)
            response = connection.get()
        return response

# 绑定B.send的签名到A.func，让IDE智能提示生效
# 移除B.send自带的self参数，避免提示混乱
orig_params = list(inspect.signature(B.send).parameters.values())
A.func.__signature__ = inspect.signature(B.send).replace(parameters=orig_params[1:])

这样做的优势：

• MyPy会严格检查func的参数是否符合B.send的要求
• 修改__signature__后，IDE会直接显示B.send的参数提示

方案二：自定义装饰器一键复用签名（更简洁） #

写一个轻量装饰器，自动复制目标方法的签名到当前函数，同时保留函数元信息：

import inspect
import functools

def reuse_signature(source_func: Callable) -> Callable:
    def decorator(target_func: Callable) -> Callable:
        # 复制源函数签名，移除self参数
        orig_params = list(inspect.signature(source_func).parameters.values())
        target_func.__signature__ = inspect.signature(source_func).replace(parameters=orig_params[1:])
        # 保留源函数的文档、名称等元信息
        return functools.wraps(source_func)(target_func)
    return decorator

class B:
    """ 外部包中的原始类 """
    def __enter__(self):
        return self
    def __exit__(self, *_):
        pass
    def send(self, msg: str, timeout: float = 5.0) -> None:
        pass
    def get(self) -> str:
        return "response"

class A:
    def __init__(self):
        self.other = B()

    @reuse_signature(B.send)
    def func(self, *args, **kwargs) -> str:
        with self.other as connection:
            connection.send(*args, **kwargs)
            response = connection.get()
        return response

这个装饰器会自动完成签名复制，无需手动修改__signature__，MyPy和IDE都能正确识别参数提示。

注意事项 #

• 若使用Python 3.9及以下版本，需安装typing-extensions库来使用ParamSpec
• 如果B是动态导入的，要确保在调用装饰器或修改签名时，B.send的签名已被正确加载

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