042、多态与鸭子类型：Python 的接口哲学与 Protocol 类型检查

042、多态与鸭子类型：Python 的接口哲学与 Protocol 类型检查

一个让我半夜加班的 Bug

去年接手一个遗留项目，代码里有个函数接收一个“文件类对象”，文档写着“支持 read 和 write 即可”。我传了一个自定义的 StreamBuffer 进去，单元测试全绿，上线后半夜报警——生产环境某个第三方库返回的对象没有 write 方法，直接 AttributeError 崩了。排查时发现，代码里到处是if hasattr(obj, 'write')这种“类型检查”，但漏了一个分支。

这个坑让我重新审视 Python 的接口哲学：我们到底该不该检查类型？怎么检查才算优雅？

多态：不是继承的专利

Java 或 C++ 里，多态通常依赖继承——子类重写父类方法，通过父类引用调用子类实现。Python 的多态更“野”：只要对象有对应方法，就能用，管你什么继承关系。

classDuck:defquack(self):print("嘎嘎")classPerson:defquack(self):print("我学鸭子叫")defmake_it_quack(thing):thing.quack()# 这里不关心类型，只关心有没有 quack 方法make_it_quack(Duck())# 嘎嘎make_it_quack(Person())# 我学鸭子叫

这就是多态——同一个接口（quack 方法），不同行为。Python 不强制你继承某个基类，只要“长得像鸭子，叫得像鸭子”，那就是鸭子。

鸭子类型：Python 的接口哲学

“如果它走起来像鸭子，叫起来像鸭子，那它就是鸭子。”这句话是鸭子类型的精髓。Python 推崇“约定优于契约”——你不需要显式声明实现了某个接口，只要你的对象有对应的方法，就能被当作那个接口使用。

别这样写：

defprocess(data):ifnotisinstance(data,list):# 这里踩过坑，限制了传入类型raiseTypeError("必须传列表")# 处理逻辑

应该这样写：

defprocess(data):# 只要支持迭代就行，管它是列表、元组还是生成器foritemindata:# 处理逻辑pass

鸭子类型让代码更灵活，但也带来隐患：如果传入的对象缺少预期的方法，运行时才报错。这就是我那个生产事故的根源——代码假设所有“文件类对象”都有 write，但实际没有。

Protocol：给鸭子类型加上“类型安全带”

Python 3.8 引入的typing.Protocol解决了这个问题。它允许你定义一个“协议”——一个类只要实现了协议中声明的方法，就被视为该协议的子类型，无需显式继承。

fromtypingimportProtocolclassWritable(Protocol):defwrite(self,data:str)->None:...# 这里定义协议classFileWriter:defwrite(self,data:str):print(f"写入文件:{data}")classNetworkWriter:defwrite(self,data:str):print(f"发送网络:{data}")classBadWriter:defsend(self,data:str):# 没有 write 方法，不符合协议passdefsave_data(writer:Writable,data:str):writer.write(data)# 类型检查器会验证 writer 是否符合 Writable 协议save_data(FileWriter(),"hello")# 通过save_data(NetworkWriter(),"world")# 通过save_data(BadWriter(),"fail")# mypy 或 Pyright 会报错：BadWriter 不符合 Writable 协议

注意：Protocol 是静态类型检查用的，运行时不会强制检查。你仍然可以传一个没有 write 的对象进去，但 IDE 和类型检查工具会提前警告你。

实战：用 Protocol 重构遗留代码

回到开头那个生产事故，我用 Protocol 重构了文件类对象的处理：

fromtypingimportProtocol,OptionalclassReadableWritable(Protocol):defread(self,size:int=-1)->bytes:...defwrite(self,data:bytes)->int:...defclose(self)->None:...defprocess_stream(stream:ReadableWritable)->None:# 这里明确要求 stream 必须实现 read、write、closedata=stream.read()# 处理数据stream.write(result)stream.close()

然后在调用处，如果传入了不符合协议的对象，mypy 会直接报错，不用等到线上崩溃。配合isinstance做运行时兜底：

defsafe_process(stream)->None:ifnothasattr(stream,'read')ornothasattr(stream,'write'):raiseValueError("stream 必须支持 read 和 write 方法")# 这里兜底process_stream(stream)

个人经验性建议

1. 鸭子类型是 Python 的灵魂，但别裸奔。小脚本里随便用，生产代码建议用 Protocol 做静态检查。我见过太多“运行时 AttributeError”的工单了。

2. Protocol 和 ABC 怎么选？如果你需要运行时检查（比如isinstance(obj, MyABC)），用 ABC。如果只是静态类型提示，Protocol 更轻量，而且不强制继承关系。我倾向于：新项目全用 Protocol，旧项目逐步迁移。

3. 别滥用hasattr做运行时检查。它只能检查属性是否存在，不能检查方法签名是否正确。而且hasattr会吞掉某些异常（比如属性访问时触发的异常），调试时很坑。

4. 写文档时明确“接口契约”。比如“这个函数接受一个支持 read(size) 和 write(data) 的对象”，配合 Protocol 类型注解，比写十行注释都管用。

5. 类型检查工具要配齐。mypy 或 Pyright 必须上，CI 里跑一遍。我见过太多人写了 Protocol 但没配类型检查，等于白写。

最后，记住一句话：Python 的接口哲学是“信任程序员，但用工具辅助”。鸭子类型给你自由，Protocol 给你安全，两者结合才是生产级的写法。