Flask用Flask-SQLAlchemy连接MySQL

安装

 pip3 install Flask-SQLAlchemy

测试环境目录结构

settings.py

DIALECT = 'mysql'
    DRIVER = 'pymysql'
    USERNAME = 'root'
    PASSWORD = '808069'
    HOST = '127.0.0.1'
    PORT = '3306'
    DATABASE = 'cms'

    SQLALCHEMY_DATABASE_URI = '{}+{}://{}:{}@{}:{}/{}?charset=utf8'.format(
        DIALECT,DRIVER,USERNAME,PASSWORD,HOST,PORT,DATABASE
    )
    SQLALCHEMY_COMMIT_ON_TEARDOWN = True
    SQLALCHEMY_TRACK_MODIFICATIONS = True

    SQLALCHEMY_POOL_SIZE = 10
    SQLALCHEMY_MAX_OVERFLOW = 5

__init__.py

from flask_sqlalchemy import SQLAlchemy
from flask import Flask

db = SQLAlchemy()

def create_app():
    app = Flask(__name__)
    app.config.from_object('settings')

    db = SQLAlchemy()
    db.init_app(app)

    return app

manage.py

from ttt import create_app

app = create_app()

if __name__ == '__main__':
    app.run()

models.py

from manage import db

class User(db.Model):
    __tablename__ = 'user'
    id = db.Column(db.INTEGER,primary_key=True)
    username = db.Column(db.String(80),unique=True)
    password = db.Column(db.String(80),nullable=False)

class CodeCountRecord(db.Model):
    __tablename = 'codecountrecord'
    id = db.Column(db.INTEGER,primary_key=True)
    count = db.Column(db.INTEGER)
    data = db.Column(db.DATE)
    user = db.Column(db.ForeignKey('user.id'))

create_tables.py

from ttt import db, create_app
from ttt.models import *


app = create_app()
app_ctx = app.app_context()  # app_ctx = app/g
with app_ctx:  # __enter__,通过LocalStack放入Local中
    db.create_all()  # 调用LocalStack放入Local中获取app，再去app中获取配置

这种方式即为离线脚本（不用启动项目）的方式创建数据库

直接右键运行models.py即可创建表

操作

增

from cms.models import User
from manage import db

def create_user():
    # 创建一个新用户对象
    user = User()
    user.username = 'fuyong'
    user.password = '123'

    # 将新创建的用户添加到数据库会话中
    db.session.add(user)
    # 将数据库会话中的变动提交到数据库中, 记住, 如果不 commit, 数据库中是没有变化的.
    db.session.commit()

create_user()

删

def delete_user():
    # 获取用户对象
    user = User.query.filter_by(id=1).first()

    # 删除用户
    db.session.delete(user)

    #提交数据库会话
    db.session.commit()

delete_user()

改

def update_user():
    # 获取用户对象
    user = User.query.filter_by(id=2).first()

    # 修改用户
    user.password = '123567'

    #提交数据库会话
    db.session.commit()

update_user()

查

def select_user():
    # 查询所有用户
    users_list = User.query.all()

    # 查询用户名称为 fuyong 的第一个用户, 并返回用户实例, 因为之前定义数据库的时候定义用户名称唯一, 所以数据库中用户名称为 test 的应该只有一个.
    user = User.query.filter_by(username='fuyong').first()
    # or
    user = User.query.filter(User.username == 'fuyong').first()

    # 模糊查询, 查找用户名以abc 结尾的所有用户
    users_list = User.query.filter(User.username.endsWith('g')).all()

    # 查询用户名不是 fuyong 的第一个用户
    user = User.query.filter(User.username != 'fuyong').first()
    

循环导入的问题

如果上面的例子继续写下去的时候，我们或许会在视图views中引入models文件以操作数据，在models文件中引入manage文件中的db以定义类和字段，

然后在manage文件中引入views文件以注册蓝图（register_blueprint），这样就出现了a引入b，b引入c，c引入a的问题，就会报错，

解决办法就是另外创建一个ext.py文件，专门用来创建db，代码如下：

from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()

　　注意：此时先不讲app传入

然后在manage.py文件中，导入db，然后初始化，将app传进去：

db.init_app(app)

这样，在视图中需要用db的之后直接从ext导入，而不再从manage里导入