我有两个用于flask_login的用户模型:
class Admin(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(30))
password_hash = db.Column(db.String(200))
class Merchant(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(30))
password_hash = db.Column(db.String(200))
现在我想在会话中加载用户:
@login_manager.user_loader
def load_user(user_id):
pass
我想知道如何从两个模型加载用户。
#You can create a permission
admin_permission = Permission(RoleNeed('admin'))
#protect a view with a principal for that need
@app.route('/admin')
@admin_permission.required()
def do_admin_index():
return Response('Only if you are an admin)