flask-sqlalchemy结合Blueprint遇到循环引入问题的解决方案

想要用flask_sqlalchemy结合Blueprint分模块写一下SQL的增删改查接口，结果发现有循环引入问题。

一开始，我在app.py中使用db = SQLAlchemy(app)创建数据库对象；并且使用app.register_blueprint(db_bp, url_prefix='/db')注册蓝图。

这使得我的依赖关系是这样的。db.py → app,py 中的db对象；app.py → db,py 用于注册蓝图，产生了循环引用。

接着我学着使用一个model.py来存放db变量，但是使用错误，下面是错误的示例。

我在model.py使用db = SQLAlchemy()创建了未注册的db对象

在app.py中使用db.init_app(app)来连接数据库，注册db对象

但在db.py中错误引用了app.py中的db对象，再次产生了循环引用

解决方法，在db.py中引入model.py中的db对象即可。

此时的依赖关系如下，db.py和app.py → model.py 用于获取db对象；app.py → db.py 用于注册蓝图。

看到一些比较麻烦的解决方案，1. 把db变量变成一个web接口，用请求的方式获取。

把app注册db和blueprint的操作都放入main函数中。或者，先封装到def create_app()函数中，然后在main函数中调用。

参考使用Flask-SQLAlchemy和Blueprints循环导入db引用 | 那些遇到过的问题

dao.py

python 复制代码

from flask import Blueprint, request, jsonify
from models import db
from models import KnowledgeEntity
from sqlalchemy.exc import SQLAlchemyError


# 创建视图函数蓝图
app = Blueprint('KnowledgeDAO', __name__)

'''
word = db.Column(db.String(255), primary_key=True)
content = db.Column(db.Text, unique=True, nullable=False)
priority = db.Column(db.String(255), unique=True, nullable=False)
association = db.Column(db.JSON, unique=True, nullable=False)
'''


# 创建用户
@app.route('/add', methods=['POST'])
def create_user():
    data = request.get_json()
    word = data.get('word')
    content = data.get('content')
    priority = data.get('priority')
    association = data.get('association')

    if not word or not priority:
        return jsonify({'message': 'Both word and priority are required'}), 400

    entity = KnowledgeEntity(word=word, content=content, priority=priority, association=association)

    try:
        db.session.add(entity)
        db.session.commit()
        return jsonify(1), 201
    except SQLAlchemyError as e:
        db.session.rollback()  # 回滚事务以撤销之前的操作
        error_message = str(e)
        return f'Error: {error_message}', 500


# 查询所有用户
@app.route('/selectAll', methods=['POST'])
def get_users():


    users = KnowledgeEntity.query.all()
    print('users',users)
    # user_list = [{'id': user.id, 'username': user.username, 'email': user.email} for user in users]
    return jsonify(1)

model.py

python 复制代码

from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()

class KnowledgeEntity(db.Model):
    __tablename__ = 'knowledge'
    word = db.Column(db.String(255), primary_key=True)
    content = db.Column(db.Text, unique=True, nullable=False)
    priority = db.Column(db.String(255), unique=True, nullable=False)
    association = db.Column(db.JSON, unique=True, nullable=False)

    def __init__(self, word, content=None, priority=None, association=None):
        self.word = word
        self.content = content
        self.priority = priority
        self.association = association

app.py（改进后的）

python 复制代码

from flask import Flask, request, jsonify,g
from flask_cors import CORS
from flask_sqlalchemy import SQLAlchemy
from models import db
# 导入视图函数
from DAO import KnowledgeDAO

app = Flask(__name__)
# 配置选项，用于控制 Flask 在将 Python 字典转换为 JSON 数据时是否按照键的字母顺序对键进行排序，默认情况下，它的值为 True，表示会对键进行排序。
app.config['JSON_SORT_KEYS'] = False
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://root:123456@127.0.0.1:3306/ennote'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

# 注册跨域
CORS(app, resources=r'/*')  # 注册CORS, "/*" 允许访问所有api

# 连接数据库
db.init_app(app)

# 注册蓝图
app.register_blueprint(KnowledgeDAO.app, url_prefix='/knowledge')

if __name__ == '__main__':


    # 创建应用上下文
    with app.app_context():
        # 在这里执行需要应用上下文的操作
        # 例如，访问数据库或使用Flask的全局变量
        db.create_all()

    app.run(host='127.0.0.1', port=5000, debug=True)