from fastapi import FastAPI, Depends, HTTPException
from tortoise.contrib.fastapi import register_tortoise
from models import User
from schemas import UserCreate
app = FastAPI()
# 初始化数据库
register_tortoise(
app,
db_url="sqlite://db.sqlite3",
modules={"models": ["models"]},
generate_schemas=True,
add_exception_handlers=True,
)
@app.post("/users/", status_code=201)
async def create_user(user_data: UserCreate):
# 密码哈希处理(实际项目应使用passlib)
hashed_password = f"hashed_{user_data.password}"
try:
user = await User.create(
username=user_data.username,
hashed_password=hashed_password,
email=user_data.email
)
except Exception as e:
raise HTTPException(
status_code=400,
detail="Username already exists"
)
return {
"id": user.id,
"username": user.username,
"email": user.email
}
@app.get("/users/{user_id}")
async def get_user(user_id: int):
user = await User.get_or_none(id=user_id)
if not user:
raise HTTPException(status_code=404, detail="User not found")
return {
"id": user.id,
"username": user.username,
"email": user.email,
"created_at": user.created_at.isoformat()
}
复制代码
课后Quiz
题目1:以下哪种场景最适合使用异步ORM?
A) 单用户的桌面应用程序
B) 需要处理数千并发哀求的API服务
C) 执行复杂事件的财务系统
D) 数据堆栈的批量数据处理 答案:B
解析:异步ORM在高并发IO密集型场景下能明显提升吞吐量,而ACD场景更多需要的是事件完整性或盘算能力。 题目2:如何避免在ORM查询时发生SQL注入?
A) 直接拼接字符串
B) 使用ORM的参数化查询
C) 手动过滤特殊字符
D) 限制查询字段长度 答案:B
解析:Tortoise-ORM的查询方法会主动进行参数化处理,有效防止SQL注入,这是最安全的做法。
常见报错解决方案