SQLAlchemy是一个ORM框架。flask-sqlalchemy是一个简化了SQLAlchemy操作的flask扩展。
安装flask-sqlalchemy: pip install flask-sqlalchemy
安装Mysql驱动: pip install MySQL-python # Python2中安装MySQL-python,Python3中安装 PyMySQL
demo.py(模型类的定义,数据库的添加):
# coding:utf-8
from flask import Flask
from flask_sqlalchemy import SQLAlchemy # 导入
app = Flask(__name__)
# 通过类对象加载配置。 (配置数据库)
class Config(object):
"""配置参数"""
# sqlalchemy的配置参数
SQLALCHEMY_DATABASE_URI = "mysql://root:密码@127.0.0.1:3306/数据库名"
# 设置sqlalchemy自动跟踪数据库中数据的修改 (修改数据库数据后,对应的模型类对象会自动修改)(添加对象时,自动返回id给对象)
SQLALCHEMY_TRACK_MODIFICATIONS = True
app.config.from_object(Config) # 通过类对象加载配置参数
# 创建数据库sqlalchemy工具对象
db = SQLAlchemy(app)
# 模型类 (继承db.Model)
class Role(db.Model):
"""用户角色表"""
__tablename__ = "tbl_roles"
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(32), unique=True)
users = db.relationship("User", backref="role") # 关联属性。关联的所有用户 (一对多)
# backref="role" 表示会在User模型类中自动创建关联属性role。
def __repr__(self):
"""定义之后,可以让显示对象的时候更直观"""
return "Role object: name=%s" % self.name
# 表名的常见规范
# ihome -> ih_user 数据库名缩写_表名
# tbl_user tbl_表名
# 数据库模型类
class User(db.Model):
"""用户表"""
__tablename__ = "tbl_users" # 指明对应的数据库表名
id = db.Column(db.Integer, primary_key=True) # 整型的主键,会默认设置为自增主键
name = db.Column(db.String(64), unique=True)
email = db.Column(db.String(128), unique=True)
password = db.Column(db.String(128))
role_id = db.Column(db.Integer, db.ForeignKey("tbl_roles.id")) # 外键 (一对多)
def __repr__(self):
return "User object: name=%s" % self.name
if __name__ == '__main__':
# 清除数据库里的所有表 (不推荐使用)
db.drop_all()
# 创建所有的表 (根据模型类创建)
db.create_all()
# 创建对象
role1 = Role(name="admin")
# 通过会话将对象添加到数据库中 (session是与数据库的链接会话)
db.session.add(role1)
# 提交任务到数据库中
db.session.commit() # SQLALCHEMY_TRACK_MODIFICATIONS设置为True后,会自动将id传给对象role1
# db.session.rollback() # 回滚
role2 = Role(name="stuff")
db.session.add(role2)
db.session.commit()
us1 = User(name='wang', email='[email protected]', password='123456', role_id=role1.id)
us2 = User(name='zhang', email='[email protected]', password='201512', role_id=role2.id)
us3 = User(name='chen', email='[email protected]', password='987654', role_id=role2.id)
us4 = User(name='zhou', email='[email protected]', password='456789', role_id=role1.id)
# 一次添加多条数据
db.session.add_all([us1, us2, us3, us4])
db.session.commit()