FastAPI中怎么实现数据库集成-开发者知识库平台

FastAPI中怎么实现数据库集成

FastAPI

1054

2024/5/3 11:28:56

栏目: 大数据

要在FastAPI中实现数据库集成，可以使用第三方库如SQLAlchemy或者ORM（对象关系映射）。以下是使用SQLAlchemy的基本步骤：

安装SQLAlchemy库：

pip install SQLAlchemy

创建一个数据库连接：

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()

创建一个模型类来定义数据库表结构：

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)

创建数据库表：

Base.metadata.create_all(bind=engine)

在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应用中实现数据库集成，实现对数据库的增删改查操作。

辰迅云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>

FastAPI中怎么实现数据库集成

最新知识库

相关标签