47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
"""Metrics view"""
|
|
from base64 import b64encode
|
|
|
|
from django.conf import settings
|
|
from django.db import connections
|
|
from django.db.utils import OperationalError
|
|
from django.http import HttpRequest, HttpResponse
|
|
from django.views import View
|
|
from django_prometheus.exports import ExportToDjangoView
|
|
|
|
|
|
class MetricsView(View):
|
|
"""Wrapper around ExportToDjangoView, using http-basic auth"""
|
|
|
|
def get(self, request: HttpRequest) -> HttpResponse:
|
|
"""Check for HTTP-Basic auth"""
|
|
auth_header = request.META.get("HTTP_AUTHORIZATION", "")
|
|
auth_type, _, given_credentials = auth_header.partition(" ")
|
|
credentials = f"monitor:{settings.SECRET_KEY}"
|
|
expected = b64encode(str.encode(credentials)).decode()
|
|
|
|
if auth_type != "Basic" or given_credentials != expected:
|
|
response = HttpResponse(status=401)
|
|
response["WWW-Authenticate"] = 'Basic realm="authentik-monitoring"'
|
|
return response
|
|
|
|
return ExportToDjangoView(request)
|
|
|
|
|
|
class LiveView(View):
|
|
"""View for liveness probe, always returns Http 201"""
|
|
|
|
def dispatch(self, request: HttpRequest) -> HttpResponse:
|
|
return HttpResponse(status=201)
|
|
|
|
|
|
class ReadyView(View):
|
|
"""View for liveness probe, always returns Http 201"""
|
|
|
|
def dispatch(self, request: HttpRequest) -> HttpResponse:
|
|
db_conn = connections["default"]
|
|
try:
|
|
_ = db_conn.cursor()
|
|
except OperationalError:
|
|
return HttpResponse(status=503)
|
|
return HttpResponse(status=201)
|