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

2024-11-04
北京
本文字数：1853 字
阅读完需：约 6 分钟

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

场景示例：

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

创建表

from flask import Flask# 实例化 Flask的类，并且绑定modulefrom flask_sqlalchemy import SQLAlchemyfrom flask_sqlalchemy.session import Sessionfrom 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())

复制代码