我正在嘗試在用戶(hù)向我的網(wǎng)站注冊(cè)時(shí)設(shè)置默認(rèn)角色，當(dāng)前在用戶(hù)注冊(cè)時(shí)未設(shè)置任何角色。我已經(jīng)創(chuàng)建了所需的角色，因此只需要以某種方式對(duì)其進(jìn)行定義。不確定如何。我的代碼幾乎是快速入門(mén)指南中的復(fù)制粘貼。無(wú)論如何，這里是：# 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='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')
2 回答
MMMHUHU
TA貢獻(xiàn)1834條經(jīng)驗(yàn) 獲得超8個(gè)贊
我使用此git問(wèn)題中的信息對(duì)此進(jìn)行了修復(fù)：
@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()
HUH函數(shù)
TA貢獻(xiàn)1836條經(jīng)驗(yàn) 獲得超4個(gè)贊
如果您已經(jīng)創(chuàng)建了角色“用戶(hù)”，則可以在創(chuàng)建/注冊(cè)用戶(hù)時(shí)將該角色添加到用戶(hù)。
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()
添加回答
舉報(bào)
0/150
提交
取消