如何从嵌套的僵尸类中获取值？ [英] How to get values from nested pydantic classes?

问题描述

class mail(BaseModel):
    mailid: int
    email: str
    
class User(BaseModel):
    id: int
    name: str
    mails: List[mail]

data = {
    'id': 123,
    'name': 'Jane Doe',
    'mails':[
        {'mailid':1,'email':'[e-mail]'}, 
        {'mailid':2,'email':'[e-mail]'}
    ]
}
    
userobj = User(**data)

如何获取mail类中email或mailid从mail类中的值？

当我尝试使用print(mail.email)时，它抛出错误为

AttributeError: type object 'mail' has no attribute 'email'

当我使用此命令时，也出现错误

print(userobj.mails.email)

AttributeError: 'list' object has no attribute 'email'

当我使用它时，我得到的数据如下

print(userobj.mails)
[mail(mailid=1, email='[e-mail]'), mail(mailid=2, email='[e-mail]')]

我想要的输出必须是当我想打印电子邮件时我必须收到'[e-mail]','[e-mail]'

推荐答案

您似乎正在尝试访问类而不是实例。

这里有一种方法：

from pydantic import BaseModel
from typing import List

class Mail(BaseModel):
    mailid: int
    email: str

one_mail = {"mailid": 1, "email": "[e-mail]"}

mail = Mail(**one_mail)

print(mail)
# mailid=1 email='[e-mail]'

准备好后，让我们调整User模型：

class User(BaseModel):
    id: int
    name: str
    mails: List[Mail]

data = {
    "id": 123, 
    "name": "Jane Doe",
    "mails":[
        {"mailid": 1, "email": "[e-mail]"}, 
        {"mailid": 2, "email": "[e-mail]"}
    ]
}

userobj = User(**data)

print(userobj.mails)
# [Mail(mailid=1, email='[e-mail]'), Mail(mailid=2, email='[e-mail]')]

如果您只需要其中一个电子邮件地址，则需要指定其中一个，例如：

print(userobj.mails[0].email)
# [e-mail]

为了获得所有电子邮件地址的列表，您需要迭代mails，例如：

print([mail.email for mail in userobj.mails])
# ['[e-mail]', '[e-mail]']

附注：PYDANIC支持email string validation。

这篇关于如何从嵌套的僵尸类中获取值？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！