takahe/core/middleware.py

76 lines
2.1 KiB
Python
Raw Permalink Normal View History

2022-12-05 17:55:30 +00:00
from time import time
2022-12-20 15:26:39 +00:00
from django.conf import settings
from django.core.exceptions import MiddlewareNotUsed
from core import sentry
2022-11-18 07:09:04 +00:00
from core.models import Config
2022-12-15 07:35:04 +00:00
class HeadersMiddleware:
"""
2022-12-15 07:35:04 +00:00
Deals with Accept request headers, and Cache-Control response ones.
"""
def __init__(self, get_response):
self.get_response = get_response
def __call__(self, request):
accept = request.headers.get("accept", "text/html").lower()
request.ap_json = (
"application/json" in accept
or "application/ld" in accept
or "application/activity" in accept
)
response = self.get_response(request)
2022-12-15 07:35:04 +00:00
if "Cache-Control" not in response.headers:
2022-12-17 22:35:22 +00:00
response.headers["Cache-Control"] = "no-store, max-age=0"
return response
2022-11-18 07:09:04 +00:00
class ConfigLoadingMiddleware:
"""
Caches the system config every request
"""
refresh_interval: float = 5.0
2022-12-05 17:55:30 +00:00
2022-11-18 07:09:04 +00:00
def __init__(self, get_response):
self.get_response = get_response
2022-12-05 17:55:30 +00:00
self.config_ts: float = 0.0
2022-11-18 07:09:04 +00:00
def __call__(self, request):
# Allow test fixtures to force and lock the config
if not getattr(Config, "__forced__", False):
if (
not getattr(Config, "system", None)
or (time() - self.config_ts) >= self.refresh_interval
):
Config.system = Config.load_system()
self.config_ts = time()
2022-11-18 07:09:04 +00:00
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
2022-12-20 15:26:39 +00:00
def show_toolbar(request):
"""
Determines whether to show the debug toolbar on a given page.
"""
return settings.DEBUG and request.user.is_authenticated and request.user.admin