def render_to_json(response):
"""
Determine the appropriate content and create the JSON response
"""
# determine the status code
if hasattr(response, 'status_code'):
status_code = response.status_code
elif issubclass(type(response), Http404):
status_code = 404
elif issubclass(type(response), Exception):
status_code = 500
else:
status_code = 200
data = {}
# TODO: add optional process for fragments
if isinstance(response, dict):
for key in ['fragments', 'inner-fragments', 'append-fragments',
'prepend-fragments']:
if key in response:
data.update({
key: response.pop(key)
})
data.update({
'status': status_code,
'statusText': REASON_PHRASES.get(status_code, 'UNKNOWN STATUS CODE'),
'content': response
})
return JSONResponse(data)
def process_response(self, request, response):
if random() >= 0.5 and DRF_CHAOS_ENABLED:
time.sleep(randint(0, 3))
response = HttpResponse()
status_code = choice(REASON_PHRASES.keys())
response.status_code = status_code
response.reason_phrase = REASON_PHRASES.get(
status_code, 'UNKNOWN STATUS CODE')
response.content = "drf-chaos: {}".format(response.reason_phrase)
return response
def httpreason(value, arg=False):
"""
Uses django's REASON_PHRASES to change a status_code into a textual reason.
Optional True/False argument allows you to return a string with code number *and* phrase. Defaults to False"""
try:
value_int = int(value)
except Exception:
return ''
phrase = REASON_PHRASES.get(value_int, 'UNKNOWN STATUS CODE')
if arg:
phrase = '{0}: {1}'.format(value, phrase)
return phrase
def httpreason(value, arg=False):
"""
Uses django's REASON_PHRASES to change a status_code into a textual reason.
Optional True/False argument allows you to return a string with code number *and* phrase. Defaults to False"""
try:
value_int = int(value)
except Exception:
return ''
phrase = REASON_PHRASES.get(value_int, 'UNKNOWN STATUS CODE')
if arg:
phrase = '{0}: {1}'.format(value, phrase)
return phrase
def process_response(self, request, response):
if random() >= 0.5 and DRF_CHAOS_ENABLED:
time.sleep(randint(0, 3))
response = HttpResponse()
status_code = choice(REASON_PHRASES.keys())
response.status_code = status_code
response.reason_phrase = REASON_PHRASES.get(
status_code,
'UNKNOWN STATUS CODE'
)
response.content = "drf-chaos: {}".format(
response.reason_phrase
)
return response
def custom_rest_exception_handler(exc, context):
response = exception_handler(exc, context)
if isinstance(exc, exceptions.NotAuthenticated):
response.status_code = 401
response.reason_phrase = REASON_PHRASES.get(response.status_code)
return response
