社区首页 >专栏 >SQLAlchemy学习-9.一对多和多对一关系

SQLAlchemy学习-9.一对多和多对一关系

上海-悠悠

发布于 2022-08-26 07:55:18

3.3K00

代码可运行

文章被收录于专栏：从零开始学自动化测试从零开始学自动化测试

运行总次数：0

代码可运行

前言

一对多和多对一关系

一对多关系

一对多关系表设计，一个Parent类关联多个Child类

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import create_engine, Column, Integer, String, ForeignKey
from sqlalchemy.orm import relationship

# 拼接配置dialect + driver://username:passwor@host:port/database
DB_URI = 'mysql+pymysql://root:123456@localhost:3306/web'
Base = declarative_base()

class Parent(Base):  # 一
    __tablename__ = 'parent'
    id = Column(Integer, primary_key=True)
    name = Column(String(64), nullable=False)
    full_name = Column(String(64))
    # 在父表类中通过 relationship() 方法来引用子表的类集合
    children = relationship("Child")

class Child(Base):  # 多
    __tablename__ = 'child'
    id = Column(Integer, primary_key=True)
    name = Column(String(64), nullable=False)
    full_name = Column(String(64))
    # 在子表类中通过 foreign key (外键)引用父表的参考字段
    parent_id = Column(Integer, ForeignKey('parent.id'))

if __name__ == '__main__':
    engine = create_engine(DB_URI)
    Base.metadata.create_all(engine)  # 将模型映射到数据库中

与一对一关系主要区别是relationship不用设置 uselist=False参数

同步后生成的表

新增数据

添加数据

from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine

engine = create_engine('mysql+pymysql://root:123456@localhost:3306/web')
# 把当前的引擎绑定给这个会话
Session = sessionmaker(bind=engine)
# 实例化
session = Session()

# 添加数据
parent = Parent(name='yy', full_name='yoyo')
parent.children = [
    Child(name='c1', full_name='c1 full name')
]
session.add(parent)
session.commit()

# 如果parent 已经存在
parent = session.query(Parent).filter(Parent.name == 'yy')
children = [
    Child(name='c4', full_name='c4 full name', parent_id=parent.first().id),
    Child(name='c5', full_name='c5 full name', parent_id=parent.first().id),
]
session.add_all(children)
session.commit()

查询数据

通过父类，查询子类（单向查询）

    parent = session.query(Parent).get(1)
    print(parent.children)

多对一关系

多对一关系相比上面的一对多而言是双向的关系在最新版本的 sqlalchemy 中对 relationship 引进了 back_populates 参数，两个参数的效果完全一致。

backref 和 back_populates 两个参数的区别

backref 只需要在 Parent 类中声明 children，Child.parent 会被动态创建。
back_populates 必须在两个类中显式地使用 back_populates，更显繁琐，理解更直观

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import create_engine, Column, Integer, String, ForeignKey
from sqlalchemy.orm import relationship
# 拼接配置dialect + driver://username:passwor@host:port/database
DB_URI = 'mysql+pymysql://root:123456@localhost:3306/web'
Base = declarative_base()

class Parent(Base):  # 一
    __tablename__ = 'parent'
    id = Column(Integer, primary_key=True)
    name = Column(String(64), nullable=False)
    full_name = Column(String(64))
    # 父类的 relationship() 方法中使用 relationship.back_populates 参数
    children = relationship("Child", back_populates="parent", lazy="dynamic")

class Child(Base):  # 多
    __tablename__ = 'child'
    id = Column(Integer, primary_key=True)
    name = Column(String(64), nullable=False)
    full_name = Column(String(64))
    parent_id = Column(Integer, ForeignKey('parent.id'))
    parent = relationship("Parent", order_by='Parent.id', back_populates="children")
    # 子表类中附加一个 relationship() 方法

if __name__ == '__main__':
    engine = create_engine(DB_URI)
    Base.metadata.create_all(engine)  # 将模型映射到数据库中

查询数据（双向查询）

通过父类查询子类,或子类查询父类

from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine


engine = create_engine('mysql+pymysql://root:123456@localhost:3306/web')
# 把当前的引擎绑定给这个会话
Session = sessionmaker(bind=engine)
# 实例化
session = Session()

parent = session.query(Parent).get(1)

print(parent.children.all())
children = session.query(Child).get(1)
print(children.parent)