在flask-security中设置默认角色

Question

我正在尝试在用户向我的网站注册时设置默认角色，当前在用户注册时未设置任何角色。我已经创建了所需的角色，因此只需要以某种方式对其进行定义。不确定如何。我的代码几乎是快速入门指南中的复制粘贴。无论如何，这里是：# Define modelsroles_users = db.Table('roles_users',        db.Column('user_id', db.Integer(), db.ForeignKey('user.id')),        db.Column('role_id', db.Integer(), db.ForeignKey('role.id')))class Role(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 User(db.Model, UserMixin):    id = db.Column(db.Integer, primary_key=True)    email = db.Column(db.String(255), unique=True)    password = db.Column(db.String(255))    active = db.Column(db.Boolean())    confirmed_at = db.Column(db.DateTime())    roles = db.relationship('Role', secondary=roles_users, backref=db.backref('users', lazy='dynamic'))# Setup Flask-Securityuser_datastore = SQLAlchemyUserDatastore(db, User, Role)security = Security(app, user_datastore)import users.getUser as getUser#@app.before_first_requestdef create_user():    db.create_all()    user_datastore.create_user(email='[e-mail]', password='password')    db.session.commit()@flask_sijax.route(app, '/')def test():    print user_datastore.create_role(name='User', description='Generic user role')    db.session.commit()    return render_template('test.html')

Answer 1

我使用此git问题中的信息对此进行了修复：

@user_registered.connect_via(app)

def user_registered_sighandler(app, user, confirm_token):

    default_role = user_datastore.find_role("User")

    user_datastore.add_role_to_user(user, default_role)

    db.session.commit()

Answer 2

如果您已经创建了角色“用户”，则可以在创建/注册用户时将该角色添加到用户。

def create_user():

    db.create_all()

    user = user_datastore.create_user(email='[e-mail]', password='password')

    default_role = user_datastore.find_role(name="User")

    user_datastore.add_role_to_user(user, default_role)

    db.session.commit()