"""
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