使用Flask-SQLALchemy操作MySQL数据库

7564321

1、简介

Flask-SQLAlchemy 是一个 Flask 扩展，简化了在 Flask程序中使用SQLAlchemy 的操作。SQLAlchemy 是一个很强大的关系型数据库框架，支持多种数据库后台。 SQLAlchemy 提供了高层 ORM，也提供了使用数据库原生 SQL 的低层功能。

2、安装

$ pip install flask-sqlalchemy

3、配置连接mysql

$ vim models.py

from flask importFlask

from flask_script import Manager

from flask_sqlalchemy import SQLAlchemy

app =Flask(__name__)

app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql://kevin:123456@10.1.1.5/kevin'

app.config['SQLALCHEMY_COMMIT_ON_TEARDOWN'] = True

manager = Manager(app)

db = SQLAlchemy(app)



if __name__ =='__main__':

manager.run()

4、定义模型

$ vim models.py（只列出修改部分）

db = SQLAlchemy(app)

class Role(db.Model):

    __tablename__ = 'roles'

    id = db.Column(db.Integer,primary_key=True)

    name = db.Column(db.String(255),unique=True)

    users = db.relationship('User',backref='role')

    def __repr__(self):

        return '<Role %r>' %self.name



class User(db.Model):

    __tablename__ = 'users'

    id = db.Column(db.Integer,primary_key=True)

    username = db.Column(db.String(255),unique=True,index=True)

    password = db.Column(db.String(255))

    role_id = db.Column(db.Integer,db.ForeignKey('roles.id'))

    def __repr__(self):

        return '<User %r>' %self.username

类变量 __tablename__ 定义在数据库中使用的表名。如果没有定义 __tablename__， Flask-SQLAlchemy 会使用模版名的小写作为表名。

不需要定义__init__或__repr__，如果没有定义，SQLAlchemy会自动创建__init__方法。如果定义了，定义的字段名会成为此方法所接收的关键字参数名。

5、数据库操作

5.1、创建表

创建数据库，方法是使用db.create_all()函数。如果数据库表已经存在于数据库中，那么 db.create_all()不会重新创建或者更新这个表。

查看数据库

mysql> use kevin

Database changed

mysql> show tables;

Empty set (0.00sec)

创建

$ python models.py shell

>>> from models import db

>>>db.create_all()

查看

mysql> show tables;

+-----------------+

| Tables_in_kevin |

+-----------------+

| roles           |

| users           |

+-----------------+

2 rows in set (0.00sec)

5.2、插入数据

>>> from  models import Role,User

>>>admin_role = Role(name='Admin')

>>>mod_role = Role(name='Moderator')

>>>user_role = Role(name='User')

>>>user_kevin = User(username='kevin',role=admin_role)

>>>user_susan = User(username='susan',role=user_role)

>>>user_david = User(username='david',role=user_role)

现在这些对象只存在于Python 中，还未写入数据库。因此 id 尚未赋值。

>>>print(admin_role.id)

None

>>>print(mod_role.id)

None

>>>print(user_role.id)

None

通过数据库会话管理对数据库所做的改动，在 Flask-SQLAlchemy 中，会话由 db.session表示。准备把对象写入数据库之前，先要将其添加到会话中。

>>>db.session.add(admin_role)

>>>db.session.add(mod_role)

>>>db.session.add(user_role)

>>>db.session.add(user_kevin)

>>>db.session.add(user_susan)

>>>db.session.add(user_david)

可以简写为：

>>>db.session.add_all([admin_role,mod_role,user_role,user_kevin,user_susan,user_david])

为了把对象写入数据库，要调用 commit() 方法提交会话。

>>>db.session.commit()

再次查看id属性，它们已经赋值了，表里也有数据了。

>>>print(admin_role.id)

1

>>>print(mod_role.id)

2

>>>print(user_role.id)

3

mysql> select * from roles;

+----+-----------+

| id | name      |

+----+-----------+

|  1 | Admin    |

|  2 | Moderator |

|  3 | User     |

+----+-----------+

3 rows in set (0.00sec)



mysql> select * from users;

+----+----------+----------+---------+

| id | username |password | role_id |

+----+----------+----------+---------+

|  1 | kevin   | NULL     |       1 |

|  2 | susan   | NULL     |       3 |

|  3 | david   | NULL     |       3 |

+----+----------+----------+---------+

3 rows in set (0.00sec)

5.3、修改数据

在数据库会话上调用 add() 方法也能更新模型。下面把 "Admin" 角色重命名为 "Administrator"。

>>>admin_role.name = 'Adminitrator'

>>>db.session.add(admin_role)

>>>db.session.commit()

>>>print(admin_role.name)

Adminitrator

5.4、删除数据

数据库会话还有个 delete() 方法。下面是把 "Moderator" 角色从数据库中删除。

>>>db.session.delete(mod_role)

>>>db.session.commit()

mysql> select name from roles where name='Moderator';

Empty set (0.00sec)

5.5、查询数据

Flask-SQLAlchemy 为每个模型类都提供了query 对象。最基本的模型查询是取回对应表中的所有记录。

>>>Role.query.all()

[<Roleu'Adminitrator'>, <Role u'User'>]

>>>User.query.all()

[<Useru'kevin'>, <User u'susan'>, <User u'david'>]

使用过滤器可以配置 query 对象进行更精确的数据库查询。下面这个例子查找角色为"User" 的所有用户

>>>User.query.filter_by(role=user_role).all()

[<Useru'susan'>, <User u'david'>]

filter_by() 等过滤器在 query 对象上调用，返回一个更精确的 query 对象。多个过滤器可以一起调用，直到获得所需结果。完整的列表参见 SQLAlchemy 文档。