From 0694462ae0c7953c0f5aa189ddc1cbe178b38f4f Mon Sep 17 00:00:00 2001 From: gabigab117 Date: Fri, 17 May 2024 19:48:45 +0200 Subject: [PATCH] feat: create SuperUserAuth class inheriting from JWTAuth - Added SuperUserAuth class that extends the functionality of JWTAuth. --- ninja_jwt/authentication.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/ninja_jwt/authentication.py b/ninja_jwt/authentication.py index 8f2cd02c7..b2b637663 100644 --- a/ninja_jwt/authentication.py +++ b/ninja_jwt/authentication.py @@ -137,3 +137,11 @@ class AsyncJWTTokenUserAuth( ): async def authenticate(self, request: HttpRequest, token: str) -> Any: return await self.async_jwt_authenticate(request, token) + + +class SuperUserAuth(JWTAuth): + def get_user(self, validated_token): + user = super().get_user(validated_token) + if not user.is_superuser: + raise AuthenticationFailed("User must be superuser") + return user