Python typing.Protocol实现结构化子类型的鸭子类型

Python的typing.Protocol是结构化子类型（structural subtyping）的核心工具，它允许你根据对象的行为（方法/属性）而非继承关系来定义类型。这种“鸭子类型”（Duck Typing）机制让代码更灵活，避免传统继承的 rigid 约束。本文通过实操步骤教你如何用Protocol实现结构化子类型。

传统继承的局限：为什么需要Protocol？

假设你需要定义一个“可绘制”的接口，传统做法是创建基类并强制子类继承：

class Drawable:
    def draw(self) -> None:
        raise NotImplementedError

class Circle(Drawable):
    def draw(self) -> None:
        print("画圆形")

class Square(Drawable):
    def draw(self) -> None:
        print("画方形")

但如果一个类需要同时支持“可绘制”和“可保存”两种行为，传统继承会导致多重继承问题（菱形问题）。例如：

class Saveable:
    def save(self) -> None:
        raise NotImplementedError

# 多重继承导致复杂度上升
class Circle(Drawable, Saveable):
    def draw(self) -> None:
        print("画圆形")
    def save(self) -> None:
        print("保存圆形")

这种设计让类关系变得复杂，且难以扩展。Protocol通过结构化子类型解决了这个问题。

用Protocol实现结构化子类型：3步完成

步骤1：导入Protocol并定义接口

使用typing.Protocol定义接口，只需声明方法签名，无需实现：

from typing import Protocol

class Drawable(Protocol):
    def draw(self) -> None: ...

...表示这是一个抽象方法，实现类必须提供具体实现。

步骤2：实现类满足Protocol

任何类只要实现了draw方法，就会被视为Drawable类型，无需继承：

class Circle:
    def draw(self) -> None:
        print("画圆形")

class Square:
    def draw(self) -> None:
        print("画方形")

class Text:
    def draw(self) -> None:
        print("画文本")

Circle、Square、Text都没有继承Drawable，但都满足Drawable的结构要求。

步骤3：使用Protocol进行类型检查

通过isinstance或类型注解验证对象是否符合Protocol：

def render(obj: Drawable) -> None:
    obj.draw()

render(Circle())  # 输出: 画圆形
render(Square())  # 输出: 画方形
render(Text())   # 输出: 画文本

类型检查器（如mypy）会自动验证obj是否实现了draw方法，无需运行时检查。

结构化子类型的优势：对比传统继承

实际应用：插件系统设计

假设你正在开发一个图像编辑器，需要支持多种插件（滤镜、调整器等）。用Protocol定义插件接口：

from typing import Protocol

class Filter(Protocol):
    def apply(self, image: str) -> str: ...

class BrightnessAdjuster(Protocol):
    def adjust(self, image: str, level: int) -> str: ...

插件实现类无需继承，只需满足接口：

class BlurFilter:
    def apply(self, image: str) -> str:
        return f"模糊处理: {image}"

class ContrastFilter:
    def apply(self, image: str) -> str:
        return f"对比度调整: {image}"

class GammaAdjuster:
    def adjust(self, image: str, level: int) -> str:
        return f"伽马值调整: {image} (level={level})"

插件管理器通过Protocol检查插件类型：

def apply_filters(image: str, filters: list[Filter]) -> str:
    for filter in filters:
        image = filter.apply(image)
    return image

def adjust_brightness(image: str, adjuster: BrightnessAdjuster, level: int) -> str:
    return adjuster.adjust(image, level)

# 使用示例
image = "原图"
filters = [BlurFilter(), ContrastFilter()]
adjusted_image = apply_filters(image, filters)
result = adjust_brightness(adjusted_image, GammaAdjuster(), 50)
print(result)  # 输出: 伽马值调整: 对比度调整: 模糊处理: 原图 (level=50)

注意事项：Protocol的局限性

1. 无法检查私有属性：Protocol只检查公开方法/属性，私有成员（如_private_method）不会被验证。
2. 运行时无实际类型：Protocol是类型检查工具，运行时不会创建实际类型，无法通过isinstance检查（除非显式注册）。
3. 方法签名必须完全匹配：包括参数类型、返回类型和名称，否则不满足Protocol。

通过Protocol，你可以用更灵活的方式实现鸭子类型，让代码更解耦、更易扩展。尝试在你的项目中用Protocol替代传统继承，体验结构化子类型的优势。