50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
|
from random import random
|
||
|
|
from django.conf import settings
|
||
|
|
from django.shortcuts import render
|
||
|
|
from django.views.generic import View
|
||
|
|
from django.http import HttpResponse, HttpResponseServerError
|
||
|
|
from django.core.exceptions import PermissionDenied
|
||
|
|
from django.db import connection
|
||
|
|
from .utils import get_client_ip
|
||
|
|
|
||
|
|
|
||
|
|
def error403(request, *args, **kwargs):
|
||
|
|
return render(request, 'error403.html', status=403)
|
||
|
|
|
||
|
|
|
||
|
|
def error404(request, *args, **kwargs):
|
||
|
|
return render(request, 'error404.html', status=404)
|
||
|
|
|
||
|
|
|
||
|
|
def error500(request, *args, **kwargs):
|
||
|
|
return render(request, 'error500.html', status=500)
|
||
|
|
|
||
|
|
|
||
|
|
class HealthCheckView(View):
|
||
|
|
'''
|
||
|
|
A basic healthcheck view. SELECTs a random int via the database connection
|
||
|
|
and verifies it matches. This checks that the application server, django and
|
||
|
|
the database connection are all working correctly.
|
||
|
|
'''
|
||
|
|
|
||
|
|
ALLOWED_IPS = settings.HEALTHCHECK_ALLOWED_IPS
|
||
|
|
|
||
|
|
def get(self, request, *args, **kwargs):
|
||
|
|
if settings.HEALTHCHECK_FIREWALL:
|
||
|
|
client_ip = get_client_ip(request)
|
||
|
|
if client_ip not in self.ALLOWED_IPS:
|
||
|
|
raise PermissionDenied
|
||
|
|
randomint = int(random() * (10 ** 10))
|
||
|
|
with connection.cursor() as cursor:
|
||
|
|
cursor.execute('select {}'.format(randomint))
|
||
|
|
row = cursor.fetchone()
|
||
|
|
try:
|
||
|
|
pong = row[0]
|
||
|
|
except IndexError:
|
||
|
|
pong = False
|
||
|
|
if str(pong) != str(randomint):
|
||
|
|
err = 'Failed healtcheck, expected "{}" got "{}"'
|
||
|
|
return HttpResponseServerError(err.format(randomint, pong))
|
||
|
|
else:
|
||
|
|
return HttpResponse('ok')
|