软件测试学习笔记丨Flask操作数据库一对多操作

本文转自测试人社区，原文链接：https://ceshiren.com/t/topic/23443

场景示例：

• 一个班级有多个学生
• 一个学生只在一个班级
• 一对多关系中，通过外键来关联数据，外键设置在多的一方，即学生表中

创建表

from flask import Flask
# 实例化 Flask的类，并且绑定module
from flask_sqlalchemy import SQLAlchemy
from flask_sqlalchemy.session import Session
from sqlalchemy import *

app = Flask(__name__)

username = "root"
pwd = "123456"
ip = "127.0.0.1"
port = "3306"
database = "first"
app.config['SQLALCHEMY_DATABASE_URI'] = f'mysql+pymysql://{username}:{pwd}@{ip}:{port}/{database}?charset=utf8'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True

db = SQLAlchemy(app)
session: Session = db.session

# 班级表
class ClassInfo(db.Model):
    __tablename__ = "Class"
    id = Column(Integer, primary_key=True)
    name = Column(String(80))
    # 魔法函数，定义输出格式
    def __repr__(self):
        return f"<Class：{self.name}"


# 学生表
class StudentInfo(db.Model):
    # 指定表名__tablename__属性
    __tablename__ = "Student"
    id = Column(Integer, primary_key=True)
    name = Column(String(80))
    # 添加外键
    class_id = Column(Integer, ForeignKey("Class.id"))
    # 映射，建立关联关系，要创建在多的一方
    # 第一个参数：表示这个关系的另一端是 ClassInfo 类，即所关联表的表名。比如班级id=1
    # 第二个参数：backref, 表示反向引用，需要从班级id为1 反向获取【多】的时候，使用的属性
    # studentinfo是当前表的别名
    classinfo = db.relationship("ClassInfo", backref="stu")
    def __repr__(self):
        return f"<Student：{self.name}"


if __name__ == '__main__':
    with app.app_context():
        db.create_all()
    # ====== 数据新增 ======
    # Class 班级表：添加两条数据
    with app.app_context():
        class1 = ClassInfo(id=1, name="测开21期")
        class2 = ClassInfo(id=2, name="测开22期")
        session.add_all([class1, class2])
        session.commit()

        # Student 学生表：添加两条数据
        student1 = StudentInfo(id=1, name="学生一", class_id=1)
        student2 = StudentInfo(id=2, name="学生二", class_id=1)
        student3 = StudentInfo(id=3, name="学生三", class_id=2)
        student4 = StudentInfo(id=4, name="学生四", class_id=2)
        session.add_all([student1, student2, student3, student4])
        session.commit()
        session.close()

查询操作

多查一：

with app.app_context():
    # 多查一  id为1的学生 对应的班级
    stu1 = StudentInfo.query.filter_by(id=2).first()
    # 直接通过属性获取 class_id
    print(stu1.class_id)
    # 反向获取 class类的属性
    print(stu1.classinfo.id)
    print(stu1.classinfo.name)

一查多：

with app.app_context():
    # 一查多，通过查询class的id去找到学生信息
    class1 = ClassInfo.query.filter_by(id=1).first()
    print(class1.stu[0].name)

修改操作

一改多：

with app.app_context():
    # 一改多，通过 班级 修改 学生 信息
    class1 = ClassInfo.query.filter_by(id=1).first()
    print(class1.stu)
    class1.stu[0].name = "学生一修改完成"
    session.commit()
    session.close()

多改一：

with app.app_context():
        # 多改一，通过 学生 修改 班级 信息
        stu2 = StudentInfo.query.filter_by(id=2).first()
        print(stu2.classinfo.name)
        stu2.classinfo.name = "就业班"
        session.commit()
        session.close()

删除操作

一般情况是用一删除多，不会用多删除一的方式

with app.app_context():
        # 删除用的很少，几乎不用，了解即可
        # 删除一个班级下所有的学生
        class1 = ClassInfo.query.filter_by(id=1).first()
        print(StudentInfo.class_id)
        # 注意这里用的是filter 不是filter_by
        StudentInfo.query.filter(StudentInfo.class_id == class1.id).delete()
        session.commit()
        session.close()
        # 删除之后查询结果
        print(StudentInfo.query.all())