FastAPI Tortois-ORM 字典形式返回模型对象
在FastAPI中,使用Tortoise ORM时,你可能想要以字典形式返回模型对象。Tortoise ORM提供了一个方法dict()
来将模型对象转换为字典。
以下是一个简单的例子:
首先,确保你已经安装了FastAPI和Tortoise ORM。
# 安装依赖
# pip install fastapi[all]
# pip install tortoise-orm[sqlite]
定义你的模型:
from tortoise import fields
from tortoise.models import Model
class User(Model):
id = fields.IntField(pk=True)
name = fields.CharField(max_length=50)
age = fields.IntField()
然后,使用Tortoise ORM的dict()
方法将模型实例转换为字典:
from tortoise.queryset import QuerySet
# 假设已经有了一个User实例
user = await User.get(id=1)
# 将模型实例转换为字典
user_dict = user.dict()
在FastAPI中,你可能会在一个路由中使用这个转换:
from fastapi import FastAPI
from tortoise.transactions import in_transaction
app = FastAPI()
@app.get("/users/{user_id}")
async def get_user(user_id: int):
async with in_transaction() as transaction:
user = await User.get(id=user_id)
if user:
return user.dict()
else:
return {"detail": "User not found"}
在这个例子中,我们定义了一个路由/users/{user_id}
,当访问这个路由时,它会尝试查找并返回一个用户的字典形式表示。如果用户不存在,它会返回一个错误信息。
评论已关闭