软件测试|Flask 操作数据库 - 一对多
作者:霍格沃兹测试开发学社
- 2023-09-07 北京
本文字数:1827 字
阅读完需:约 6 分钟
场景示例:
一个班级有多个学生
一个学生只在一个班级
一对多关系中,通过外键来关联数据,外键设置在多的一方,即学生表中
创建表
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())
复制代码
获取更多技术资料,请点击!
划线
评论
复制
发布于: 刚刚阅读数: 6
霍格沃兹测试开发学社
关注
社区:ceshiren.com 微信:ceshiren2021 2019-10-23 加入
微信公众号:霍格沃兹测试开发 提供性能测试、自动化测试、测试开发等资料,实时更新一线互联网大厂测试岗位内推需求,共享测试行业动态及资讯,更可零距离接触众多业内大佬。
评论