Skip to content
Snippets Groups Projects

Access checking based on role

Merged Andrea Callia D'Iddio requested to merge improve-permissions into master
All threads resolved!
+ 27
0
from functools import wraps
from flask import url_for
from flask_login import current_user
from werkzeug.utils import redirect
from app.constants import STAFF, STUDENT
def staff_only(func):
@wraps(func)
def inner(*args, **kwargs):
if current_user.role == STAFF:
return func(*args, **kwargs)
return redirect(url_for(f"{current_user.role}.projects"))
return inner
def students_only(func):
@wraps(func)
def inner(*args, **kwargs):
if current_user.role == STUDENT:
return func(*args, **kwargs)
return redirect(url_for(f"{current_user.role}.projects"))
return inner
Loading