tabby-web/backend/tabby/middleware.py

import logging
from tabby.app.models import User
from django.conf import settings
from django.contrib.auth import login
from pyga.requests import Tracker, Page, Session, Visitor


class BaseMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response


class TokenMiddleware(BaseMiddleware):
    def __call__(self, request):
        token_value = None
        if "auth_token" in request.GET:
            token_value = request.GET["auth_token"]
        if request.META.get("HTTP_AUTHORIZATION"):
            token_type, *credentials = request.META["HTTP_AUTHORIZATION"].split()
            if token_type == "Bearer" and len(credentials):
                token_value = credentials[0]

        user = User.objects.filter(config_sync_token=token_value).first()

        if user:
            request.session.save = lambda *args, **kwargs: None
            setattr(user, "backend", "django.contrib.auth.backends.ModelBackend")
            login(request, user)
            setattr(request, "_dont_enforce_csrf_checks", True)

        response = self.get_response(request)

        if user:
            response.set_cookie = lambda *args, **kwargs: None

        return response


class GAMiddleware(BaseMiddleware):
    def __init__(self, get_response):
        super().__init__(get_response)
        if settings.GA_ID:
            self.tracker = Tracker(settings.GA_ID, settings.GA_DOMAIN)

    def __call__(self, request):
        response = self.get_response(request)
        if settings.GA_ID and request.path in ["/", "/app"]:
            try:
                self.tracker.track_pageview(Page(request.path), Session(), Visitor())
            except Exception:
                logging.exception()

        return response