【Flask】ORM的关联关系

飞熊 • 2023年3月5日上午10:28 • Python • 阅读 69

1、表的外键关联

使用SQLAlchemy创建外键非常简单。在从表中增加一个字段，指定这个字段外键的是哪个表的哪个字段就可以了。从表中外键的字段，必须和主表的主键字段类型保持一致。

特别注意1：这种关联只关注数据表之间的外键关联，不考虑Python对象之间的关联关系.

前提需要连接数据库、创建数据库引擎、创建基类、创建session

from sqlalchemy import create_engine, Column, Integer, ForeignKey, String, TEXT, Boolean, DATE, DECIMAL
from sqlalchemy.ext.declarative import declarative_base
from datetime import date
from sqlalchemy.orm import sessionmaker
# 数据库的配置变量
HOSTNAME = '127.0.0.1'
PORT = '3306'
DATABASE = 'test'
USERNAME = 'root'
PASSWORD = 'root'
DB_URI = 'mysql+pymysql://{}:{}@{}:{}/{}?charset=utf8mb4'.format(USERNAME, PASSWORD, HOSTNAME, PORT, DATABASE)

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

#创建一个基类,所有的模型类继承Base
Base=declarative_base(engine)

#创建session
session=sessionmaker(engine)()  #注意sessionmaker 返回一个函数

创建主表模型类：

class Dept(Base):
    __tablename__='t_dept'
    dept_no=Column(name='dept_no',type_=Integer,primary_key=True,autoincrement=True)
    dept_name=Column(name='dept_name',type_=String(50),unique=True)
    city=Column(name='city',type_=String(50))

    def __str__(self):
        return f'{self.dept_no}-{self.dept_name}-{self.city}'

创建从表模型类：

class Emp(Base):
    __tablename__='t_emp'
    emp_no=Column(name='emp_no',type_=Integer,primary_key=True,autoincrement=True)
    emp_name=Column(name='emp_name',type_=String(50))
    job=Column(name='job',type_=String(50))
    hire_date=Column(name='hire_date',type_=DATE)
    sal=Column(name='sal',type_=DECIMAL(10,2))
    dept_no=Column(ForeignKey(t_dept.dept_no,ondelete='CASCADE'),name='dept_no',type_=Integer)

    def __str__(self):
        return f'{self.emp_no}{self.emp_name}{self.job}{self.hire_date}{self.sal}'

特别注意：设置外键关联，在从表中增加一个字段，指定这个字段外键的是哪个表的哪个字段就可以了。从表中外键的字段，必须和主表的主键字段类型保持一致。

dept_no=Column(ForeignKey(t_dept.dept_no,ondelete='CASCADE'),name='dept_no',type_=Integer)

根据模型类生成数据表

#根据模型类建表
Base.metadata.create_all()
#删除表
#Base.metadata.drop_all()

插入数据

#插入数据
d1=Dept(dept_name='信息部',city='北京')
session.add(d1)
session.commit()

e1=Emp(emp_name='kobe',job='销售',hire_date='2020-12-12',sal='8888.888',dept_no=1)
session.add(e1)
session.commit()

2、外键的删除选项

RESTRICT：若子表中有父表对应的关联数据，删除父表对应数据，会阻止删除。默认项
NO ACTION：在MySQL中，同RESTRICT。
CASCADE：级联删除。删除主表数据时，连通一起删除外键表的数据
SET NULL：父表对应数据被删除，子表对应数据项会设置为NULL。

举例：CASCADE

主表

从表

删除主表中dept_no为1的员工信息

d1=session.query(Dept).filter(Dept.dept_no==1).first()
session.delete(d1)
session.commit()

执行完上述代码后，从表中的关联数据一起删除

文章由极客之音整理，本文链接：https://www.bmabk.com/index.php/post/123407.html