Flask SQLAlchemy管理数据库
关系型数据库框架,支持多种数据库,提供高层ORM, 也支持执行原生SQL
$ pip install flask-sqlalchemy
$ pip install pymysql
数据库连接URL
Mysql
mysql+pymysql://username:password@server/db
创建数据库 mysql控制台
mysql> create database school4 default character set = utf8;
from flask import Flask
from flask_script import Manager
from flask_sqlalchemy import SQLAlchemy
app = Flask(__name__)
# 设置数据库连接字符串
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://root:Vff123456@127.0.0.1/school4?charset=UTF8MB4'
# 不跟踪修改,不设置会有警告
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
# 创建数据库连接
db = SQLAlchemy(app)
import application.models
import application.views
manager = Manager(app)
定义模型
from application import db
import datetime
# 用户表
class Users(db.Model):
__tablename__ = 'users'
# 主键
id = db.Column(db.Integer, primary_key=True)
# 用户名 唯一索引
username = db.Column(db.String(128), unique=True, nullable=False)
# 密码 必填字段
password = db.Column(db.String(512), nullable=False)
# 姓名 创建索引,加快查询
fullname = db.Column(db.String(128), index=True, nullable=False)
# 状态 (1: 生效 0: 禁用)
status = db.Column(db.SmallInteger, default=1, nullable=False)
# 创建时间 默认当前时间
created_time = db.Column(db.DateTime, nullable=False, default=datetime.datetime.utcnow, index=True)
def __repr__(self):
return 'username=%s' % username
根据model创建表
$ python manage.py shell
>>> from application import db
>>> db.drop_all()
>>> db.create_all()
插入数据
$ python manage.py shell
>>> user = Users(username='lisa', password='asdasdfad', fullname='李霞')
>>> db.session.add(user) # 添加到数据库会话中
>>> db.session.commit() # 提交数据
>>> db.session.rollback() # 数据库回滚会有效么?试下放在commit前面呢?
查询数据
>>> Users.query.all() # 查询全部
>>> Users.query.filter_by(username='lisa') # 返回一个新查询
>>> Users.query.filter_by(username='lisa').all() # 返回查询结果
>>> Users.query.filter(Users.username=='lisa').first_or_404()
>>> str(Users.query.filter(Users.username=='lisa3')) # 查看sql语句
>>> db.session.execute('select * from users where id = 1').first() # 原生sql语句
修改行
user = Users.query.get(1)
user.password = '123412343'
db.session.add(user)
db.session.commit()
删除行
db.session.delete(user)
db.session.commit()
关系
一对多关系
# 学生表
class Student(db.Model):
__tablename__ = 'student'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(128), unique=True, nullable=False)
# 建立指向班级的外键
class_id = db.Column(db.Integer, db.ForeignKey('class.id'))
def __repr__(self):
return 'student=%s' % self.name
# 班级表
class Class(db.Model):
__tablename__ = 'class'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(128), unique=True, nullable=False)
# 引用学生
students = db.relationship('Student', backref='sclass')
一对多关系测试
>>> from application import db
>>> db.create_all()
>>> from application.models import *
>>> student1 = Student(name='lisa')
>>> db.session.add(student1)
>>> db.session.commit()
>>> c1 = Class(name='python1904')
>>> db.session.add(c1)
>>> db.session.commit()
>>> student1.sclass = c1
一对一关系
一对一关系可以用前面介绍的一对多关系 表示,但调用db.relationship() 时要把uselist设为False
student = db.relationship('Student', backref='sclass', uselist=False)
多对多关系建模
class StudentLesson(db.Model):
__tablename__ = 'student_lesson'
student_id = db.Column(db.Integer, db.ForeignKey('student.id'), primary_key=True)
lesson_id = db.Column(db.Integer, db.ForeignKey('lesson.id'), primary_key=True)
score = db.Column(db.Integer, nullable=True)
class Student(db.Model):
__tablename__ = 'student'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(128), unique=True, nullable=False)
lessons = db.relationship('StudentLesson', foreign_keys=[StudentLesson.student_id], backref=db.backref('student', lazy='joined'), lazy='dynamic', cascade='all, delete-orphan') `
def __repr__(self):
return 'student=%s' % self.name
class Lesson(db.Model):
__tablename__ = 'lesson'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(128), unique=True, nullable=False)
students = db.relationship('StudentLesson', foreign_keys=[StudentLesson.lesson_id], backref=db.backref('lesson', lazy='joined'), lazy='dynamic', cascade='all, delete-orphan')
def __repr__(self):
return 'lesson=%s' % self.name
多对多关系测试
>>> from application.models import *
>>> from application import db
>>> db.create_all()
>>> student1 = Student(name='carmack')
>>> student2 = Student(name='lisa')
>>> db.session.add_all([student1, student2])
>>> db.session.commit()
>>> lesson1 = Lesson(name='数学')
>>> lesson2 = Lesson(name='语文')
>>> db.session.add_all([lesson1, lesson2])
>>> db.session.commit()
>>> student_lesson = StudentLesson(student=student1, lesson=lesson1)
>>> db.session.add(student_lesson)
>>> db.session.commit()
常用的SQLAlchemy关系选项
- backref
在关系的另一个模型中添加反向引用 - primaryjoin
明确指定两个模型之间使用的联结条件。只在模棱两可的关系中需要指定 - lazy
指定如何加载相关记录。可选值有 select(首次访问时按需加载)、immediate(源对象加 载后就加载)、joined(加载记录,但使用联结)、subquery(立即加载,但使用子查询), noload(永不加载)和 dynamic(不加载记录,但提供加载记录的查询) - uselist
设为False,不使用列表
使用Flask-Migrate实现数据库迁移
安装migrate
pip install flask-migrate
配置使用migrate
初始化,使用app和db进行migrate对象的初始化
from flask_migrate import Migrate
#绑定app和数据库
migrate = Migrate(app, db)
在Manager()对象上添加迁移指令
from flask_migrate import Migrate, MigrateCommand
app = Flask(__name__)
manage = Manager(app=app)
manage.add_command(‘db’, MigrateCommand)
操作:
python manage.py db init 初始化出migrations的文件,只调用一次
python manage.py db migrate -m ‘comment’ 生成迁移文件
python manage.py db upgrade 执行迁移文件中的升级
python manage.py db downgrade 执行迁移文件中的降级
python manage.py db —help 帮助文档