20、Python 类型检查与 pydantic 库的应用

Python 类型检查与 pydantic 库的应用

1. Python 中的协议与类型检查

在 Python 中，我们可以定义一些函数和协议来处理特定的任务。例如，有如下代码：

def render_menu(self) -> Menu:
    """ No implementation needed """
    ...

def load_restaurant(restaurant: Restaurant):
    # code to load restaurant
    # ...

我们可以将导入的模块传递给 load_restaurant 函数：

import restaurant
from load_restaurant import load_restaurant
# Loads our restaurant model
load_restaurant(restaurant)

在 main.py 中，调用 load_restaurant 函数时类型检查可以正常通过。这里的 restaurant 模块满足我们定义的 Restaurant 协议。协议在类型检查时非常智能，当传入一个模块时，它会忽略 render_menu 中的 self 参数。