Question

我正在尝试创建一个骨架，我正在使用flask安全性进行登录/管理。

在我的app文件夹中，我有一个创建应用程序的功能：

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_security import Security, SQLAlchemyUserDatastore

from .views.home import home_blueprint
from .views.contact import contact_blueprint
from .views.login import login_blueprint, bcrypt, login_manager
from .models import db
from .models.user import User, Role

def create_app():
    app = Flask(__name__, instance_relative_config=True)
    db.init_app(app)
    bcrypt.init_app(app)
    login_manager.init_app(app)
    app.register_blueprint(home_blueprint)
    app.register_blueprint(contact_blueprint)
    app.register_blueprint(login_blueprint)
    app.config.from_object('config')
    user_datastore = SQLAlchemyUserDatastore(db, User, Role)
    security = Security(app, user_datastore)
    return app

现在，当我尝试创建一个虚拟管理员用户进行测试时，问题出现了。为此，我在骨架根文件夹中有一个create_db.py文件。

from app import create_app
from sqlalchemy import exists
from flask_security.utils import encrypt_password

from app.models import db
from app.models.user import Role, User

create_app().app_context().push()
db.create_all()

user_role = Role(name='user')
db.session.add(user_role)
db.session.commit()

super_user_role = Role(name='superuser')
db.session.add(super_user_role)
db.session.commit()

user_role = Role(name='user')
super_user_role = Role(name='superuser')
admin_user = user_datastore.create_user(
    username='admin',
    email='admin',
    password=encrypt_password('admin'),
    confirmed_at=datetime.datetime.now(),
    roles=[user_role, super_user_role]
)
db.session.commit()

当我运行这个时，我得到NameError：user_datastore没有定义？

出于故事目的，我首先实现了flask-login，并且能够创建数据库。

Answer 1

在文件create_db.py中，变量user_datastore不存在因此错误。您可以在create_dy.py（未经测试）中执行以下操作：

app = create_app()
app.app_context().push()
# get the app security instance
security = app.extensions.get('security')
# or like this
# security = app.extensions['security']


# ... etc

# Use the security api to create users
admin_user = security.datastore.create_user(
    username='admin',
    email='admin',
    password=encrypt_password('admin'),
    confirmed_at=datetime.datetime.now(),
    roles=[user_role, super_user_role]
)

# ... etc

此外，我还不确定您需要login_manager.init_app(app)功能create_app()。

试图实现烧瓶安全性

1 个答案: