| from django.conf import settings |
| from django import http |
| |
| class XViewMiddleware(object): |
| """ |
| Adds an X-View header to internal HEAD requests -- used by the documentation system. |
| """ |
| def process_view(self, request, view_func, view_args, view_kwargs): |
| """ |
| If the request method is HEAD and either the IP is internal or the |
| user is a logged-in staff member, quickly return with an x-header |
| indicating the view function. This is used by the documentation module |
| to lookup the view function for an arbitrary page. |
| """ |
| if request.method == 'HEAD' and (request.META.get('REMOTE_ADDR') in settings.INTERNAL_IPS or (request.user.is_authenticated() and request.user.is_staff)): |
| response = http.HttpResponse() |
| response['X-View'] = "%s.%s" % (view_func.__module__, view_func.__name__) |
| return response |
