如何解决Tortoise ORM for Python 没有返回实体的关系Pyndantic,FastAPI
我正在使用 Tortoise ORM 作为异步 orm 库制作一个示例 Fast Api 服务器,但我似乎无法返回我定义的关系。这些是我的关系:
# Category
from tortoise.fields.data import DatetimeField
from tortoise.models import Model
from tortoise.fields import UUIDField,CharField
from tortoise.fields.relational import ManyToManyField
from tortoise.contrib.pydantic import pydantic_model_creator
class Category(Model):
id = UUIDField(pk=True)
name = CharField(max_length=255)
description = CharField(max_length=255)
keywords = ManyToManyField(
"models.Keyword",related_name="categories",through="category_keywords"
)
created_on = DatetimeField(auto_Now_add=True)
updated_on = DatetimeField(auto_Now=True)
Category_dto = pydantic_model_creator(Category,name="Category",allow_cycles = True)
# Keyword
from models.expense import Expense
from models.category import Category
from tortoise.fields.data import DatetimeField
from tortoise.fields.relational import ManyToManyRelation
from tortoise.models import Model
from tortoise.fields import UUIDField,CharField
from tortoise.contrib.pydantic import pydantic_model_creator
class Keyword(Model):
id = UUIDField(pk=True)
name = CharField(max_length=255)
description = CharField(max_length=255)
categories: ManyToManyRelation[Category]
expenses: ManyToManyRelation[Expense]
created_on = DatetimeField(auto_Now_add=True)
updated_on = DatetimeField(auto_Now=True)
class Meta:
table="keyword"
Keyword_dto = pydantic_model_creator(Keyword)
表格已正确创建。将关键字添加到类别时,数据库状态都很好。问题是当我想查询类别并包含关键字时。我有这个代码:
class CategoryRepository():
@staticmethod
async def get_one(id: str) -> Category:
category_orm = await Category.get_or_none(id=id).prefetch_related('keywords')
if (category_orm is None):
raise NotFoundHTTP('Category')
return category_orm
category_orm debug at run-time
哪种告诉我它们已加载。 然后当我不能使用 Pydantic 模型时,我有这个代码
class CategoryUseCases():
@staticmethod
async def get_one(id: str) -> Category_dto:
category_orm = await CategoryRepository.get_one(id)
category = await Category_dto.from_tortoise_orm(category_orm)
return category
调试这个,没有keywords
字段
category (pydantic) debug at run-time
查看函数from_tortoise_orm
的tortoise orm源码
@classmethod
async def from_tortoise_orm(cls,obj: "Model") -> "PydanticModel":
"""
Returns a serializable pydantic model instance built from the provided model instance.
.. note::
This will prefetch all the relations automatically. It is probably what you want.
但是我的关系没有回来。有没有人有类似经历?
解决方法
当您尝试在 Tortoise ORM 初始化之前 生成 pydantic 模型时会出现此问题。如果您查看 basic pydantic 示例,您会看到所有 pydantic_model_creator
都被称为 after Tortoise.init
。
显而易见的解决方案是在 Tortoise 初始化之后创建 pydantic 模型,如下所示:
await Tortoise.init(db_url="sqlite://:memory:",modules={"models": ["__main__"]})
await Tortoise.generate_schemas()
Event_Pydantic = pydantic_model_creator(Event)
或者更方便的方法,通过Tortoise.init_models()
使用早期模型初始化。像这样:
from tortoise import Tortoise
Tortoise.init_models(["__main__"],"models")
Tournament_Pydantic = pydantic_model_creator(Tournament)
在这种情况下,主要思想是将pydantic和db模型拆分为不同的模块,以便导入第一个不会导致提前创建第二个。并确保在创建 pydantic 模型之前调用 Tortoise.init_models()
。
可以在 here 中找到更详细的示例说明。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。