1. 程式人生 > >flask中權限訪問裝飾器

flask中權限訪問裝飾器

根據 req map ted split rst ret gpo clas

根據session裏面保存的管理員id查詢出對於的角色,根據角色查詢出權限,根據權限查詢出能夠訪問的路徑

# 權限控制裝飾器
def admin_auth(f):
@wraps(f)
def decorated_function(*args, **kwargs): # 讓某個函數來繼承我們的參數
admin = Admin.query.join(
Role
).filter(
Role.id == Admin.role_id,
Admin.id == session[‘admin_id‘]
).first()
auths = admin.role.auths
auths = list(map(lambda v: int(v), auths.split(‘,‘)))
auth_list = Auth.query().all()
urls = [v.url for v in auth_list for val in auths if v.id == val.id]
rule=request.url_rule
if rule not in urls:
abort(404)
return f(*args, **kwargs)

return decorated_function

flask中權限訪問裝飾器