from rest_framework.authtoken.models import Token
from django.contrib.auth.models import AnonymousUser
from django.db import close_old_connections
from asgiref.sync import sync_to_async
class TokenAuthMiddleware:
    Token authorization middleware for Django Channels 2
    def __init__(self, inner):
        self.inner = inner
    def __call__(self, scope):
        # Close old database connections to prevent usage of timed out connections
        sync_to_async(close_old_connections)()
        headers = dict(scope['headers'])
            token_name, token_key = headers[b'sec-websocket-protocol'].decode().split(', ')
            if token_name == 'Token':
                token = sync_to_async(Token.objects.get, thread_sensitive=True)(key=token_name)
                scope['user'] = token.user
            else:
                scope['user'] = AnonymousUser()
        except Token.DoesNotExist:
            scope['user'] = AnonymousUser()