Flask数据库——分开Models和解决循环引用

Dear 丶 2021-12-05 02:15 178阅读 0赞

分开models目的：为了让代码更加方便的管理

项目代码描述：  
[models.py][]

# -*- coding:utf-8 -*-
    from app import db
    
    class Article(db.Model):
        __tablename__ = 'article'
        id = db.Column(db.Integer, primary_key=True, autoincrement=True)

[app.py][]

from flask import Flask
    from flask_sqlalchemy import SQLAlchemy
    import config
    from models import Article
    
    app = Flask(__name__)
    app.config.from_object(config)
    db = SQLAlchemy(app)
    
    db.create_all()
    
    @app.route('/')
    def hello_world():
        return 'Hello World!'
    
    if __name__ == '__main__':
        app.run()

点击运行项目后报错：

ImportError: cannot import name 'db' from 'app'

**问题原因：** Flask中的循环引用  
![Alt][]  
**解决方法：** app引用model本身就是我们需要达到的目的，所有只有将db文件进行分离  
![Alt][Alt 1]

1.  添加文件 [exts.py][]

# -*- coding:utf-8 -*-
    
    from flask_sqlalchemy import SQLAlchemy
    
    db = SQLAlchemy()

1.  修改 [models.py][] 中的db导入文件

# -*- coding:utf-8 -*-
    
    from exts import db
    
    class Article(db.Model):
        __tablename__ = 'article'
        id = db.Column(db.Integer, primary_key=True, autoincrement=True)

1.  修改文件 [app.py][]

from flask import Flask
    import config
    from models import Article
    from exts import db
    
    app = Flask(__name__)
    app.config.from_object(config)
    db.init_app(app)
    app.app_context().push()
    
    db.create_all()
    
    @app.route('/')
    def hello_world():
        return 'Hello World!'
    
    if __name__ == '__main__':
        app.run()

db此时需要利用另外语句进行初始化

db.init_app(app)
    app.app_context().push()

[models.py]: http://models.py
[app.py]: http://app.py
[Alt]: /images/20211205/bf9b8dfb6ea246798f7044775ee7c1e0.png
[Alt 1]: /images/20211205/a12fc8d3bd1646778bad224d753c6bac.png
[exts.py]: http://exts.py