Flask之数据库操作

Posted 2021-01-12 alexzhang92

4.2 数据库基本操作

在Flask-SQLAlchemy中，插入、修改、删除操作，均由数据库会话管理。会话用db.session表示。在准备把数据写入数据库前，要先将数据添加到会话中然后调用commit()方法提交会话。

数据库会话是为了保证数据的一致性，避免因部分更新导致数据不一致。提交操作把会话对象全部写入数据库，如果写入过程发生错误，整个会话都会失效。

数据库会话也可以回滚，通过db.session.rollback()方法，实现会话提交数据前的状态。

在Flask-SQLAlchemy中，查询操作是通过query对象操作数据。最基本的查询是返回表中所有数据，可以通过过滤器进行更精确的数据库查询。

将数据添加到会话中示例：

user = User(name=‘python‘)
db.session.add(user)
db.session.commit()

 

在视图函数中定义模型类

from flask import Flask
from flask_sqlalchemy import SQLAlchemy


app = Flask(__name__)

#设置连接数据库的URL
app.config[‘SQLALCHEMY_DATABASE_URI‘] = ‘mysql://root:[email protected]:3306/Flask_test‘

#设置每次请求结束后会自动提交数据库中的改动
app.config[‘SQLALCHEMY_COMMIT_ON_TEARDOWN‘] = True

app.config[‘SQLALCHEMY_TRACK_MODIFICATIONS‘] = True
#查询时会显示原始SQL语句
app.config[‘SQLALCHEMY_ECHO‘] = True
db = SQLAlchemy(app)

class Role(db.Model):
    # 定义表名
    __tablename__ = ‘roles‘
    # 定义列对象
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(64), unique=True)
    us = db.relationship(‘User‘, backref=‘role‘)

    #repr()方法显示一个可读字符串
    def __repr__(self):
        return ‘Role:%s‘% self.name

class User(db.Model):
    __tablename__ = ‘users‘
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(64), unique=True, index=True)
    email = db.Column(db.String(64),unique=True)
    pswd = db.Column(db.String(64))
    role_id = db.Column(db.Integer, db.ForeignKey(‘roles.id‘))

    def __repr__(self):
        return ‘User:%s‘%self.name
if __name__ == ‘__main__‘:
    db.drop_all()
    db.create_all()
    ro1 = Role(name=‘admin‘)
    ro2 = Role(name=‘user‘)
    db.session.add_all([ro1,ro2])
    db.session.commit()
    us1 = User(name=‘wang‘,email=‘[email protected]‘,pswd=‘123456‘,role_id=ro1.id)
    us2 = User(name=‘zhang‘,email=‘[email protected]‘,pswd=‘201512‘,role_id=ro2.id)
    us3 = User(name=‘chen‘,email=‘[email protected]‘,pswd=‘987654‘,role_id=ro2.id)
    us4 = User(name=‘zhou‘,email=‘[email protected]‘,pswd=‘456789‘,role_id=ro1.id)
    db.session.add_all([us1,us2,us3,us4])
    db.session.commit()
    app.run(debug=True)

 

常用的SQLAlchemy查询过滤器

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

常用的SQLAlchemy查询执行器

方法	说明
all()	以列表形式返回查询的所有结果
first()	返回查询的第一个结果，如果未查到，返回None
first_or_404()	返回查询的第一个结果，如果未查到，返回404
get()	返回指定主键对应的行，如不存在，返回None
get_or_404()	返回指定主键对应的行，如不存在，返回404
count()	返回查询结果的数量
paginate()	返回一个Paginate对象，它包含指定范围内的结果

创建表：

db.create_all()

 

删除表

db.drop_all()

 

插入一条数据

ro1 = Role(name=‘admin‘)
db.session.add(ro1)
db.session.commit()
#再次插入一条数据
ro2 = Role(name=‘user‘)
db.session.add(ro2)
db.session.commit()

 

一次插入多条数据

us1 = User(name=‘wang‘,email=‘[email protected]‘,pswd=‘123456‘,role_id=ro1.id)
us2 = User(name=‘zhang‘,email=‘[email protected]‘,pswd=‘201512‘,role_id=ro2.id)
us3 = User(name=‘chen‘,email=‘[email protected]‘,pswd=‘987654‘,role_id=ro2.id)
us4 = User(name=‘zhou‘,email=‘[email protected]‘,pswd=‘456789‘,role_id=ro1.id)
db.session.add_all([us1,us2,us3,us4])
db.session.commit()

 

查询:filter_by精确查询

返回名字等于wang的所有人

User.query.filter_by(name=‘wang‘).all()

 

first()返回查询到的第一个对象

User.query.first()

 

all()返回查询到的所有对象

User.query.all()

 

filter模糊查询，返回名字结尾字符为g的所有数据。

User.query.filter(User.name.endswith(‘g‘)).all()

 

get()，参数为主键，如果主键不存在没有返回内容

User.query.get()

 

逻辑非，返回名字不等于wang的所有数据。

User.query.filter(User.name!=‘wang‘).all()

 

逻辑与，需要导入and，返回and()条件满足的所有数据。

from sqlalchemy import and_
User.query.filter(and_(User.name!=‘wang‘,User.email.endswith(‘163.com‘))).all()

 

逻辑或，需要导入or_

from sqlalchemy import or_
User.query.filter(or_(User.name!=‘wang‘,User.email.endswith(‘163.com‘))).all()

 

not_ 相当于取反

from sqlalchemy import not_
User.query.filter(not_(User.name==‘chen‘)).all()

 

查询数据后删除

user = User.query.first()
db.session.delete(user)
db.session.commit()
User.query.all()

 

更新数据

user = User.query.first()
user.name = ‘dong‘
db.session.commit()
User.query.first()

 

使用update

User.query.filter_by(name=‘zhang‘).update({‘name‘:‘li‘})

 

关联查询示例：角色和用户的关系是一对多的关系，一个角色可以有多个用户，一个用户只能属于一个角色。

查询角色的所有用户：

#查询roles表id为1的角色
ro1 = Role.query.get(1)
#查询该角色的所有用户
ro1.us

 

查询用户所属角色：

#查询users表id为3的用户
us1 = User.query.get(3)
#查询用户属于什么角色
us1.role