mirror of
https://github.com/evgen-app/chess_rpg_backend.git
synced 2024-11-26 19:44:01 +03:00
33 lines
955 B
Python
33 lines
955 B
Python
|
from channels.db import database_sync_to_async
|
||
|
|
from django.core.exceptions import PermissionDenied
|
||
|
|
|
||
|
|
from game.models import Player
|
||
|
|
from game.services.jwt import read_jwt
|
||
|
|
|
||
|
|
|
||
|
|
@database_sync_to_async
|
||
|
|
def get_player(jwt):
|
||
|
|
payload = read_jwt(jwt)
|
||
|
|
if not payload:
|
||
|
|
raise PermissionDenied
|
||
|
|
try:
|
||
|
|
return Player.objects.get(id=payload)
|
||
|
|
except User.DoesNotExist:
|
||
|
|
return AnonymousUser()
|
||
|
|
|
||
|
|
|
||
|
|
class QueryAuthMiddleware:
|
||
|
|
"""Custom middleware to read user auth token from string."""
|
||
|
|
|
||
|
|
def __init__(self, app):
|
||
|
|
# Store the ASGI application we were passed
|
||
|
|
self.app = app
|
||
|
|
|
||
|
|
async def __call__(self, scope, receive, send):
|
||
|
|
# Look up user from query string (you should also do things like
|
||
|
|
# checking if it is a valid user ID, or if scope["user"] is already
|
||
|
|
# populated).
|
||
|
|
scope["user"] = await get_user(int(scope["query_string"]))
|
||
|
|
|
||
|
|
return await self.app(scope, receive, send)
|