flask(三)之Flask-SQLAlchemy
作者：互联网
01-介绍
Flask-SQLAlchemy是一个Flask扩展，简化了在Flask应用中使用SQLAlchemy的操作。SQLAlchemy提供了高层ORM，也提供了使用数据库原生SQL的低层功能。
# 安装 pip install flask-sqlalchemy
在Flask-SQLAlchemy中，数据库使用URL指定。
应用使用的数据库URL必须保存到Flask配置对象的 SQLALCHEMY_DATABASE_URI 键中。
建议把 SQLALCHEMY_TRACK_MODIFICATIONS 键 设为 False，以便在不需要跟踪对象变化时降低内存消耗。
02-数据库的连接
from flask import Flask
# 1.导入
from flask_sqlalchemy import SQLAlchemy
app = Flask(__name__)
#2.定义要连接的数据库
DB_URI = "mysql+pymysql://root:123456@127.0.0.1:3306/learn_sqlalchemy?charset=utf8"
#3.添加到到配置中
app.config['SQLALCHEMY_DATABASE_URI'] = DB_URI
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] =False
# 4.实例化一个对象，将app传进去
db = SQLAlchemy(app)
@app.route('/')
def hello_world():
return 'Hello World!'
if __name__ == '__main__':
app.run()
03-定义模型
class User(db.Model):
__tablename__ = 'user'
id = db.Column(db.Integer,primary_key=True,autoincrement=True)
username = db.Column(db.String(50),nullable=False)
def __repr__(self):
return self.username
class Article(db.Model):
__tablename__ = "article"
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
title = db.Column(db.String(50), nullable=False)
uid = db.Column(db.Integer,db.ForeignKey("user.id"))
author = db.relationship("User",backref='article')
def __repr__(self):
return self.title
__tablename__ 定义在数据库中的表名。
04-一对多的关系
例：一个角色对应多个用户：
class Role(db.Model):
# ...
users = db.relationship('User', backref='role')
class User(db.Model):
# ...
role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))
db.relationship()中的第一个参数指的是这个关系的另一端是哪个模型，backref参数向User模型添加一个role属性，从而定义反向关系。
05-操作
# 创建表 db.create_all() # 删除表 db.drop_all() # 插入行 admin_role = Role(name='admin') user_role= User(username='hyp', role=admin_role) # 对数据库的改动童年过数据库会话管理，在Flask-SQLAlchemy中，会话由 db.session 表示。在写入数据库之前，要先添加到会话中： db.session.add(admin_role) db.session.add(user_role) # 或者 db.session.add_all([admin_role, user_role]) # 为了写入数据库，要调用 commit()方法提交会话： db.session.commit() # 数据库回话能保证数据库的一致性。数据库回话也可以回滚，调用 db.session.rollback() # 修改行 admin_role.name= 'lcy' db.session.add(admin_role) db.session.commit() # 删除行 db.session.delete(admin_role) db.session.commit() # 注意：删除、更新和插入一样，提交数据会话才执行。 # 查询行 # 每个模型类都提供了query对象。 Role.query.all() Role.query.filter_by(role=user_role).all() # 查看生成的原生SQL查询语句，只需要把query对象转换成字符串： str(User.query.filter_by(role=user_role)) # first()方法 只返回第一个结果 user_role = Role.query.filter_by(name='User').first()
06-集成Python Shell
若想把对象添加到入列表中，必须使用 app.shell_context_processor 装饰器创建并注册一个shell上下文处理器。
@app.shell_context_processor
def make_shell_context():
return dict(db=db, User=User, Role=Role)
这个shell上下文处理器函数返回一个字典，包含数据库实例和模型。
07-使用 Flask-Migrate实现数据库迁移
Flask应用还可以使用Flask-Migrate扩展，是对Alembic的轻量级包装。
# 安装 pip install flask-migrate
manage.py
from flask_script import Manager
from flask_migrate_demo import app
from exts import db
import models #这个一定要导入
from flask_migrate import Migrate,MigrateCommand
manager = Manager(app)
Migrate(app,db)
manager.add_command("db",MigrateCommand) #把所有命令放到db里面
if __name__ == '__main__':
manager.run()
初始化
python manage.py db init
创建迁移脚本
python manage.py db migrate -m '第一次提交'
生成到数据库
python manage.py db upgrade
08-alembic数据迁移工具
alembic是用来做ORM模型与数据库的迁移与映射。alembic使用方式跟git有点类似，表现在两个方面，第一个，alemibi的所有命令都是以alembic开头；
第二，alembic的迁移文件也是通过版本进行控制的。
# 安装 pip install alembic
使用：
# model.py
from sqlalchemy import Column,Integer,String,create_engine
from sqlalchemy.ext.declarative import declarative_base
DB_URI = "mysql+pymysql://root:123456@127.0.0.1:3306/alembic_demo?charset=utf8"
engine = create_engine(DB_URI)
Base = declarative_base(engine)
class User(Base):
__tablename__ = 'user'
id = Column(Integer,primary_key=True,autoincrement=True)
username = Column(String(50),nullable=False)
在终端初始化，创建一个仓库
alembic init learn_alembic
修改配置文件，指定连接的数据库
# alembic.ini sqlalchemy.url = mysql+pymysql://root:123456@127.0.0.1:3306/alembic_demo?charset=utf8
将models所在的目录路径添加到env.py，并指定target_metadata
import sys,os # 1.__file__：当前文件（env.py） #2.os.path.dirname(__file__)：获取当前文件的目录 #3.os.path.dirname(os.path.dirname(__file__))：获取当前文件目录的上一级目录 #4.sys.path: python寻找导入的包的所有路径 sys.path.append(os.path.dirname(os.path.dirname(__file__))) import models target_metadata = models.Base.metadata
生成迁移脚本
alembic revision --autogenerate -m "第一次提交"
将生成的迁移脚本映射到数据库中
alembic upgrade head
以后如果想要添加或修改模型，重复最后两步即可。
常用命令：
init:创建一个alembic仓库 rebision:创建一个新的版本文件 --autogenerate:自动将当前模型的修改，生成迁移脚本 -m:本次迁移做了哪些修改 upgrade:将指定版本的迁移文件映射到数据库中，会执行版本文件中的upgrade函数 head:代表当前的迁移脚本的版本号 downgrade:会执行指定版本的迁移文件中的downgrade函数 heads:展示当前可用的heads脚本文件 history:列出所有的迁移版本及其信息 current:展示当前数据库中的版本号
经典错误 1.FAILED:Target databases is not up to date. 原因：主要是heads和current不相同。current落后于heads的版本 解决办法：将current移动到head上。alembic upgrade head 2.FAILED:Can't locate revision identified by 'xxxxxxx' 原因：数据库中存的版本号不在迁移脚本文件中 解决办法：删除数据的alembic_version表中的数据，重新执行alembic upgrade head
用alembic工具：数据库中会自动生成一张表alembic_version
在数据库中可以查看当前的版本号
在cmd终端也可以通过current命令查看
alembic current
Flask-SQLAlchemy下使用alembic：
1. config.py
DB_URI = "mysql+pymysql://root:123456@127.0.0.1:3306/flask_alembic_demo?charset=utf8" SQLALCHEMY_DATABASE_URI = DB_URI
（2）flask_alembic_demo.py
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
import config
app = Flask(__name__)
app.config.from_object(config)
db = SQLAlchemy(app)
class User(db.Model):
__tablename__ = 'user'
id = db.Column(db.Integer,primary_key=True,autoincrement=True)
username = db.Column(db.String(50),nullable=False)
@app.route('/')
def hello_world():
return 'Hello World!'
if __name__ == '__main__':
app.run()
（3）初始化
alembic init alembic
（4）alembic.ini
sqlalchemy.url = mysql+pymysql://root:123456@127.0.0.1:3306/flask_alembic_demo?charset=utf8
（5）env.py
import sys,os sys.path.append(os.path.dirname(os.path.dirname(__file__))) import flask_alembic_demo # 用的是db.Model target_metadata = flask_alembic_demo.db.Model.metadata
（6）生成迁移脚本
alembic revision --autogenerate -m "first commit"
（7）upgrade到数据库
alembic upgrade head
（8）添加字段
假入想添加一个字段age
class User(db.Model):
__tablename__ = 'user'
id = db.Column(db.Integer,primary_key=True,autoincrement=True)
username = db.Column(db.String(50),nullable=False)
age = db.Column(db.Integer)
重复步骤即可
alembic revision --autogenerate -m "add column age" alembic upgrade head
标签：__,SQLAlchemy,flask,数据库,db,Flask,role,app,alembic 来源： https://www.cnblogs.com/pgxpython/p/10422840.html