本文探讨了如何在python中为函数属性添加类型注解，以解决pep 232定义的函数属性与pep 484定义的类型注解结合时的挑战。通过引入可调用类（即装饰器模式），我们展示了如何为函数包装器定义并类型化属性，从而在保持代码清晰度和类型安全性的同时，实现如函数缓存等功能。
在Python中，函数不仅是可执行的代码块，它们也可以拥有自己的属性，这在PEP 232中有所定义。这些函数属性常用于存储与函数相关的状态，例如缓存、计数器或配置信息。然而，随着PEP 484引入的类型注解（Type Hints）成为现代Python开发的重要组成部分，如何为这些函数属性添加类型注解以实现静态类型检查，却是一个不那么直观的问题。直接在函数定义后为属性赋值（例如 foo.cache = {}）虽然可行，但这种方式无法在静态分析阶段提供类型信息，降低了代码的可维护性和健壮性。
传统函数属性的局限性
考虑一个常见的场景：为函数添加一个简单的缓存机制。一个未经类型注解的实现可能如下所示：
def foo(s: str):
try:
print(foo.cache[s])
except KeyError: # 修正：原代码为Exception，应为KeyError更精确
print('NEW')
foo.cache[s] = 'CACHE' + s
# 在函数定义后添加属性
foo.cache = {} # 这是一个字典，但其类型并未被静态检查器感知在这种实现中，foo.cache 是一个运行时添加的属性。虽然它能正常工作，但像 mypy 这样的静态类型检查工具无法得知 foo.cache 的具体类型（例如 dict[str, str]），也无法在编译时检测到对 foo.cache 的错误访问。如果尝试在函数内部直接注解 foo.cache 的类型，Python语法并不支持这种内联的函数属性类型声明。
解决方案：使用可调用类实现类型安全
为了解决函数属性的类型注解问题，我们可以采用一种更具Pythonic风格的模式：使用可调用类（Callable Class）作为装饰器来包装函数。这种方法允许我们将函数属性定义为类的属性，从而利用类的类型注解能力，实现对函数属性的类型安全管理。
立即学习“Python免费学习笔记（深入）”；
下面是一个使用可调用类实现带类型注解缓存的示例：
import typing
class Cacheable:
"""
一个可调用类，用作装饰器，为被装饰函数提供一个类型化的缓存属性。
"""
cache: dict[str, str] # 定义缓存属性及其类型
_call: typing.Callable[[str], None] # 定义被包装函数签名
def __init__(self, call: typing.Callable[[str], None]) -> None:
"""
初始化Cacheable实例。
Args:
call: 被装饰的原始函数。
"""
self.cache = {} # 初始化缓存字典
self._call = call # 存储原始函数
def __call__(self, s: str) -> None:
"""
使Cacheable实例可被调用，转发调用到原始函数。
Args:
s: 原始函数的输入参数。
"""
return self._call(s)
@Cacheable
def foo(s: str) -> None:
"""
一个示例函数，使用Cacheable装饰器实现缓存功能。
"""
try:
print(foo.cache[s]) # 访问类型化的缓存属性
except KeyError:
print('new')
foo.cache[s] = f'cache{s}'
# 运行示例
print("--- 第一次调用 'a' ---")
foo('a')
print("--- 第二次调用 'a' ---")
foo('a')
print("--- 第一次调用 'b' ---")
foo('b')
# 静态类型检查的优势示例 (使用 mypy 验证)
# 尝试访问不存在的属性，mypy 会报错：
# mypy -> "Cacheable" has no attribute "someotherattribute"
# foo.someotherattribute = {}
# print(foo.someotherattribute)代码解析
-
Cacheable 类定义:
- cache: dict[str, str]: 这是关键所在。我们在这里为 cache 属性声明了类型，它是一个键和值都是字符串的字典。
- _call: typing.Callable[[str], None]: 这个属性用于存储被装饰的原始函数。它的类型注解明确了原始函数接受一个 str 参数并返回 None。
- __init__(self, call: typing.Callable[[str], None]) -> None: 构造函数接收原始函数作为参数，并将其存储在 _call 中。同时，它初始化了 cache 字典。
- __call__(self, s: str) -> None: 这个特殊方法使得 Cacheable 类的实例可以像函数一样被调用。当 foo('a') 被调用时，实际上是调用了 Cacheable 实例的 __call__ 方法，该方法进而调用了原始的 foo 函数（存储在 self._call 中）。
-
@Cacheable 装饰器:
- 当 foo 函数被 @Cacheable 装饰时，Python 会执行 foo = Cacheable(foo)。这意味着 foo 不再是原始的函数对象，而是一个 Cacheable 类的实例。
- 因此，当你在 foo 函数内部（或者外部）通过 foo.cache 访问属性时，你实际上是在访问这个 Cacheable 实例的 cache 属性。
-
类型安全性:
- 由于 Cacheable 类明确声明了 cache 属性的类型，静态类型检查工具（如 mypy）能够理解 foo.cache 的类型。
- 如果你尝试访问一个 Cacheable 实例中不存在的属性（例如 foo.someotherattribute），mypy 会立即报告错误，从而在运行时之前捕获潜在的属性访问错误。这极大地提高了代码的健壮性。
注意事项与总结
- 适用场景: 这种模式特别适用于需要为函数添加状态（如缓存、计数器、锁）且希望这些状态能够被静态类型检查的场景。
- 装饰器原理: 理解装饰器的工作原理是掌握此模式的关键。装饰器本质上是一个函数或可调用对象，它接收一个函数作为输入，并返回一个新函数（或可调用对象）。
- 代码可读性: 尽管引入了一个额外的类定义，但它通过明确的类型注解提高了代码的可读性和可维护性，尤其是在大型项目中。
- 替代方案: 对于更复杂的缓存需求，可以考虑使用Python标准库中的 functools.lru_cache 装饰器，它提供了开箱即用的缓存功能，并且在Python 3.9+ 中支持类型注解。然而，如果需要自定义缓存逻辑或管理多个函数属性，可调用类模式提供了更大的灵活性。
通过采用可调用类作为装饰器，我们不仅能够优雅地为Python函数属性添加类型注解，还能在开发阶段就利用静态类型检查的优势，从而编写出更健壮、更易于维护的代码。这种方法弥补了Python在函数属性类型注解方面的语法限制，是实现类型安全函数状态管理的一种有效策略。
