title: FastAPI中的复杂查询与原子更新指南
date: 2025/05/02 20:33:32
updated: 2025/05/02 20:33:32
author: cmdragon
excerpt:
FastAPI 结合 Tortoise-ORM 实现复杂查询与原子更新。通过 Q 对象构建多条件查询,支持 AND、OR、NOT 逻辑运算符,动态组合查询条件。使用 F 表达式进行原子更新,避免竞态条件,确保数据一致性。示例包括订单状态与金额的复杂查询、库存扣减的原子操作,以及商品促销的价格更新。常见错误包括字段拼写错误、类型不匹配和空结果集,需通过模型检查和异常处理解决。
categories:
tags:
扫描二维码
关注或者微信搜一搜:编程智域 前端至全栈交流与成长
探索数千个预构建的 AI 应用,开启你的下一个伟大创意:https://tools.cmdragon.cn/
在开始前确保已安装必要依赖:
pip install fastapi uvicorn tortoise-orm pydantic
创建订单模型示例(models.py):
from tortoise.models import Model
from tortoise import fields
class Product(Model):
id = fields.IntField(pk=True)
name = fields.CharField(max_length=255)
stock = fields.IntField(default=0)
price = fields.DecimalField(max_digits=10, decimal_places=2)
class Order(Model):
id = fields.IntField(pk=True)
status = fields.CharField(max_length=20) # pending/completed/canceled
total_amount = fields.DecimalField(max_digits=10, decimal_places=2)
product = fields.ForeignKeyField('models.Product', related_name='orders')
created_at = fields.DatetimeField(auto_now_add=True)
创建对应的Pydantic模型(schemas.py):
from pydantic import BaseModel
from datetime import datetime
class OrderOut(BaseModel):
id: int
status: str
total_amount: float
product_id: int
created_at: datetime
class Config:
orm_mode = True
Q对象是Tortoise-ORM的条件表达式构造器,支持逻辑运算符:
&
表示AND|
表示OR~
表示NOT示例:查询金额大于100且状态为pending的订单
from tortoise.expressions import Q
async def get_orders():
return await Order.filter(
Q(total_amount__gt=100) & Q(status="pending")
).all()
在路由中实现动态过滤(main.py):
from fastapi import APIRouter, Query
from tortoise.expressions import Q
router = APIRouter()
@router.get("/orders", response_model=list[OrderOut])
async def search_orders(
min_amount: float = Query(None),
max_amount: float = Query(None),
status: str = Query(None)
):
query = Q()
if min_amount:
query &= Q(total_amount__gte=min_amount)
if max_amount:
query &= Q(total_amount__lte=max_amount)
if status:
query &= Q(status=status)
return await Order.filter(query).prefetch_related('product')
查询过去7天内金额超过500的已完成订单,或金额低于100的待处理订单:
from datetime import datetime, timedelta
async def complex_query():
seven_days_ago = datetime.now() - timedelta(days=7)
return await Order.filter(
Q(
Q(created_at__gte=seven_days_ago) &
Q(total_amount__gt=500) &
Q(status='completed')
) |
Q(
Q(total_amount__lt=100) &
Q(status='pending')
)
).order_by('-created_at')
F表达式直接在数据库层面执行运算,避免竞态条件。示例:安全扣减库存
from tortoise.expressions import F
async def decrease_stock(product_id: int, quantity: int):
await Product.filter(id=product_id).update(
stock=F('stock') - quantity
)
同时更新多个字段:
async def update_product_price(product_id: int, new_price: float):
await Product.filter(id=product_id).update(
price=new_price,
last_updated=datetime.now(),
version=F('version') + 1
)
只有当库存充足时才允许扣减:
async def safe_purchase(product_id: int, quantity: int):
updated = await Product.filter(
id=product_id,
stock__gte=quantity
).update(stock=F('stock') - quantity)
if not updated:
raise HTTPException(400, "库存不足")
实现商品促销接口:
@router.post("/products/{product_id}/promotion")
async def create_promotion(
product_id: int,
discount: float = Body(..., gt=0, lt=1)
):
# 原子更新价格并记录操作
updated = await Product.filter(id=product_id).update(
price=F('price') * (1 - discount),
promotion_count=F('promotion_count') + 1
)
if not updated:
raise HTTPException(404, "商品不存在")
# 获取更新后的对象
product = await Product.get(id=product_id)
return {
"new_price": float(product.price),
"promotion_count": product.promotion_count
}
FieldError: Unknown field 'total_amout' for model Order
原因:字段名拼写错误(amout → amount)
解决:检查模型定义和查询字段是否一致
ValidationError: 1 validation error for OrderOut...
原因:Decimal字段自动转换为float时精度丢失
解决:在Pydantic模型中使用Decimal
类型并配置json_encoders
DoesNotExist: Object does not exist
原因:查询条件过于严格导致无结果
解决:添加异常处理或使用first()
代替get()
当需要同时满足三个条件时,Q对象应该如何组合?
A) Q(a) | Q(b) | Q(c)
B) Q(a) & Q(b) & Q(c)
C) Q(a) & (Q(b) | Q(c))
答案:B。& 运算符用于AND条件组合
为什么要使用F表达式而不是先查询再更新?
A) 减少数据库查询次数
B) 避免并发导致的数据不一致
C) 两种方式效果相同
答案:B。F表达式保证原子操作,防止竞态条件
如何防止通过Q对象构造的查询出现SQL注入?
A) 手动转义参数
B) 使用ORM的内置参数化查询
C) 限制用户输入字段
答案:B。Tortoise-ORM会自动处理查询参数化
余下文章内容请点击跳转至 个人博客页面 或者 扫码关注或者微信搜一搜:编程智域 前端至全栈交流与成长
,阅读完整的文章:FastAPI中的复杂查询与原子更新指南 | cmdragon's Blog
参与评论
手机查看
返回顶部