SQLAlchemy
SQLAlchemy 是一个非常流行的 Python ORM,它提供了与各种数据库的集成,并且可以方便地实现数据库操作。FastAPI 提供了与 SQLAlchemy 的集成,可以方便地使用 SQLAlchemy ORM 操作数据库。
下面是一个使用 FastAPI 和 SQLAlchemy 的示例:
代码语言:javascript复制from fastapi import FastAPI
from sqlalchemy.orm import Session
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
SQLALCHEMY_DATABASE_URL = "sqlite:///./test.db"
engine = create_engine(SQLALCHEMY_DATABASE_URL)
Base = declarative_base()
app = FastAPI()
@app.on_event("startup")
async def startup():
Base.metadata.create_all(bind=engine)
@app.on_event("shutdown")
async def shutdown():
Base.metadata.drop_all(bind=engine)
@app.get("/users/{user_id}")
async def read_user(user_id: int, db: Session = Depends(get_db)):
user = db.query(User).filter(User.id == user_id).first()
if not user:
raise HTTPException(status_code=404, detail="User not found")
return user
@app.post("/users/")
async def create_user(user: UserCreate, db: Session = Depends(get_db)):
db_user = User(**user.dict())
db.add(db_user)
db.commit()
db.refresh(db_user)
return db_user
在上面的示例中,我们定义了一个名为 User 的 SQLAlchemy 模型,并在 startup 和 shutdown 事件中创建和删除数据库。然后我们在请求处理函数中使用 get_db 函数获取数据库会话,并使用 SQLAlchemy ORM 操作数据库。