FastAPI 创建 SQLAlchemy 会话

在本文中，我们将介绍如何在 FastAPI 中使用 SQLAlchemy 创建会话（session）。会话是与数据库交互的主要组件，它允许我们执行查询、插入和更新数据库中的数据。

SQLAlchemy 简介

SQLAlchemy 是一种强大的 Python ORM（对象关系映射）工具，它提供了对多种数据库的支持，并允许我们使用 Python 对象进行数据库操作。通过 SQLAlchemy，我们可以通过定义数据库模型类来创建表格，并使用这些模型类执行数据库操作。

FastAPI 和 SQLAlchemy 集成

FastAPI 是一个现代、快速（高性能）的 Web 框架，它使用了 Python 类型提示和异步机制，能够提供高效的 API 开发体验。

与 SQLAlchemy 相结合，FastAPI 提供了一个强大的工具集，可以轻松地进行数据库操作。我们可以使用 FastAPI 提供的 Depends 类来创建一个可重用的 SQLAlchemy 会话。

创建 SQLAlchemy 会话

在 FastAPI 中，我们可以使用 SQLAlchemy 提供的 sessionmaker 类来创建会话。我们首先需要导入 sessionmaker，以及 SQLAlchemy 提供的 create_engine 函数。

from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine

# 设置数据库连接字符串
SQLALCHEMY_DATABASE_URL = "sqlite:///./test.db"

# 创建数据库引擎
engine = create_engine(SQLALCHEMY_DATABASE_URL)

# sessionmaker 是一个类，接收一个 engine 作为参数并返回一个会话类
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

在上述代码中，我们首先定义了数据库连接字符串 SQLALCHEMY_DATABASE_URL，它指定了数据库的类型、位置和名称。然后，我们使用 create_engine 函数创建了一个数据库引擎。接下来，我们通过 sessionmaker 类创建了一个会话类 SessionLocal，该类与我们之前创建的引擎绑定在一起。

使用 SQLAlchemy 会话

我们可以通过在 FastAPI 中使用 Depends 类来创建一个可重用的 SQLAlchemy 会话。

from fastapi import Depends
from sqlalchemy.orm import Session

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

在上述代码中，我们定义了一个名为 get_db 的函数，它返回一个会话对象。通过使用 yield 关键字，我们可以在函数返回之前使用这个会话对象。最后，在 finally 块中，我们关闭了会话。

在 FastAPI 中使用 SQLAlchemy 会话

现在我们可以在 FastAPI 的路径操作函数中使用 get_db 函数来访问数据库会话了。

from fastapi import FastAPI, Depends
from sqlalchemy.orm import Session

app = FastAPI()

@app.get("/users/")
def get_users(db: Session = Depends(get_db)):
    users = db.query(User).all()
    return users

在上述代码中，我们定义了一个 GET 请求的路径操作函数 /users/，并使用 db: Session = Depends(get_db) 参数来获取数据库会话。在函数体中，我们使用会话对象 db 执行了一个查询操作，获取了所有用户的数据，并将其返回。

总结

通过本文的介绍，我们了解了如何在 FastAPI 中使用 SQLAlchemy 创建会话。会话是进行数据库操作的核心组件，它允许我们执行查询、插入和更新数据库中的数据。通过 FastAPI 提供的依赖注入功能，我们可以轻松地在路径操作函数中使用会话对象进行数据库操作。这种结合了 FastAPI 和 SQLAlchemy 的开发方式，能够使我们更加高效地进行 Web API 的开发工作。

希望通过本文的学习，你对 FastAPI 和 SQLAlchemy 的集成有所了解，并能够在实际的项目中灵活运用。

如果想深入学习 FastAPI 或 SQLAlchemy 的相关知识，建议查阅官方文档和相关教程，进一步提升自己的开发技能。