create custom TokenAuthentication class

xshapira · Dec 27, 2023 · c1d05cb · c1d05cb
1 parent 5cca509
commit c1d05cb
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/auth/__init__.py b/auth/__init__.py
diff --git a/auth/custom_token_auth.py b/auth/custom_token_auth.py
@@ -0,0 +1,13 @@
+from rest_framework import authentication
+
+
+class TokenCookieAuthentication(authentication.TokenAuthentication):
+    def authenticate(self, request):
+        # Check for token in cookies first
+        token = request.COOKIES.get("authToken")
+
+        if token:
+            return self.authenticate_credentials(token)
+
+        # Otherwise check normal auth sources
+        return super().authenticate(request)