22 lines
785 B
Python
22 lines
785 B
Python
"""Metrics view"""
|
|
from base64 import b64encode
|
|
from django.conf import settings
|
|
from django.views import View
|
|
from django.http import HttpRequest, HttpResponse, Http404
|
|
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', '')
|
|
token_type, _, credentials = auth_header.partition(' ')
|
|
creds = f"monitor:{settings.SECRET_KEY}"
|
|
expected = b64encode(str.encode(creds)).decode()
|
|
|
|
if token_type != 'Basic' or credentials != expected:
|
|
raise Http404
|
|
|
|
return ExportToDjangoView(request)
|