initial commit

1 files changed, 22 insertions, 0 deletions

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