mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
84 lines
2.8 KiB
Python
84 lines
2.8 KiB
Python
from __future__ import unicode_literals
|
|
import sys
|
|
|
|
from django.conf import settings
|
|
from django.db import ProgrammingError
|
|
from django.http import HttpResponseRedirect
|
|
from django.shortcuts import render
|
|
from django.urls import reverse
|
|
|
|
|
|
BASE_PATH = getattr(settings, 'BASE_PATH', False)
|
|
LOGIN_REQUIRED = getattr(settings, 'LOGIN_REQUIRED', False)
|
|
|
|
|
|
class LoginRequiredMiddleware(object):
|
|
"""
|
|
If LOGIN_REQUIRED is True, redirect all non-authenticated users to the login page.
|
|
"""
|
|
def __init__(self, get_response):
|
|
self.get_response = get_response
|
|
|
|
def __call__(self, request):
|
|
if LOGIN_REQUIRED and not request.user.is_authenticated():
|
|
# Redirect unauthenticated requests to the login page. API requests are exempt from redirection as the API
|
|
# performs its own authentication.
|
|
api_path = reverse('api-root')
|
|
if not request.path_info.startswith(api_path) and request.path_info != settings.LOGIN_URL:
|
|
return HttpResponseRedirect('{}?next={}'.format(settings.LOGIN_URL, request.path_info))
|
|
return self.get_response(request)
|
|
|
|
|
|
class APIVersionMiddleware(object):
|
|
"""
|
|
If the request is for an API endpoint, include the API version as a response header.
|
|
"""
|
|
def __init__(self, get_response):
|
|
self.get_response = get_response
|
|
|
|
def __call__(self, request):
|
|
api_path = reverse('api-root')
|
|
response = self.get_response(request)
|
|
if request.path_info.startswith(api_path):
|
|
response['API-Version'] = settings.REST_FRAMEWORK_VERSION
|
|
return response
|
|
|
|
|
|
class ExceptionHandlingMiddleware(object):
|
|
"""
|
|
Intercept certain exceptions which are likely indicative of installation issues and provide helpful instructions
|
|
to the user.
|
|
"""
|
|
def __init__(self, get_response):
|
|
self.get_response = get_response
|
|
|
|
def __call__(self, request):
|
|
return self.get_response(request)
|
|
|
|
def process_exception(self, request, exception):
|
|
|
|
# Don't catch exceptions when in debug mode
|
|
if settings.DEBUG:
|
|
return
|
|
|
|
# Determine the type of exception
|
|
if isinstance(exception, ProgrammingError):
|
|
template_name = 'exceptions/programming_error.html'
|
|
elif isinstance(exception, ImportError):
|
|
template_name = 'exceptions/import_error.html'
|
|
elif (
|
|
sys.version_info[0] >= 3 and isinstance(exception, PermissionError)
|
|
) or (
|
|
isinstance(exception, OSError) and exception.errno == 13
|
|
):
|
|
template_name = 'exceptions/permission_error.html'
|
|
else:
|
|
template_name = '500.html'
|
|
|
|
# Return an error message
|
|
type_, error, traceback = sys.exc_info()
|
|
return render(request, template_name, {
|
|
'exception': str(type_),
|
|
'error': error,
|
|
}, status=500)
|