Files
WebTech/application/decorators.py

17 lines
527 B
Python

from typing import Callable, Any
from flask_login import current_user # type: ignore
from functools import wraps
from flask import redirect, url_for
def admin_required(f: Callable[..., Any]) -> Callable[..., Any]:
@wraps(f)
def decorated_function(*args: ..., **kwargs: ...):
if not current_user.is_authenticated:
return redirect(url_for("login"))
if not current_user.is_admin:
return redirect(url_for("index"))
return f(*args, **kwargs)
return decorated_function