测试开发之路--Flask 之旅 (三):数据库

来源：https://testerhome.com

背景

通过前两次的努力，我们对环境有了增删查改以及部署和查看日志的能力。 现在已经处于将就可用的状态。但其实还差了很重要的东西，就是权限的管理。 因为不能说每个用户上来都能随便的重启和删除环境吧，太容易出事故了。所以我们想起码有最基本的隔离性。

Flask扩展模块

上一次我们使用了Flask-WTF这个针对于表单的扩展模块。 这次为了加入权限管理，我们需要再安装以下这些扩展模块。

• Flask-SQLAlchemy
• Flask-MySQLdb
• Flask-Security
• Flask-Login

涉及到权限就需要有用户的概念，我们需要数据库来存储我们用户和环境的信息，所以要使用Flask-SQLAlchemy和Flask-MySQLdb(我使用的是mysql)。为了解决用户登录的问题引入Flask-Login， 为了生成角色和权限的概念引入Flask-Security。这里解释一下Flask-Security，它无法单独使用，更像是Flask-Login和Flask-SQLAlchemy的扩展。 是为了增强他们的权限控制能力而存在的。所以再加入Flask-Security后，Flask-SQLAlchemy和Flask-Login的使用方式都跟以前有些不太一样了。接下来我会一个一个介绍

创建数据库

我们先解决数据库的问题吧。 再安装好上面的所有模块后(注：缺一不可)，我们首先要在数据库中创建名叫env的库。供我们使用

再启动app前设置一下数据库参数：

代码语言：javascript

复制

# 连接mysql数据库的配置
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://root:root@172.27.1.221:3306/env'

然后我们创建一个models.py

代码语言：javascript

复制

from __init__ import *
from flask_security import RoleMixin, UserMixin
# Create database connection object
db = SQLAlchemy(app)
# Define models
roles_users = db.Table('roles_users',
                       db.Column('user_id', db.Integer(), db.ForeignKey('user.id')),
                       db.Column('role_id', db.Integer(), db.ForeignKey('role.id')))
class Role(db.Model, RoleMixin):
    id = db.Column(db.Integer(), primary_key=True)
    name = db.Column(db.String(80), unique=True)
    description = db.Column(db.String(255))
class User(db.Model, UserMixin):
    id = db.Column(db.Integer, primary_key=True)
    email = db.Column(db.String(255), unique=True)
    password = db.Column(db.String(255))
    active = db.Column(db.Boolean())
    confirmed_at = db.Column(db.DateTime())
    roles = db.relationship('Role', secondary=roles_users,
                            backref=db.backref('users', lazy='dynamic'))
    envs = db.relationship('Env', backref=db.backref('user'))
class Env(db.Model):
    def __init__(self, name, user_id):
        self.name = name
        self.user_id = user_id
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(255), unique=True)
    user_id = db.Column(db.Integer, db.ForeignKey('user.id'))

看到这么一坨代码可能有点懵逼。我来解释解释吧。 Flask-SQLAlchemy 是一个ORM框架，它组织了数据库表到类的影射。所以我们可以使用管理这些类对象的方式管理数据库。 你看到上面我们创建了 User，Env和Role 类，分别对应了用户，环境和角色。它们都集成了db.Model。我们把SQLAlchemy(app) 赋值给db，这是初始化我们数据库的方式。 然后你会发现User和Role又分别继承了UserMixin 和 RoleMixin。暂时可以不用管它， 这个是Flask-Security的东西。 表之间的关系我用外键来定义，额，虽然我知道外键的性能不好。开发人员几乎从来不用。但是我这么个小玩意就不在意这些有的没的了，一共就没几条数据。看到上面对roles_users的定义可能大家会觉得有点懵逼。 这是SQLAlchemy 处理多对多关系的方式：产生一张中间表。 SQLAlchemy处理表关系的方式就是relationship方法。这部分内容有点多~~ 具体详细的API定义请大家参考官方文档：http://www.pythondoc.com/flask-sqlalchemy/quickstart.html

接下来我们只需要调用下面的代码就会在数据库创建出这些表了。

代码语言：javascript

复制

db.create_all()

3张表加一个roles_users的表。

数据库操作

现在我们有表了，我们想初始化一些数据。 所以我们先看看一些基础的操作。 例如我们想创建一个环境信息，我们可以这么做：

代码语言：javascript

复制

env = Env()
env.name = 'test01'
env.user_id = '1'
db.session.add(env)
db.session.commit()

这是添加一条数据的方式。 只要是增删改的操作，只要没有运行db.session.commit()， 就不会真正的入库。 这是为了保证事务性。增加了一条数据后，现在我们来查询数据：

代码语言：javascript

复制

env = Env.query.filter_by(name='test01').first()

由于我们model的类都继承了db.Models。 所以我们可以使用一些方法来进行查询。 也可以使用一些复杂一点的查询方式，如下：

代码语言：javascript

复制

Env.query.filter(name.endswith('01')).all()

使用 Flask-Security 创建User和Role

OK，现在我们看看一开始没有说明的UserMixin 和 RoleMixin。上面说过他们是Flask-Security针对Flask-SQLAlchemy做的扩展。通过Flask-Security我们可以很方便的管理用户权限但是它对我们的model有一定的要求。请看下图：

就是说我们一定要有User和Role这两张表以及包含相应的字段。 这样Flask-Security才能够帮助我们生成权限管理的解决方案。举个简单的例子，通过使用Flask-Security， 我们可以用下面的方式创建用户：

代码语言：javascript

复制

from urls import db， User,  Role
from flask_security import SQLAlchemyUserDatastore, Security
# Setup Flask-Security
user_datastore = SQLAlchemyUserDatastore(db, User, Role)
security = Security(app, user_datastore)
db.create_all()
admin = user_datastore.create_user(email='admin@4paradigm.com', password='admin')
# 生成普通用户角色和admin用户角色
user_datastore.create_role(name='User', description='Generic user role')
admin_role = user_datastore.create_role(name='Admin', description='Admin user role')
# 为admin添加Admin角色
user_datastore.add_role_to_user(admin, admin_role)
db.session.commit()

上面是我初始化数据库的代码。 为了能使用Flask-Security。我们使用SQLAlchemyUserDatastore来封装db， 使用Security封装我们的web app。 接下来就可以看到我们使用很方便的方式创建user和role。 并且使用add_role_to_user的方式为一个用户添加角色。Flask-Security为我们提供了很多有用的方法。请看下面的截图：

更多的API请查看官方文档：https://pythonhosted.org/Flask-Security/api.html

未完待续

上面说的这些方法都可以很有效的管理我们数据库中User和Role的关系。再下一篇帖子中，我会详细介绍如何使用Flask-Security来做权限控制。