Start work on admin panel

1 files changed, 21 insertions, 0 deletions

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