2

我正在尝试在用户注册我的网站时设置默认角色，目前在用户注册时没有设置角色。

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

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

# 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')
4

2 回答 2

10

我使用此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()
于 2013-06-17T15:07:54.737 回答
-1

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

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()
于 2013-06-17T13:13:42.987 回答