takahe/core/middleware.py

from time import time

from django.core.exceptions import MiddlewareNotUsed

from core import sentry
from core.models import Config


class ConfigLoadingMiddleware:
    """
    Caches the system config every request
    """

    refresh_interval: float = 30.0

    def __init__(self, get_response):
        self.get_response = get_response
        self.config_ts: float = 0.0

    def __call__(self, request):
        if (
            not getattr(Config, "system", None)
            or (time() - self.config_ts) >= self.refresh_interval
        ):
            Config.system = Config.load_system()
            self.config_ts = time()
        response = self.get_response(request)
        return response


class SentryTaggingMiddleware:
    """
    Sets Sentry tags at the start of the request if Sentry is configured.
    """

    def __init__(self, get_response):
        if not sentry.SENTRY_ENABLED:
            raise MiddlewareNotUsed()
        self.get_response = get_response

    def __call__(self, request):
        sentry.set_takahe_app("web")
        response = self.get_response(request)
        return response