如何在Python函数定义时直接标记其符合指定的typing.Protocol？

问题场景 #

用typing.Protocol定义了函数类型FType，但符合该协议的函数分散在不同文件中，希望直接标记函数并让类型检查器验证兼容性，同时避免代码重复。

方案1：先声明类型再定义函数 #

这是最简洁的原生写法，无需额外工具，同时能被mypy等类型检查器正确验证：

from typing import Protocol

class FType(Protocol):
    def __call__(self, a: int) -> float: ...

# 先声明函数的类型为FType
f: FType
# 再定义函数，mypy会自动检查是否符合Protocol
def f(a: int) -> float:
    return float(a)

# 错误示例：不符合Protocol的函数会被mypy报错
g: FType
def g(a: int) -> str:
    return str(a)  # mypy提示：Incompatible return type (got "str", expected "float")

这种写法没有代码重复，编辑器可以通过查找FType的引用快速定位所有标记的函数，类型检查也能正常工作。

方案2：用装饰器实现类注解式标记 #

如果你更偏好类似:FType的简洁标记风格，可以实现一个轻量装饰器，既做标记又触发类型检查：

from typing import Protocol, TypeVar, cast

# 定义泛型TypeVar用于绑定Protocol
T = TypeVar('T')

def implements(protocol: type[T]) -> callable[[T], T]:
    def decorator(func: T) -> T:
        # 用cast告诉类型检查器函数符合目标Protocol
        return cast(T, func)
    return decorator

class FType(Protocol):
    def __call__(self, a: int) -> float: ...

# 用装饰器标记函数符合FType，写法更直观
@implements(FType)
def f(a: int) -> float:
    return float(a)

# 错误示例：不符合的函数会被mypy检测到
@implements(FType)
def g(a: int) -> str:
    return str(a)  # mypy提示：Argument 1 to "implements" has incompatible type "Callable[[int], str]"; expected "FType"

这种方式的优势是标记更醒目，装饰器的命名也能清晰表达函数的协议实现关系，同时不影响函数的正常运行。

为什么之前的写法失效？ #

你之前尝试的f: FType = f写法，mypy会先将左侧的f推断为FType类型，再赋值右侧的同一个变量，导致类型检查被绕过。而先声明类型再定义函数的方式，类型检查器会先确认变量的预期类型，再验证后续的函数定义是否匹配，因此能正确报错。

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