Add djangorestframework-jwt auth handler

Related to tfranzel#28
jayvdb · May 3, 2020 · 2f48a47 · 2f48a47
1 parent c21a177
commit 2f48a47
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/drf_spectacular/contrib/authentication.py b/drf_spectacular/contrib/authentication.py
@@ -21,6 +21,20 @@ def get_security_definition(self, auto_schema):
         }
 
 
+class JWTScheme(OpenApiAuthenticationExtension):
+    target_class = 'rest_framework_jwt.authentication.JSONWebTokenAuthentication'
+    name = 'jwtAuth'
+
+    def get_security_definition(self, auto_schema):
+        from rest_framework_jwt.settings import api_settings
+
+        return {
+            'type': 'http',
+            'scheme': 'bearer',
+            'bearerFormat': api_settings.JWT_AUTH_HEADER_PREFIX,
+        }
+
+
 class DjangoOAuthToolkitScheme(OpenApiAuthenticationExtension):
     target_class = 'oauth2_provider.contrib.rest_framework.OAuth2Authentication'
     name = 'oauth2'