mirror of
https://github.com/netbox-community/netbox.git
synced 2024-05-10 07:54:54 +00:00
Include the API version in responses
This commit is contained in:
@ -1,5 +1,6 @@
|
||||
from django.http import HttpResponseRedirect
|
||||
from django.conf import settings
|
||||
from django.core.urlresolvers import reverse
|
||||
|
||||
|
||||
BASE_PATH = getattr(settings, 'BASE_PATH', False)
|
||||
@ -17,7 +18,22 @@ class LoginRequiredMiddleware(object):
|
||||
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 = '/{}api/'.format(BASE_PATH)
|
||||
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
|
||||
|
Reference in New Issue
Block a user