Egor fureunoir Gorbunov
a81f734e23
Fixes: (1) Removed all `# type: ignore` annotations across the codebase; (2) Fixed usage of Django Model methods by eliminating unnecessary `# type: ignore` directives; (3) Adjusted usage of functions like `get()` to align with method expectations, removing incorrect comments; Extra: (1) Deleted `pyrightconfig.json` as part of migration to a stricter type-checked environment; (2) Minor code cleanup, including formatting changes and refactoring import statements in adherence to PEP8 recommendations.
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
from contextlib import suppress
|
|
from typing import Iterable
|
|
|
|
from channels.middleware import BaseMiddleware
|
|
from django.contrib.auth.models import AnonymousUser
|
|
from django.utils.functional import LazyObject
|
|
from rest_framework_simplejwt.authentication import JWTAuthentication
|
|
|
|
|
|
class _LazyUser(LazyObject):
|
|
def _setup(self):
|
|
self._wrapped = AnonymousUser()
|
|
|
|
|
|
def _extract_jwt_from_subprotocols(subprotocols: Iterable[str] | None) -> str | None:
|
|
if not subprotocols:
|
|
return None
|
|
items = list(subprotocols)
|
|
if len(items) >= 2 and items[0].lower() == "bearer" and items[1]:
|
|
return items[1]
|
|
if len(items) == 1 and items[0]:
|
|
return items[0]
|
|
return None
|
|
|
|
|
|
class JWTAuthMiddleware(BaseMiddleware):
|
|
async def __call__(self, scope, receive, send):
|
|
scope["user"] = _LazyUser()
|
|
|
|
token = _extract_jwt_from_subprotocols(scope.get("subprotocols"))
|
|
if token:
|
|
jwt_auth = JWTAuthentication()
|
|
with suppress(Exception):
|
|
|
|
class _Req:
|
|
def __init__(self, token_str: str):
|
|
self.META = {"HTTP_X_EVIBES_AUTH": f"Bearer {token_str}"}
|
|
|
|
user, _ = jwt_auth.authenticate(_Req(token))
|
|
scope["user"] = user
|
|
|
|
return await super().__call__(scope, receive, send)
|
|
|
|
|
|
def JWTAuthMiddlewareStack(inner):
|
|
return JWTAuthMiddleware(inner)
|