Python 在flask安全性中设置默认角色
当用户向我的站点注册时,我正在尝试设置默认角色,当前用户注册时未设置任何角色 我已经创建了我需要的角色,所以我只需要以某种方式定义它。但不知道怎么办 我的代码基本上是从《快速入门指南》中复制粘贴的。无论如何,这是:Python 在flask安全性中设置默认角色,python,flask,flask-sqlalchemy,flask-security,Python,Flask,Flask Sqlalchemy,Flask Security,当用户向我的站点注册时,我正在尝试设置默认角色,当前用户注册时未设置任何角色 我已经创建了我需要的角色,所以我只需要以某种方式定义它。但不知道怎么办 我的代码基本上是从《快速入门指南》中复制粘贴的。无论如何,这是: # Define models roles_users = db.Table('roles_users', db.Column('user_id', db.Integer(), db.ForeignKey('user.id')), db.Column(
# 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='my@email.com', 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')
如果已经创建了角色“用户”,则可以在创建/注册用户时将该角色添加到该用户
def create_user():
db.create_all()
user = user_datastore.create_user(email='my@email.com', password='password')
default_role = user_datastore.find_role(name="User")
user_datastore.add_role_to_user(user, default_role)
db.session.commit()
我使用以下信息修复了它:
@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()
我添加了使用
user\u registered.connect(user\u registered\u sighandler)
而不是decorator来完成这项工作。