python sqlalchemy 关联两个表查询

原创

mob649e815cb099 2025-01-10 09:28:10 ©著作权

文章标签 数据库 User sqlalchemy 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e815cb099的原创作品，请联系作者获取转载授权，否则将追究法律责任

使用 SQLAlchemy 关联两个表的查询方法

在这篇文章中，我们将学习如何使用 Python 的 SQLAlchemy 库来关联两个表并进行查询。SQLAlchemy 是一个强大的数据库工具，可以帮助我们使用 Python 更方便地与数据库进行交互。

流程概述

我们将通过以下步骤来实现表的关联查询：

步骤	描述
1	安装 SQLAlchemy
2	创建两个表的模型
3	创建数据库并初始化表
4	插入数据
5	进行关联查询

现在让我们逐步进行。

步骤 1：安装 SQLAlchemy

首先，你需要确保你的开发环境中安装了 SQLAlchemy。你可以使用 pip 来安装它。打开你的命令行，并运行以下命令：

pip install sqlalchemy

步骤 2：创建两个表的模型

创建两个关联的表模型。假设我们有一个用户表 User 和一个文章表 Post，其中每个用户可以拥有多篇文章。

# 导入 SQLAlchemy 的基础类和数据类型
from sqlalchemy import create_engine, Column, Integer, String, ForeignKey
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship

# 创建基础类
Base = declarative_base()

# 定义用户模型
class User(Base):
    __tablename__ = 'users'  # 表名

    id = Column(Integer, primary_key=True)  # 主键
    name = Column(String)   # 用户名
    posts = relationship("Post", back_populates="user")  # 关联关系

# 定义文章模型
class Post(Base):
    __tablename__ = 'posts'  # 表名

    id = Column(Integer, primary_key=True)  # 主键
    title = Column(String)  # 文章标题
    user_id = Column(Integer, ForeignKey('users.id'))  # 外键，关联用户ID
    user = relationship("User", back_populates="posts")  # 关联关系

代码说明：

Base 是所有模型类的基类。
User 模型具有 id 和 name 字段，以及一个与 Post 的一对多关联。
Post 模型具有 id、title 和 user_id 字段，以及与 User 的多对一关联。

步骤 3：创建数据库并初始化表

接下来，我们需要创建一个数据库并将表结构写入数据库中。

# 创建数据库引擎
engine = create_engine('sqlite:///example.db')  # 使用 SQLite 数据库

# 创建所有表
Base.metadata.create_all(engine)  # 创建所有模型对应的表

代码说明：

create_engine：定义数据库的连接字符串，在这里我们使用 SQLite 数据库。
create_all：根据模型生成表。

步骤 4：插入数据

现在我们可以在表中插入一些数据。我们将插入一个用户和几篇文章。

from sqlalchemy.orm import sessionmaker

# 创建一个会话
Session = sessionmaker(bind=engine)
session = Session()

# 插入一个用户
new_user = User(name="Alice")
session.add(new_user)

# 插入几篇文章
new_post1 = Post(title="Post 1", user=new_user)
new_post2 = Post(title="Post 2", user=new_user)
session.add(new_post1)
session.add(new_post2)

# 提交会话
session.commit()  # 将所有变更提交到数据库

代码说明：

sessionmaker：创建会话的工厂。
session.add：将新创建的用户和文章添加到会话中。
session.commit：将所有更改保存到数据库。

步骤 5：进行关联查询

现在我们可以查询数据，看看我们存储的内容。我们想要查询每个用户及其撰写的文章。

# 查询所有用户及其文章
users = session.query(User).all()  # 查询所有用户

for user in users:
    print(f'User: {user.name}')  # 打印用户名
    for post in user.posts:  # 遍历用户的所有文章
        print(f'    Post: {post.title}')  # 打印文章标题

代码说明：