Pydantic 别名技巧：将现有字段别名指向嵌套键值

本文旨在解决在使用 Pydantic 处理遗留 API 数据时，如何将响应中的嵌套字段值映射到现有字段的问题。通过 computed_field 和 Field(exclude=True) 的组合，以及 serialization_alias 和 validation_alias 属性与 AliasPath 的结合使用，可以优雅地实现字段别名，避免手动删除原始字段，并保持数据模型的清晰和简洁。

在使用 Pydantic 处理来自遗留 API 的数据时，经常会遇到响应字段结构不符合预期的情况。一个常见的场景是，需要将响应中的嵌套字段的值映射到模型中已存在的字段上，而不是保留原始的嵌套结构。本文将介绍两种使用 Pydantic 实现这种字段别名的方法，并提供示例代码。

方法一：使用 computed_field 和 Field(exclude=True)

这种方法适用于 Pydantic v1.0 及以上版本。其核心思想是，首先将原始嵌套字段定义为一个 Pydantic 模型，然后使用 computed_field 创建一个计算字段，该字段从嵌套模型中提取所需的值。同时，使用 Field(exclude=True) 将原始嵌套字段排除在序列化结果之外。

以下是一个示例：

from pydantic import BaseModel, Field, computed_field

class Logo(BaseModel):
    url: str = ''


class Survey(BaseModel):
    logo: Logo = Field(exclude=True)

    @computed_field
    @property
    def logo_url(self) -> str:
        return self.logo.url

a = Survey(logo={'url': 'foo'})
print(a.model_dump())
#{'logo_url': 'foo'}

在这个例子中，Logo 模型表示 API 响应中的 logo 字段，它包含一个 url 属性。Survey 模型包含一个 logo 字段，类型为 Logo，并且使用 exclude=True 排除在序列化结果之外。logo_url 是一个计算字段，它使用 @computed_field 装饰器标记，并使用 @property 定义为属性。logo_url 属性从 logo 字段的 url 属性中提取值。

优点：

• 实现简单，易于理解。
• 不需要修改原始 API 响应数据。

缺点：

• 需要定义额外的 Logo 模型。
• logo 字段仍然存在于模型实例中，只是在序列化时被排除。

方法二：使用 serialization_alias 和 validation_alias 以及 AliasPath

这种方法适用于 Pydantic v2.0 及以上版本，使用了 serialization_alias 和 validation_alias 属性，以及 AliasPath。serialization_alias 用于指定序列化时使用的字段名，validation_alias 用于指定验证时使用的字段名。AliasPath 用于指定嵌套字段的路径。

NameGPT名称生成器

免费AI公司名称生成器，AI在线生成企业名称，注册公司名称起名大全。

0

查看详情

以下是一个示例：

from pydantic import BaseModel, Field, AliasPath

class Survey(BaseModel):
    logo_url: str = Field(
        ...,
        serialization_alias="logo",
        validation_alias=AliasPath('logo', 'url')
    )

a = Survey.model_validate({'logo': {'url': 'foo'}})
print(a.model_dump(by_alias=True))
# {'logo': 'foo'}

在这个例子中，logo_url 字段使用了 serialization_alias="logo"，这意味着在序列化时，该字段将被命名为 logo。validation_alias=AliasPath('logo', 'url') 指定了验证时，logo_url 字段的值应该从 logo 字典的 url 键中获取。

优点：

• 代码更加简洁。
• 不需要定义额外的模型。
• 可以直接将嵌套字段的值映射到目标字段。

缺点：

• 需要理解 serialization_alias 和 validation_alias 的含义。
• 只适用于 Pydantic v2.0 及以上版本。

总结

本文介绍了两种使用 Pydantic 将现有字段别名指向嵌套键值的方法。第一种方法使用 computed_field 和 Field(exclude=True)，适用于 Pydantic v1.0 及以上版本。第二种方法使用 serialization_alias 和 validation_alias 以及 AliasPath，适用于 Pydantic v2.0 及以上版本。选择哪种方法取决于您的具体需求和 Pydantic 版本。在实际应用中，建议根据 API 响应的结构和复杂度，选择最合适的方法来实现字段别名。

注意事项：

• 在使用 serialization_alias 时，需要确保在调用 model_dump() 方法时，设置 by_alias=True，才能使别名生效。
• 在使用 validation_alias 时，需要确保使用 model_validate() 方法来验证数据，才能使别名生效。
• 在处理复杂的嵌套结构时，可以组合使用 AliasPath 和其他 Pydantic 功能，例如 validator，来实现更灵活的字段别名。

以上就是Pydantic 别名技巧：将现有字段别名指向嵌套键值的详细内容，更多请关注php中文网其它相关文章！