From bff5795ed10c923f0e26781ed53ff04fd2284923 Mon Sep 17 00:00:00 2001 From: Gertjan van den Burg Date: Mon, 18 Mar 2019 18:30:21 +0000 Subject: Start work on admin panel --- app/admin/decorators.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 app/admin/decorators.py (limited to 'app/admin/decorators.py') diff --git a/app/admin/decorators.py b/app/admin/decorators.py new file mode 100644 index 0000000..f42b582 --- /dev/null +++ b/app/admin/decorators.py @@ -0,0 +1,21 @@ +# -*- coding: utf-8 -*- + +from functools import wraps + +from flask import current_app, request +from flask_login import current_user +from flask_login.config import EXEMPT_METHODS + + +def admin_required(func): + @wraps(func) + def decorated_view(*args, **kwargs): + if request.method in EXEMPT_METHODS: + return func(*args, **kwargs) + elif current_app.config.get("LOGIN_DISABLED"): + return func(*args, **kwargs) + elif not current_user.is_admin: + return current_app.login_manager.unauthorized() + return func(*args, **kwargs) + + return decorated_view -- cgit v1.2.3