如何从嵌套的pydantic类中获取值？

Question

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':'aeajhs@gmail.com'}, 
        {'mailid':2,'email':'aeajhsds@gmail.com'}
    ]
}
    
userobj = User(**data)

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

当我尝试使用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='aeajhs@gmail.com'), mail(mailid=2, email='aeajhsds@gmail.com')]

我的愿望输出必须是当我想要打印电子邮件时，我必须得到'aeajhs@gmail.com','aeajhsds@gmail.com'

Answer 1

看起来您正在尝试访问类而不是实例

这里有一种方法：

from pydantic import BaseModel
from typing import List

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

one_mail = {"mailid": 1, "email": "aeajhs@gmail.com"}

mail = Mail(**one_mail)

print(mail)
# mailid=1 email='aeajhs@gmail.com'

有了这些，让我们调整User模型：

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

data = {
    "id": 123, 
    "name": "Jane Doe",
    "mails":[
        {"mailid": 1, "email": "aeajhs@gmail.com"}, 
        {"mailid": 2, "email": "aeajhsds@gmail.com"}
    ]
}

userobj = User(**data)

print(userobj.mails)
# [Mail(mailid=1, email='aeajhs@gmail.com'), Mail(mailid=2, email='aeajhsds@gmail.com')]

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

print(userobj.mails[0].email)
# aeajhs@gmail.com

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

print([mail.email for mail in userobj.mails])
# ['aeajhs@gmail.com', 'aeajhsds@gmail.com']

作为旁注：Pydantic支持email string validation