From 3fca03988b42adaf8e67cc7137dd1fdba327e197 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jes=C3=BAs?= <heckyel@hyperbola.info>
Date: Tue, 22 Mar 2022 00:39:40 +0800
Subject: initial commit

---
 utils/decorators.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 utils/decorators.py

(limited to 'utils/decorators.py')

diff --git a/utils/decorators.py b/utils/decorators.py
new file mode 100644
index 0000000..3b12ead
--- /dev/null
+++ b/utils/decorators.py
@@ -0,0 +1,22 @@
+"""
+Decorators
+"""
+# from config import only_get
+from utils.errors import error_method
+from database.models import UserModel
+from flask_jwt_extended import get_jwt_identity
+
+
+def superuser(func):
+    def wrapper(*args, **kwargs):
+        """Check if user is admin"""
+        user_id = get_jwt_identity()
+        user_admin = UserModel.objects(id=user_id, admin=True)
+        if user_admin:
+            return func(*args, **kwargs)
+        else:
+            response = error_method()
+            return response
+
+    wrapper.__name__ = func.__name__
+    return wrapper
-- 
cgit v1.2.3