Added type converter to JsonPayload.param

This commit is contained in:
Joel Collins 2019-01-09 16:18:49 +00:00
parent 8367bd7192
commit 783cc05ac9

View file

@ -2,7 +2,7 @@ import pprint
import copy import copy
class JsonResponse: class JsonPayload:
def __init__(self, request): def __init__(self, request):
""" """
Object to wrap up simple functionality for parsing a JSON response. Object to wrap up simple functionality for parsing a JSON response.
@ -12,16 +12,21 @@ class JsonResponse:
# Store raw response data # Store raw response data
self.data = request.get_data() self.data = request.get_data()
def param(self, key, default=None): def param(self, key, default=None, convert=None):
"""Check if a key exists in a JSON/dictionary payload, and returns it.""" """Check if a key exists in a JSON/dictionary payload, and returns it."""
# If no JSON payload exists, return early # If no JSON payload exists, return early
if not self.json: if not self.json:
return None return None
if key in self.json: if key in self.json:
return self.json[key] val = self.json[key]
else: else:
return default val = default
if convert:
val = convert(val)
return val
def axes_to_array(coordinate_dictionary, axis_keys=['x', 'y', 'z'], base_array=None): def axes_to_array(coordinate_dictionary, axis_keys=['x', 'y', 'z'], base_array=None):