mirror of
https://github.com/ZwareBear/awx.git
synced 2026-04-19 06:21:48 -05:00
1a3d452a2d
Trying to get the object could raise an unhandleable permission error if the user wasn't logged in for certain views
83 lines
3.1 KiB
Python
83 lines
3.1 KiB
Python
# Copyright (c) 2015 Ansible, Inc.
|
|
# All Rights Reserved.
|
|
|
|
# Django REST Framework
|
|
from rest_framework import renderers
|
|
from rest_framework.request import override_method
|
|
|
|
|
|
class BrowsableAPIRenderer(renderers.BrowsableAPIRenderer):
|
|
'''
|
|
Customizations to the default browsable API renderer.
|
|
'''
|
|
|
|
def get_default_renderer(self, view):
|
|
renderer = super(BrowsableAPIRenderer, self).get_default_renderer(view)
|
|
# Always use JSON renderer for browsable OPTIONS response.
|
|
if view.request.method == 'OPTIONS' and not isinstance(renderer, renderers.JSONRenderer):
|
|
return renderers.JSONRenderer()
|
|
return renderer
|
|
|
|
def get_context(self, data, accepted_media_type, renderer_context):
|
|
# Store the associated response status to know how to populate the raw
|
|
# data form.
|
|
try:
|
|
setattr(renderer_context['view'], '_raw_data_response_status', renderer_context['response'].status_code)
|
|
return super(BrowsableAPIRenderer, self).get_context(data, accepted_media_type, renderer_context)
|
|
finally:
|
|
delattr(renderer_context['view'], '_raw_data_response_status')
|
|
|
|
def get_raw_data_form(self, data, view, method, request):
|
|
# Set a flag on the view to indiciate to the view/serializer that we're
|
|
# creating a raw data form for the browsable API. Store the original
|
|
# request method to determine how to populate the raw data form.
|
|
if request.method in {'OPTIONS', 'DELETE'}:
|
|
return
|
|
try:
|
|
setattr(view, '_raw_data_form_marker', True)
|
|
setattr(view, '_raw_data_request_method', request.method)
|
|
return super(BrowsableAPIRenderer, self).get_raw_data_form(data, view, method, request)
|
|
finally:
|
|
delattr(view, '_raw_data_form_marker')
|
|
delattr(view, '_raw_data_request_method')
|
|
|
|
def get_rendered_html_form(self, data, view, method, request):
|
|
# Never show auto-generated form (only raw form).
|
|
obj = getattr(view, 'object', None)
|
|
if obj is None and hasattr(view, 'get_object') and hasattr(view, 'retrieve'):# and view.check_permissions():
|
|
try:
|
|
obj = view.get_object()
|
|
except Exception:
|
|
obj = None
|
|
with override_method(view, request, method) as request:
|
|
if not self.show_form_for_method(view, method, request, obj):
|
|
return
|
|
if method in ('DELETE', 'OPTIONS'):
|
|
return True # Don't actually need to return a form
|
|
|
|
def get_filter_form(self, data, view, request):
|
|
# Don't show filter form in browsable API.
|
|
return
|
|
|
|
|
|
class PlainTextRenderer(renderers.BaseRenderer):
|
|
|
|
media_type = 'text/plain'
|
|
format = 'txt'
|
|
|
|
def render(self, data, media_type=None, renderer_context=None):
|
|
if not isinstance(data, basestring):
|
|
data = unicode(data)
|
|
return data.encode(self.charset)
|
|
|
|
|
|
class DownloadTextRenderer(PlainTextRenderer):
|
|
|
|
format = "txt_download"
|
|
|
|
|
|
class AnsiTextRenderer(PlainTextRenderer):
|
|
|
|
media_type = 'text/plain'
|
|
format = 'ansi'
|