首页 > 编程问答 >基于父模型归档的 Pydantic 联合判别器

基于父模型归档的 Pydantic 联合判别器

时间:2024-09-25 03:41:14浏览次数:1  
标签:python python-3.x pydantic pydantic-v2

我有这样的模型:

class Foo(BaseModel):
    protocol: str
    protocol_params: Union[ProtocolOneParam, ProtocolTwoParam]

ProtocolOneParam ProtocolTwoParam 没有具有可区分值的相同字段,因此我可以将它们用作 Discriminator ,而我可以理解哪个模型应该用于 protocol_params 的唯一方法是通过值 protocol 可以是 "protocol1" "protocol2"

if protocol == "protocol1" protocol_params 应该使用 ProtocolOneParam 模型进行反序列化等等。

在实际用例中,有超过 5 个协议,其中没有任何可区分的字段。

pydantic 有什么方法可以实现我所需要的吗?


当然,Pydantic 可以实现想要的功能。可以使用 @validator root_validator 来根据 protocol 字段的值动态设置 protocol_params 的类型。

from typing import Union, Literal
from pydantic import BaseModel, validator

class ProtocolOneParam(BaseModel):
    param1: str

class ProtocolTwoParam(BaseModel):
    param2: int

class Foo(BaseModel):
    protocol: Literal["protocol1", "protocol2"]
    protocol_params: Union[ProtocolOneParam, ProtocolTwoParam]

    @validator('protocol_params', pre=True)
    def validate_protocol_params(cls, value, values):
        if values['protocol'] == "protocol1":
            return ProtocolOneParam(**value)
        elif values['protocol'] == "protocol2":
            return ProtocolTwoParam(**value)
        else:
            raise ValueError(f"Invalid protocol: {values['protocol']}")

# 示例
foo1 = Foo(protocol="protocol1", protocol_params={"param1": "hello"})
foo2 = Foo(protocol="protocol2", protocol_params={"param2": 123})

print(foo1) # protocol='protocol1', protocol_params=ProtocolOneParam(param1='hello')
print(foo2) # protocol='protocol2', protocol_params=ProtocolTwoParam(param2=123)

解释:

  1. 使用 Literal 类型限制 protocol 字段的取值范围,使其只能是 "protocol1" "protocol2"
  2. @validator('protocol_params', pre=True) 装饰器定义了一个验证器,它会在 protocol_params 字段被赋值之前运行。
  3. 在验证器中,我们根据 values['protocol'] 的值,将 value 参数(即原始数据)转换成对应的 ProtocolOneParam ProtocolTwoParam 对象。
  4. 如果 protocol 的值不匹配,则抛出 ValueError 异常。

对于拥有更多协议的情况,可以扩展 validate_protocol_params 验证器,根据 protocol 的值选择对应的模型。

这种方法可以有效地解决的问题,并且易于扩展和维护。

标签:python,python-3.x,pydantic,pydantic-v2
From: 77887248

相关文章

  • 如何在 Python 中获得类似 Cron 的调度程序?
    我正在寻找一个Python库,它将提供at和cron类似的功能。我很想有一个纯Python解决方案,而不是依赖于安装在盒子上的工具;这样我就可以在没有cron的机器上运行。对于那些不熟悉cron的人:您可以根据如下表达式来安排任务:02**7/usr/bin/run......
  • 我找不到一种方法让我的 python print 语句在几秒钟后自行删除
    我正在尝试制作一款基于文本的冒险游戏,我希望能够让文本自行删除。举个例子,游戏将使用print()语句打印文本,5秒后文本将被删除或对玩家隐藏。我试图找到其他人与此问题相关的问题,并且我找不到任何信息。我不知道该使用什么命令或与之相关的任何内容,请帮忙。Ivetriedtof......
  • 在 python 中可视化四元数
    我在无人机上安装了一个IMU,每0.1秒收集一次四元数数据(w,x,y,z)。现在我想将四元数数据与实际的无人机方向(视频数据)进行比较。所以我想创建某种盒子对象来显示基于四元数数据的方向。我实现了以下教程,将四元数转换为欧拉以进行可视化:https://www.youtube.com/watch?......
  • 有没有办法在 python 中获取特定的键盘输入
    我正在为学校开发一个项目,它目前有行输入(“按Enter继续”),它可以工作,但只要用户按Enter键程序继续,输入是什么并不重要,而且我我希望它仅在按下特定键时才起作用。我查了一下,曾经有一个键盘模块,但由于某种莫名其妙的原因它被删除了,那么还有其他方式获取输入吗?你绝对可以......
  • python+flask计算机毕业设计基于微信小程序的法律问题咨询系统设计与实现(程序+开题+论
    本系统(程序+源码+数据库+调试部署+开发环境)带论文文档1万字以上,文末可获取,系统界面在最后面。系统程序文件列表开题报告内容研究背景随着互联网的飞速发展和智能手机的普及,人们获取信息和解决问题的途径日益多样化。在法律服务领域,传统的线下咨询方式已难以满足公众日益增......
  • python+flask计算机毕业设计基于人脸识别的医疗保险系统的设计与实现(程序+开题+论文)
    本系统(程序+源码+数据库+调试部署+开发环境)带论文文档1万字以上,文末可获取,系统界面在最后面。系统程序文件列表开题报告内容研究背景随着科技的飞速发展和人口老龄化的加剧,医疗保险系统面临着前所未有的挑战与机遇。传统医疗保险管理方式依赖于人工审核与纸质记录,不仅效率......
  • python+flask计算机毕业设计基于微信小程序的河南省美食分享平台(程序+开题+论文)
    本系统(程序+源码+数据库+调试部署+开发环境)带论文文档1万字以上,文末可获取,系统界面在最后面。系统程序文件列表开题报告内容研究背景在移动互联网时代,智能手机和社交媒体已成为人们日常生活不可或缺的一部分。微信小程序作为腾讯推出的一种轻量级应用形态,凭借其无需安装、......
  • python+flask计算机毕业设计基于微信小程序的网络文学管理平台(程序+开题+论文)
    本系统(程序+源码+数据库+调试部署+开发环境)带论文文档1万字以上,文末可获取,系统界面在最后面。系统程序文件列表开题报告内容研究背景随着互联网的迅猛发展,网络文学已成为当代文化生活中不可或缺的一部分,它不仅丰富了人们的阅读体验,还促进了文学创作的多元化与普及化。然而......
  • Python不同方式正倒序遍历的时间开销
    fromtimeitimporttimeitli=[iforiinrange(1000000)]deffor_loop(n):#使用for直接遍历ret=0foriinli:ret=li[i]deffor_loop_enumerate(n):#使用enumerate进行遍历ret=0foridx,iinenumerate(li):re......
  • 【入门岛·第2关】python基础
    目录Python实现wordcountVscode连接InternStudiodebug笔记Python实现wordcountimportstringdefwordcount(text):#去掉标点符号,并将文本转换为小写text=text.translate(str.maketrans('','',string.punctuation)).lower()#按空格分割文本为单词......