Web框架——Flask系列之Flask-SQLAlchemy数据库的基本操作（九）_db.session.rollback()

数据库基本操作

• 在Flask-SQLAlchemy中，插入、修改、删除操作，均由数据库会话管理。

会话用 db.session 表示。在准备把数据写入数据库前，要先将数据添加到会话中然后调用db.session.commit()方法提交会话。

1. db.session.add（role）添加到数据库的session中
2. db.session.add_all（[user1，user2]）添加多个信息到session中
3. db.session.commit（）提交数据库的修改（包括增/删/改）
4. db.session.rollback（）数据库的回滚操作
5. db.session.delete（user）删除数据库（需跟上commit）

• 在 Flask-SQLAlchemy 中，查询操作是通过 query 对象操作数据。

最基本的查询是返回表中所有数据，可以通过过滤器进行更精确的数据库查询。

一、在视图函数中定义模型类

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)

#设置连接数据库的URL
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:mysql@127.0.0.1:3306/test'
#设置数据库追踪信息,压制警告
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
#创建SQLAlchemy对象,读取app中配置信息
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)
    password = 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相关联的表
    db.drop_all()
    #创建所有和db相关联的表
    db.create_all()
    app.run(debug=True)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

二、关键代码格式,说明

• 一对多关系

class Role(db.Model):
    ...
    #关键代码
    us = db.relationship('User', backref='role', lazy='dynamic')
    ...

class User(db.Model):
    ...
    role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))
1
2
3
4
5
6
7
8
9

• 其中realtionship描述了Role和User的关系。
• 第一个参数为对应参照的类"User"
• 第二个参数backref为类User,反向引用属性
• 第三个参数lazy决定了什么时候SQLALchemy从数据库中加载数据

1. 如果设置为子查询方式(subquery)，则会在加载完Role对象后，就立即加载与其关联的对象，这样会让总查询数量减少，但如果返回的条目数量很多，就会比较慢
   设置为 subquery 的话，role.users 返回所有数据列表
2. 另外,也可以设置为动态方式(dynamic)，这样关联对象会在被使用的时候再进行加载，并且在返回前进行过滤，如果返回的对象数很多，或者未来会变得很多，那最好采用这种方式

三、常见的操作语句

• db.session.add(obj) 添加对象
• db.session.add_all([obj1,obj2,…]) 添加多个对象
• db.session.delete(obj) 删除对象
• db.session.commit() 提交会话
• db.session.rollback() 回滚
• db.session.remove() 移除会话
  
  

代码演示：

往roles表中添加一条记录：role;往users表中添加、修改、删除一条记录：user

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)

# 数据库链接地址
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:root@127.0.0.1/flask_sql_demo'
# 动态追踪修改设置，如未设置只会提示警告,不建议开启，未来版本中会移除
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

db = SQLAlchemy(app)

'''
两张表
角色（管理员/普通用户）
用户（角色ID）
'''

# 数据库的模型，需要继承db.Model
# 角色表
class Role(db.Model):
    # 定义表名
    __tablename__ = 'roles'

    # 定义字段
    # db.Column表示是一个字段
    id = db.Column(db.Integer,primary_key=True)
    name = db.Column(db.String(16),unique=True)

# 用户表
class User(db.Model):
    # 定义表名
    __tablename__ = 'users'

    # 定义字段
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(16), unique=True)
    # db.ForeignKey('roles.id')表示是外键，需要传参数：表名.字段名
    role_id = db.Column(db.Integer,db.ForeignKey('roles.id'))





@app.route('/')

def index():
    return 'Hello flask!'

if __name__ == '__main__':
    # 删除表
    db.drop_all()
    # 创建表
    db.create_all()

    # 创建测试数据
    # 增加数据
    role = Role(name='admin')
    db.session.add(role)
    db.session.commit()

    user = User(name='Zep',role_id=role.id)
    db.session.add(user)
    db.session.commit()

    # 修改user这条记录时，不需要再db.session.add(user)，因为之前已经add过了，直接commit即可
    user.name = 'Zep03'
    db.session.commit()

    # 删除user这条记录
    db.session.delete(user)
    db.session.commit()

    user = User(name='Zep', role_id=role.id)
    db.session.add(user)
    db.session.commit()

    app.run(debug=True)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

运行结果：

四、常用的SQLAlchemy查询过滤器

• 用来过滤数据,返回查询的结果集
  

1. 通过filter_by（列名=值，列名=值）的方法来查询数据：
   
   
2. 通过filter(类名.列名值 ，类名.列名值)的方法来查询数据：
   
   
3. 完整使用方法：User.query.filter().offset().limit.order_by().all()

offset表示“偏移”，offset(2)表示跳过2条记录，从第3条开始查


4. order_by(类名.列名.desc()):按照列名的降序排序

order_by(类名.列名.asc()):按照列名的升序排序（默认就是按照升序）

from sqlalchemy import func
db.session.query(User.role_id,func.count(User.role_id)).group_by(User.role_id).all()

1
2
3

五、常用的SQLAlchemy查询执行器

• 用来执行结果集,得到具体数据
  

1. all()方法，以列表的形式返回查询的所有结果，列表中存放的是一个个对象（一条条记录）：
   
   
2. 通过first()和get()方法获取表中的某一条记录： 注意：get(主键的值)，返回指定主键对应的行

3. 关联查询：通过Role的对象来获取关联的字段的列表，ro.users

4. 关联查询通过User的对象来获取role:

5. 显示信息： def repr(self):
‘’‘定义之后，可以让显示对象的时候更加直观’’’
return “User object:name=%s” %self.name

六、练习数据库查询操作

• 代码准备

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)

#配置信息
app.config["SQLALCHEMY_DATABASE_URI"] = "mysql+pymysql://root:123456@127.0.0.1:3306/basic8"
#设置压制警告信息,如果True会追踪数据库变化,会增加显著开销,所以建议设置为False
app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False

#创建SQLAlchemy类对象,关联app
db = SQLAlchemy(app)

#编写模型类,继承db.Model
#角色,用户之间的关系
class Role(db.Model):
    __tablename__ = "roles" #指定表名称
    #参数1:表示整数类型,  参数2:表示主键
    id = db.Column(db.Integer,primary_key=True)
    #角色名唯一的
    name = db.Column(db.String(64),unique=True)


    #需要设置关系属性relationship(不会产生字段),设置在一方
    #给Role添加了users关系属性, 查询格式: role.users
    #给User添加了role关系属性(反向引用),查询格式: user.role
    users = db.relationship('User',backref='role')

    #为了方便的看到对象输出的内容__repr__, 如果是普通类__str__
    def __repr__(self):
        return "<Role:%s>"%self.name

# 用户(多方)
class User(db.Model):
    __tablename__ = "users"  # 指定表名称
    #参数1:表示整数类型,  参数2:表示主键
    id = db.Column(db.Integer,primary_key=True)
    #用户名唯一的
    name = db.Column(db.String(64),unique=True)
    #邮箱密码
    email = db.Column(db.String(64),unique=True)
    password = db.Column(db.String(64))

    #外键
    role_id = db.Column(db.Integer,db.ForeignKey(Role.id))

    #为了方便的看到对象输出的内容__repr__, 如果是普通类__str__
    def __repr__(self):
        return "<User:%s,%s,%s,%s>"%(self.id,self.name,self.email,self.password)

if __name__ == '__main__':

    #为了演示方便,先删除数据库表,和模型类关联的表
    db.drop_all()

    #创建表,所有继承自dbModel的表
    db.create_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='wang@163.com', password='123456', role_id=ro1.id)
    us2 = User(name='zhang', email='zhang@189.com', password='201512', role_id=ro2.id)
    us3 = User(name='chen', email='chen@126.com', password='987654', role_id=ro2.id)
    us4 = User(name='zhou', email='zhou@163.com', password='456789', role_id=ro1.id)
    us5 = User(name='tang', email='tang@itheima.com', password='158104', role_id=ro2.id)
    us6 = User(name='wu', email='wu@gmail.com', password='5623514', role_id=ro2.id)
    us7 = User(name='qian', email='qian@gmail.com', password='1543567', role_id=ro1.id)
    us8 = User(name='liu', email='liu@itheima.com', password='867322', role_id=ro1.id)
    us9 = User(name='li', email='li@163.com', password='4526342', role_id=ro2.id)
    us10 = User(name='sun', email='sun@163.com', password='235523', role_id=ro2.id)
    db.session.add_all([us1, us2, us3, us4, us5, us6, us7, us8, us9, us10])
    db.session.commit()

    app.run(debug=True)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

• 需求: 编写方法,查询以下内容

1. 查询所有用户数据
2. 查询有多少个用户
3. 查询第1个用户
4. 查询id为4的用户[3种方式]
5. 查询名字结尾字符为g的所有数据[开始/包含]
6. 查询名字不等于wang的所有数据[2种方式]
7. 查询名字和邮箱都以 li 开头的所有数据[2种方式]
8. 查询password是 123456 或者 email 以 itheima.com 结尾的所有数据
9. 查询id为 [1, 3, 5, 7, 9] 的用户列表
10. 查询name为liu的角色数据
11. 查询所有用户数据，并以邮箱排序
12. 每页3个，查询第2页的数据

七、代码参考

查询:filter_by精确查询
返回名字等于wang的所有人

User.query.filter_by(name='wang').all()
1

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

User.query.first()
1

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

User.query.all()
1

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

User.query.filter(User.name.endswith('g')).all()
1

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

User.query.get()
1

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

User.query.filter(User.name!='wang').all()
1

6. not_ 相当于取反

from sqlalchemy import not_
User.query.filter(not_(User.name=='chen')).all()
1
2

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

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

8. 逻辑或，需要导入or_

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

9. 查询数据后删除

user = User.query.first()
db.session.delete(user)
db.session.commit()
User.query.all()
1
2
3
4

10. 更新数据

user = User.query.first()
user.name = 'dong'
db.session.commit()
User.query.first()
1
2
3
4

关联查询示例：

角色和用户的关系是一对多的关系，一个角色可以有多个用户，一个用户只能属于一个角色。

• 查询角色的所有用户

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

• 查询用户所属角色

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