溫馨提示×

FastAPI中怎么實現數據庫集成

小億
119
2024-05-11 11:28:56
欄目: 大數據

要在FastAPI中實現數據庫集成,可以使用第三方庫如SQLAlchemy或者ORM(對象關系映射)。以下是使用SQLAlchemy的基本步驟:

  1. 安裝SQLAlchemy庫:
pip install SQLAlchemy
  1. 創建一個數據庫連接:
from sqlalchemy import create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

SQLALCHEMY_DATABASE_URL = "sqlite:///./test.db"

engine = create_engine(SQLALCHEMY_DATABASE_URL)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

Base = declarative_base()
  1. 創建一個模型類來定義數據庫表結構:
from sqlalchemy import Column, Integer, String

class User(Base):
    __tablename__ = "users"

    id = Column(Integer, primary_key=True, index=True)
    name = Column(String)
    email = Column(String)
  1. 創建數據庫表:
Base.metadata.create_all(bind=engine)
  1. 在FastAPI應用中使用數據庫:
from fastapi import FastAPI, Depends
from sqlalchemy.orm import Session

app = FastAPI()

def get_db():
    db = SessionLocal()
    try:
        yield db
    finally:
        db.close()

@app.get("/users/{user_id}")
def read_user(user_id: int, db: Session = Depends(get_db)):
    user = db.query(User).filter(User.id == user_id).first()
    return {"id": user.id, "name": user.name, "email": user.email}

通過以上步驟,就能在FastAPI應用中實現數據庫集成,實現對數據庫的增刪改查操作。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女