FastAPI教程-集成SQLAlchemy:在路由中执行添加数据

实现

在路由函数中,注入依赖,依赖传入 get_db,调用增删改查函数时,传入 db

修改 main.py 内容如下:

@app.post('/user')
async def save_user(username:str,password:str, db = Depends(get_db)):
    user = User(username = username,password = password)
    await save(db,user)
    return {
        "code":200,
        "msg":"添加成功!"
    }

完整代码

from fastapi import FastAPI, Depends

from core.database import get_db
from crud.user import save
from models.user import User

app = FastAPI()

@app.post('/user')
async def save_user(username:str,password:str, db = Depends(get_db)):
    user = User(username = username,password = password)
    await save(db,user)
    return {
        "code":200,
        "msg":"添加成功!"
    }

测试