重构代码时出现 flask-migrate 问题

Question

我使用 flask-migrate 获得了 Python-Flask 应用程序的以下文件结构：

我的问题是

1-我无法在 manage.py

中使用 db 和 create_app

当我这样做时：

$ python manage.py db init

我遇到以下错误：

File "/app/main/model/model.py", line 25, in <module>
    class User(db.Model):
NameError: name 'db' is not defined

(db定义在main.init.py)

我尝试了不同的选择，但都没有成功。

我想将 manage.py 、 model.py 和 main.init.py 保存在单独的文件中。

2- 在 model .py 中我需要 db 。如何让 db 对 model.py 可用？

下面是manage.py

# This file take care of the migrations
# in model.py we have our tables
import os
import unittest

from flask_migrate import Migrate, MigrateCommand
from flask_script import Manager

from app.main import create_app
from app.main import db
# # We import the tables into the migrate tool
from app.main.model import model


app = create_app(os.getenv('BOILERPLATE_ENV') or 'dev')

app.app_context().push()

manager = Manager(app)

migrate = Migrate(app, db)

manager.add_command('db', MigrateCommand)

#### If I add  model.py here all should be easier , but still I have the 
#### issue with 
#### from app.main import create_app , db


@manager.command
def run():
    app.run()


@manager.command
def test():
    """Runs the unit tests."""
    tests = unittest.TestLoader().discover('app/test', pattern='test*.py')
    result = unittest.TextTestRunner(verbosity=2).run(tests)
    if result.wasSuccessful():
        return 0
    return 1

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

这是app.init.py，其中定义了db和create_app

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_bcrypt import Bcrypt
from flask_cors import CORS

from .config import config_by_name

from flask_restful import Resource, Api
# from flask_restplus import Resource
from app.main.controller.api_controller import gconnect, \
     showLogin, createNewTest, getTest, getTests, getIssue, createNewIssue

db = SQLAlchemy()
flask_bcrypt = Bcrypt()


def create_app(config_name):
    app = Flask(__name__)
    app.config.from_object(config_by_name[config_name])
    cors = CORS(app,
                supports_credentials=True,
                resources={r"/api/*":
                           {"origins":
                            ["http://localhost:3000",
                             "http://127.0.0.1:3000"]}})
    api = Api(app)
    db.init_app(app)
    flask_bcrypt.init_app(app)

    api.add_resource(gconnect, '/api/gconnect')
    api.add_resource(showLogin, '/login')
    api.add_resource(createNewTest, '/api/test')
    api.add_resource(getTest, '/api/test/<int:test_id>')
    api.add_resource(getTests, '/api/tests')
    api.add_resource(getIssue, '/api/issue/<int:issue_id>')
    api.add_resource(createNewIssue, '/api/issue')

    return app

这是我的模型（为了简单起见，只是 table 之一）

from sqlalchemy import Column, ForeignKey, Integer, String, DateTime
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship, backref
from sqlalchemy import create_engine
from sqlalchemy.sql import func


# # # This will let sql alchemy know that these clasess
# # # are special Alchemy classes
# Base = declarative_base()



class User(db.Model):

    __tablename__ = 'user'

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(250), nullable=False)
    email = db.Column(db.String(250), nullable=False)
    pictures = db.Column(db.String(250))
    role = db.Column(db.String(25), nullable=True)

我的问题是：

1-我无法在 manage.py

中使用 db 和 create_app

当我这样做时：

$ python manage.py 数据库初始化

我遇到以下错误：

文件“/app/main/model/model.py”，第 25 行，在 class 用户（db.Model）： NameError：名称 'db' 未定义

(db定义在main.init.py)

我尝试了不同的选择，但都没有成功。

我想将 manage.py 、 model.py 和 main.init.py 保存在单独的文件中。

2- 在模型 .py 中我需要数据库。我如何让数据库对 model.py 可用？

Answer 1

一个简单的解决方案是在 __init__.py 之外创建一个单独的初始化文件。例如init.py 初始化 sqlalchemy 和其他扩展的地方。这样它们就可以在所有模块中导入，而不会出现任何循环依赖问题。

然而，更优雅的解决方案是使用 Flask 的 current_app 和 g 代理。它们旨在帮助 Flask 用户规避循环依赖的任何问题。

通常你在 __init__.py 模块中初始化 flask app 并且 __init__.py 模块有时必须从它的子模块导入一些变量。当子模块尝试导入已初始化的扩展时，这会出现问题

根据一般经验，外部模块应该从它们的子模块导入，而不是相反。

所以这里有一种方法可以解决您的问题（引用自 here）：

** __init__.py

from flask import g

def get_db():
    if 'db' not in g:
        g.db = connect_to_database()

    return g.db

@app.teardown_appcontext
def teardown_db():
    db = g.pop('db', None)

    if db is not None:
        db.close()


def init_db():
    db = get_db()

现在您可以通过以下方式轻松地将您的数据库连接导入任何其他模块：

from flask import g

db = g.db
db.do_something()

重构代码时出现 flask-migrate 问题

flask-migrate issue while refactoring code

python

python-3.x

flask-sqlalchemy

factory-pattern

flask-migrate