Added type converter to JsonPayload.param
This commit is contained in:
parent
8367bd7192
commit
783cc05ac9
1 changed files with 9 additions and 4 deletions
|
|
@ -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):
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue