在flask-security中设置默认角色 [英] Setting a default role in flask-security

问题描述

我已经创建了角色I需要，所以我只需要以某种方式定义它。不知道如何。

我的代码是从快速入门指南几乎复制粘贴。无论如何，这里是：

$ p $ ＃定义模型
roles_users = db.Table（'roles_users'，$ b $ db.Column（'user_id'，db.Integer（），db.ForeignKey（'user.id'）），
db.Column（'role_id'，db.Integer（），db.ForeignKey （b.Model，RoleMixin）：
id = db.Column（db.Integer（），primary_key = True）
name = db.Column（db.String（80），unique = True）
description = db.Column（db.String（255））
$ b $ 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（'角色'，secondary = roles_users，backref = db.backref（'users'，lazy ='dynamic'））

＃安装Flask-Security
user_datastore = SQLAlchemyUserDatastore（db，User，Role ）
安全= Security（app，user_datastore）


导入users.getUser as getUser

#@app.before_first_request
def create_user（）：
$ db.create_all（）
user_datastore.create_user（email='[e-mail]'，password ='password'）
db.session.commit（）

@ （））
def test（）：
print user_datastore.create_role（name ='User'，description ='Generic user role'）
db.session。 commit（）
return render_template（'test.html'）


解决方案我使用这个 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（）
  

I am trying to set a default role when a user registers with my site, currently no roles are set when the user registers.

I have created the roles I need, so I just need to define it somehow. Not sure how though.

The code I have is pretty much copy paste from the quick start guide. Anyway, here it is:

# Define models
roles_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-Security
user_datastore = SQLAlchemyUserDatastore(db, User, Role)
security = Security(app, user_datastore)


import users.getUser as getUser

#@app.before_first_request
def 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')

解决方案

I fixed it using information from this git issue with this:

@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()

这篇关于在flask-security中设置默认角色的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！