diff --git a/pillarsdk/api.py b/pillarsdk/api.py index 369b2d2..a8d6ddf 100644 --- a/pillarsdk/api.py +++ b/pillarsdk/api.py @@ -4,7 +4,6 @@ import requests import json import logging import platform -import sys from . import utils from . import exceptions @@ -96,7 +95,7 @@ class Api(object): else: http_headers['Content-Type'] = "application/json" return self.http_call(url, method, - data=json.dumps(body), + data=utils.dumps(body), headers=http_headers, verify=True) diff --git a/pillarsdk/utils.py b/pillarsdk/utils.py index 7c1fe5e..8f1e36c 100644 --- a/pillarsdk/utils.py +++ b/pillarsdk/utils.py @@ -39,6 +39,11 @@ class PillarJSONEncoder(json.JSONEncoder): return json.JSONEncoder.default(self, obj) +def dumps(mongo_doc, **kwargs): + """json.dumps() for PillarSDK documents.""" + return json.dumps(mongo_doc, cls=PillarJSONEncoder, **kwargs) + + def join_url(url, *paths): """ Joins individual URL strings together, and returns a single string.