Flask学习三：模型操作

无知的小菜鸡

已于 2023-12-28 15:49:29 修改

阅读量1.3k

点赞数 27

文章标签： flask

于 2023-12-16 11:29:34 首次发布

本文链接：https://blog.csdn.net/weixin_41897680/article/details/134769073

版权

ORM

flask 通过Model操作数据库，不管你的数据库是MySQL还是Sqlite，flask自动帮你生成相应数据库类型的sql语句，所以不需要关注sql语句和类型，对数据的操作flask帮我们自动完成，只需要会写Model就可以了

flask使用对象关系映射（简称为ORM）框架去操控数据库

ORM对象关系映射是一种程序技术，用于实现面向对象变成语言里不同类型系统的数据之间的转换，将对对象的操作转换为原生的SQL语句。简单易用，可以有效减少重复的SQL；性能损耗少；设计灵活，可以轻松实现复杂查询；可移植性好

数据库连接

Flask使用Python自带的ORM：SQLAlchemy。针对于Flask的支持，需要安装相应的插件

pip install flask-sqlalchemy

为什么使用sqlite而不是mysql

SQLite和MySQL是两种常见的数据库管理系统，它们之间有一些重要的区别。

数据库类型：SQLite是一种嵌入式数据库，它将整个数据库作为一个文件存储在磁盘上。而MySQL是一种客户端/服务器数据库，它以独立的服务器进程运行，并通过网络接口来处理客户端请求。
部署和配置：由于SQLite是一个嵌入式数据库，它非常容易部署和配置。你只需要在应用程序中包含SQLite的库文件即可。MySQL需要单独安装和配置数据库服务器，并通过网络连接来访问。
性能和扩展性：SQLite适用于小型应用程序和个人项目，因为它在处理大量数据和高并发访问时可能性能较低。相比之下，MySQL被设计用于处理大型数据集和高负载的应用程序，它具有更好的性能和扩展性。
功能和支持：MySQL是一个成熟的数据库管理系统，提供了广泛的功能和丰富的工具。它支持复杂的查询语言（如SQL），事务处理和高级安全特性。相比之下，SQLite是一个轻量级的数据库，功能相对较少。
并发处理：MySQL支持多个并发连接，可以同时处理多个客户端请求。SQLite在写操作时会锁定整个数据库文件，因此在高并发访问情况下可能会出现性能问题。

综上所述，SQLite适用于小型应用程序和个人项目，而MySQL适用于中大型应用程序和高负载环境。选择哪种数据库取决于你的具体需求和项目规模。

连接配置
直接看老师讲的视频好了，容易理解。千峰教育 Flask2入门数据库连接

数据迁移

当数据库和表配置好后，需要进行数据迁移，这样会在项目中生成真正的数据库和表。
基本步骤如下：

使用cmd打开项目目录（app.py所在的目录）
依次输入下面的命令
- flask db init 创建迁移文件夹migrates，只调用一次
- flask db migrate 生成迁移文件
- flask db upgrade 执行迁移文件中的升级
- flask db downgrade 执行迁移文件中的降级

初始化
在这里插入图片描述
执行完会在项目根目录下生成对应的文件夹instance和migrations

在这里插入图片描述
生成迁移文件

会生成数据库db文件

并且会生成一个操作记录的py文件

生成表

实际上就是执行的上一次生成文件里的upgrade函数，执行完成后，我们可以使用类似dberver的工具来进行查看
在这里插入图片描述
修改表结构
在修改表结构时，比如新添加一列。必须要先修改对应的模型，然后在重新进行生成迁移文件、执行迁移文件中的升级。

无法生成迁移文件
如果执行flask db migrate 提示INFO [alembic.env] No changes in schema detected. 这时你可以将你的所有的模型类导入到app.py中，再次执行可以生成对应的迁移文件

数据库的常见操作

新增数据

@baseBlue.route("/add-user/")
def addUser():
    # 创建一个对象
    user = BaseModel()
    user.name = "张三"
    user.email = "2345@163.com"
    user.password = "123"
    user.create_time = datetime.now()
    # 将对象添加进session中
    db.session.add(user)
    # 将数据提交到数据库
    db.session.commit()
    data = {"code": 0, "data": '', "msg": "添加成功"}
    return jsonify(data)

在这里插入图片描述

如果添加多条数据时可以使用add_all，例如

 users =[]
 for i in range(10):
     user = BaseModel()
     user.name = "张三" + str(i)
     user.email = "2345@163.com"
     user.password = "123"
     user.create_time = datetime.now()
     users.append(user)
 
 db.session.add_all(users)
 db.session.commit()

在这里插入图片描述
同时可以添加异常处理

def addUser():
    # 创建一个对象
    user = BaseModel()
    user.name = "张三"
    user.email = "2345@163.com"
    user.password = "123"
    user.create_time = datetime.now()
    try:
        # 将对象添加进session中
        db.session.add(user)
        # 将数据提交到数据库
        db.session.commit()
        data = {"code": 0, "data": '', "msg": "添加成功"}
    except Exception as e:
        data = {"code": 1, "data": '', "msg": "添加失败"}
        db.session.rollback()  # 回滚
        db.session.flush()  # 清空session
        print("addUser error："+str(e))
    return jsonify(data)

删除数据

def delUser(id):
    try:
        user = BaseModel.query.get(id)
        db.session.delete(user)
        db.session.commit()
        data = {"code": 0, "data": '', "msg": "删除成功"}
    except Exception as e:
        data = {"code": 1, "data": '', "msg": "删除失败"}
        db.session.rollback()  # 回滚
        db.session.flush()  # 清空session
        print("delUser error："+str(e))
    return jsonify(data)

修改数据

def updateUser(id):
    try:
        user = BaseModel.query.get(id)
        user.name="李四"
        db.session.commit()
        data = {"code": 0, "data": '', "msg": "修改成功"}
    except Exception as e:
        data = {"code": 1, "data": '', "msg": "修改失败"}
        db.session.rollback()  # 回滚
        db.session.flush()  # 清空session
        print("updateUser error："+str(e))
    return jsonify(data)

查询数据

过滤器

filter() 把过滤器添加到原查询上，返回一个新的查询
limit() 使用指定的值限制原查询返回的结果数量，返回一个新的查询
offset() 偏移原查询返回的结果，返回一个新的查询
order_by() 根据指定条件对原查询结果进行排序，返回一个新查询
group_by() 根据指定条件对原查询结果进行分组，返回一个新的查询

常用查询

all() 以列表形式返回查询的所有结果，返回列表
first() 返回查询的第一个结果，如果没有结果返回None
get() 返回指定主键对应的行，如果没有对应的行，则返回None
count() 返回查询结果的数量
paginate() 返回一个Paginate对象，它包含指定范围内的结果

逻辑运算

and_ 逻辑与，filter()默认为逻辑与，多个条件时用逗号分隔
or_ 逻辑或
not_ 逻辑非

查询属性

contains() ， contains('3') 模糊查找，类似于sql中的like
in_()， in_([1,2,3]) 其中之一
startswith() 和 endswidth() 以什么开始和以什么结束

根据用户名模糊查询

def queryUser(name):
    try:
        users = BaseModel.query.filter(BaseModel.name.like("%"+name+"%")).all()
        res = [user.to_dict() for user in users]
        data = {"code": 0, "data": res, "msg": "查询成功"}
    except Exception as e:
        data = {"code": 1, "data": '', "msg": "查询失败"}
        print("queryUser error："+str(e))
    return jsonify(data)

注意：查询返回的结果是模型对象，因此需要将模型对象转换为字典

class BaseModel(db.Model):
    # 表的名字
    __tablename__ = 'base_user'

    # 表的结构
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)  # 主键
    name = db.Column(db.String(20), unique=False, nullable=False)  # 用户名
    email = db.Column(db.String(20), unique=True, nullable=False)  # 邮箱
    password = db.Column(db.String(20), nullable=False)  # 密码
    rank = db.Column(db.String(20), nullable=False,
                     default='user')  # 用户等级，admin为管理员，user为普通用户
    create_time = db.Column(db.DateTime, nullable=False)  # 创建时间

    # 将模型对象转换为字典
    def to_dict(self):
        return {c.name: getattr(self, c.name) for c in self.__table__.columns}

在这里插入图片描述
query.filter就相当于sql中的where查询，如果存在多个查询条件时可以如下操作

from sqlalchemy import and_, or_

# 示例查询条件
condition1 = BaseModel.column1 == value1
condition2 = BaseModel.column2 > value2
condition3 = BaseModel.column3.like("%" + value3 + "%")

# 使用 and_ 方法将多个条件组合为与关系
query = BaseModel.query.filter(and_(condition1, condition2, condition3))

# 或者使用 or_ 方法将多个条件组合为或关系
query = BaseModel.query.filter(or_(condition1, condition2, condition3))

# 执行查询
results = query.all()

或者

from sqlalchemy import or_

conditions = [
    User.username == 'user1',
    User.id > 5,
    # ...
]

query = User.query
query = query.filter(or_(*conditions))

users = query.all()

分页查询

SQLAlchemy内部提供了paginate方法来实现分页的功能。

# 分页查询
@baseBlue.route("/get-all-user", methods=['POST'])
def getAllUser():
    # 获取前端传递过来的参数
    page = int(request.form.get("page"))
    page_size = int(request.form.get("pageSize"))
    p = BaseModel.query.paginate(
        page=page, per_page=page_size, error_out=False)
    # 将模型对象转换为字典
    data = [item.to_dict() for item in p.items]
    res = {
        "code": 0,
        "data": {
            "total": p.total,
            "page": p.page,
            "pageSize": p.per_page,
            "list": data
        },
        "msg": "查询成功"
    }
    return jsonify(res)

paginate对象的属性：

items：返回当前页的内容列表
has_next：是否还有下一页
has_prev：是否还有上一页
next(error_out=False)：返回下一页的Pagination对象
prev(error_out=False)：返回上一页的Pagination对象
page：当前的页码（从1）开始
pages：总页数
pre_page：每页显示的数量
prev_num：上一页页码数
next_num：下一页页码数
total：查询返回的记录总数

在这里插入图片描述

多表操作

一对多

学生、班级表

class Grade(db.Model):
    # 表名
    __tablename__ = 'grade'
    # 表结构
    id = db.Column(db.Integer, primary_key=True, autoincrement=True, doc="主键")
    name = db.Column(db.String(10), unique=True, nullable=False, doc="班级名称")
    # 建立联系
    # 参数1：表名，类的名称，这里要用字符串
    # 参数2：反向引用的名称，grade对象，让student去反过来得到grade对象的名称
    #                      这样Student表可以直接使用Grade中的字段，而不是使用班级id再去表里查询
    # 参数3：懒加载
    # 这里的students不是表字段，是一个类属性
    students = db.relationship('Student', backref="grade", lazy=dynamic)


# 学生表
class Student(db.Model):
    __tablename__ = 'student'
    # 字段
    id = db.Column(db.Integer, primary_key=True, autoincrement=True, doc="主键")
    name = db.Column(db.String(10), unique=True, nullable=False, doc="学生姓名")
    grade_id = db.Column(db.Integer, db.ForeignKey(
        'grade.id'), nullable=False, doc="班级id")

lazy属性

lazy属性：懒加载，可以延迟在使用关联属性的时候才建立关联

lazy='dynamic'：会返回一个query对象（查询集），可以继续使用其他的查询方法，比如all()
lazy='select'：首次访问到属性的时候，就会全部加载该属性的数据
lazy='joined'：在对关联的两个表进行join操作是，从而获取到所有相关的对象
lazy=True：返回一个可用的列表对象，同select

增加数据

def addGradeStudent():
    student = Student()
    student.name = '张三'
    student.grade_id = 1
    try:
        db.session.add(student)
        db.session.commit()
        data = {"code": 0, "data": '', "msg": "添加成功"}
    except Exception as e:
        data = {"code": 1, "data": '', "msg": "添加失败"}
        db.session.rollback()  # 回滚
        db.session.flush()  # 清空session

修改、删除操作基本上也与单表操作一致。需要注意的是，比如在删除班级时，这个班级里有学生。删除后学习表里的学生不会被删除，但是学习对应的班级id会变为null

查询

# 查询学生所在班级的班级名称
def getGradeName():
    try:
        student = Student.query.get(1)
        # 这里使用了反向引用，这样就不许要再查询班级表
        gradeName = student.grade.name
        res = {"code": 0, "data": gradeName, "msg": "查询成功"}
        data = {"code": 0, "data": res, "msg": "查询成功"}
    except Exception as e:
        data = {"code": 1, "data": '', "msg": "查询失败"}
        print("getGradeName error："+str(e))

# 查询班级下的所有学生
    grade = Grade.query.get(1)
    students = grade.students

多对多

多对多需要创建一个中间表，比如用户收藏电影，需要有用户表、收藏表、电影表，收藏表就是这个中间表。

用户表、电影表、收藏表

# 用户表
class User(db.Model):
    __tablename__ = 'user'
    id = db.Column(db.Integer, primary_key=True, autoincrement=True, doc="主键")
    username = db.Column(db.String(10), unique=True, nullable=False, doc="用户名")
    # 类属性，不是表字段。
    collections = db.relationship("Collection", back_populates="user")

# 电影表
class movie(db.Model):
    __tablename__ = 'movie'
    id = db.Column(db.Integer, primary_key=True, autoincrement=True, doc="主键")
    name = db.Column(db.String(10), unique=True, nullable=False, doc="电影名称")
    score = db.Column(db.Float, nullable=False, doc="评分")
    collections = db.relationship("Collection", back_populates="movie")

# 收藏表
class Collection(db.Model):
    id = db.Column(db.Integer, primary_key=True,doc="主键")
    user_id = db.Column(db.Integer, db.ForeignKey("user.id"),doc="用户id")
    movie_id = db.Column(db.Integer, db.ForeignKey("movie.id"),doc="电影id")

    user = db.relationship("User", back_populates="collections")
    movie = db.relationship("Movie", back_populates="collections")