from django.contrib.auth import login
from django.utils.deprecation import MiddlewareMixin

from rest_framework.authtoken.models import Token


class TokenAuthLoginMiddleware(MiddlewareMixin):

    def process_request(self, request):
        if 'token' in request.GET:
            token = request.GET.get('token')
            if token:
                try:
                    token = Token.objects.get(key=token)
                    user = token.user
                    login(request, user)
                except Token.DoesNotExist:
                    pass