flask-security ImportError: cannot import name 'db'

Question

我知道这可能是一个循环导入错误，但我想不出解决办法。到目前为止，该应用程序非常普通。这是添加了 models.py 的 flask-security 教程。这是为向项目添加蓝图做准备。

从应用导入数据库导入错误：无法导入名称 'db'

config.py
run.py
---app
------__init__py
------models.py

run.py

from app import app

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

init.py

from config import Config
import app.models

from flask import Flask, render_template
from flask_sqlalchemy import SQLAlchemy
from flask_security import Security, SQLAlchemyUserDatastore, \
    login_required

db = SQLAlchemy()

# Create app
def create_app():
    app = Flask(__name__)
    app.config.from_object(Config)
    app.config['DEBUG'] = True
    db.init_app(app)

    user_datastore = SQLAlchemyUserDatastore(db, models.Users,
                                             models.usersRoleNames)
    security = Security(app, user_datastore)

    return app

# Views
@app.route('/')
@login_required
def home():
    return render_template('index.html')

models.py

from app import db
from flask_security import UserMixin, RoleMixin

users_roles = db.Table('users_roles',
        db.Column('user_id', db.Integer(), db.ForeignKey('users.id')),
        db.Column('role_id', db.Integer(), db.ForeignKey('users_role_names.id')))


class usersRoleNames(db.Model, RoleMixin):
    id = db.Column(db.Integer(), primary_key=True)
    name = db.Column(db.String(80), unique=True)
    description = db.Column(db.String(255))


class Users(db.Model, UserMixin):
    id = db.Column(db.Integer, primary_key=True)
    email = db.Column(db.String(255), unique=True)
    username = db.Column(db.String(255))
    password = db.Column(db.String(255))
    last_login_at = db.Column(db.DateTime())
    current_login_at = db.Column(db.DateTime())
    last_login_ip = db.Column(db.String(100))
    current_login_ip = db.Column(db.String(100))
    login_count = db.Column(db.Integer)
    active = db.Column(db.Boolean())
    confirmed_at = db.Column(db.DateTime())
    roles = db.relationship('usersRoleNames', secondary=users_roles,
                            backref=db.backref('users',
                                               lazy='dynamic'))

Answer 1

尝试在 models.py 中声明您的 db 对象并将其导入 app/__init_.py

models.py

from flask_sqlalchemy import SQLAlchemy                                         

db = SQLAlchemy()
...

app/__init__.py

def create_app():                                                               
    app = Flask(__name__)                                                       
    app.config.from_object(Config)                                              
    app.config['DEBUG'] = True                                                  

    from app.models import db, Users, usersRoleNames                            
    db.init_app(app)                                                                                                             

    user_datastore = SQLAlchemyUserDatastore(db, Users,                         
                                             usersRoleNames)                    
    security = Security(app, user_datastore)                                    

    return app

Answer 2

我找到了 flask-security 循环导入问题的简单解决方案。当跨不同文件声明模型时，此解决方案特别有用。

1。创建一个名为 `models`:

的文件夹

我们所有的模型都将保存在这个文件夹中。

示例文件夹结构可能如下所示

models/
├── _db.py
├── __init__.py
├── tickets.py
└── users.py

0 directories, 4 files

2。创建一个名为 `_db.py`:

的文件

在此文件中声明 db 对象。

文件名可以是任何东西，但文件名不能以 下划线开头 确保在对导入语句排序时 db 对象总是首先导入。

('VS Code 提供了一个自动对导入进行排序的选项，因此使用它不会导致任何问题。')

_db.py

from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()

3。从 `_db.py` 文件导入 `db` 对象：

现在我们可以从上面创建的 _db.py 文件导入 db 对象：

users.py

from ._db import db
from flask_security import RoleMixin, UserMixin

users_roles = db.Table(
    # Table definition
)

class usersRoleNames(db.Model, RoleMixin):
    # Table definition
    ...

class Users(db.Model, UserMixin):
    # Table definition
    ...

4。导入 `init.py` 文件中的所有内容（可选）：

创建一个 init.py 文件并导入其中的所有模型和 db 对象。

__init__.py

from ._db import db
from .tickets import Tickets
from .users import Users, users_roles, usersRoleNames

'''
Below part is optional 
but I like to do it anyways
'''

__all__ = [
    'db',
    'Tickets',
    'Users',
    'users_roles',
    'usersRoleNames',
]

通过导入 __init__.py 文件中的所有内容，我们可以直接从模型文件夹中导入。

所以我们可以做到

from .models import db, Tickets, Users

而不是这个

from .models._db import _db
from .models.tickets import Tickets
from .models.users import Users

只是一个想法： 我不确定以 'underscore' 开头的 python 文件名是否是一个坏习惯。如果您有意见，请分享您的想法。

flask-security ImportError: cannot import name 'db'