mirror of
https://github.com/StefBuwalda/WebTech.git
synced 2025-10-30 11:19:58 +00:00
18 lines
568 B
Python
18 lines
568 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, flash
|
|
|
|
|
|
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:
|
|
flash("Admins only!")
|
|
return redirect(url_for("index"))
|
|
return f(*args, **kwargs)
|
|
|
|
return decorated_function
|