Files
yrtv/web/auth.py

12 lines
365 B
Python
Raw Normal View History

2026-01-26 02:13:06 +08:00
from functools import wraps
from flask import session, redirect, url_for, flash
def admin_required(f):
@wraps(f)
def decorated_function(*args, **kwargs):
if session.get('is_admin'):
return f(*args, **kwargs)
flash('Admin access required', 'warning')
return redirect(url_for('admin.login'))
return decorated_function