mirror of
https://github.com/ail-project/ail-framework.git
synced 2024-09-20 07:58:29 +00:00
18 lines
570 B
Python
18 lines
570 B
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*-coding:UTF-8 -*
|
||
|
|
|
||
|
|
from functools import wraps
|
||
|
|
from flask_login import LoginManager, current_user, login_user, logout_user, login_required
|
||
|
|
|
||
|
|
from flask import request
|
||
|
|
|
||
|
|
def login_required(role="ANY"):
|
||
|
|
@wraps(role)
|
||
|
|
def decorated_view(*args, **kwargs):
|
||
|
|
if not current_user.is_authenticated:
|
||
|
|
return current_app.login_manager.unauthorized()
|
||
|
|
elif (not current_user.is_in_role(role)) and (role != "ANY"):
|
||
|
|
return login_manager.unauthorized()
|
||
|
|
return func(*args, **kwargs)
|
||
|
|
return decorated_view
|