FastAPI 中如何与 SQLAlchemy 集成？

推荐答案

在 FastAPI 中与 SQLAlchemy 集成，通常需要以下步骤：

1. 安装依赖：首先需要安装 fastapi 和 sqlalchemy 库，以及一个数据库驱动（如 psycopg2 用于 PostgreSQL 或 pymysql 用于 MySQL）。

   pip install fastapi sqlalchemy psycopg2

2. 创建数据库模型：使用 SQLAlchemy 的 ORM 定义数据库模型。

   // www.javascriptcn.com code example
   from sqlalchemy import Column, Integer, String, create_engine
   from sqlalchemy.ext.declarative import declarative_base
   from sqlalchemy.orm import sessionmaker
   
   SQLALCHEMY_DATABASE_URL = "postgresql://user:password@localhost/dbname"
   
   engine = create_engine(SQLALCHEMY_DATABASE_URL)
   SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
   
   Base = declarative_base()
   
   class User(Base):
       __tablename__ = "users"
       id = Column(Integer, primary_key=True, index=True)
       name = Column(String, index=True)
       email = Column(String, unique=True, index=True)

3. 创建 FastAPI 应用：在 FastAPI 应用中集成 SQLAlchemy，并使用依赖注入来管理数据库会话。

   // www.javascriptcn.com code example
   from fastapi import Depends, FastAPI
   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 user

4. 运行应用：启动 FastAPI 应用，并确保数据库连接正常。

   uvicorn main:app --reload

本题详细解读

1. 安装依赖

在 FastAPI 中与 SQLAlchemy 集成，首先需要安装必要的依赖库。fastapi 是 FastAPI 框架本身，sqlalchemy 是 SQLAlchemy ORM 库，而 psycopg2 是 PostgreSQL 的数据库驱动。如果你使用的是其他数据库，如 MySQL，则需要安装相应的驱动，如 pymysql。

2. 创建数据库模型

使用 SQLAlchemy 的 ORM 功能，我们可以通过定义 Python 类来映射数据库表。Base 是 SQLAlchemy 的基类，所有模型类都需要继承它。SessionLocal 是一个会话工厂，用于创建数据库会话。

3. 创建 FastAPI 应用

在 FastAPI 应用中，我们使用依赖注入来管理数据库会话。get_db 函数是一个依赖项，它会在每个请求中创建一个新的数据库会话，并在请求结束时关闭会话。通过 Depends(get_db)，我们可以将数据库会话注入到路由处理函数中。

4. 运行应用

最后，使用 uvicorn 启动 FastAPI 应用。--reload 参数使得在代码更改时自动重新加载应用，方便开发调试。

通过以上步骤，FastAPI 与 SQLAlchemy 的集成就完成了。这种集成方式使得在 FastAPI 中使用 SQLAlchemy 进行数据库操作变得非常方便和高效。