aljaz/bitfinex-api-py
1
0
Fork 0
mirror of https://github.com/aljazceru/bitfinex-api-py.git synced 2025-12-20 07:14:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Improve JSONEncoder class in bfxapi.utils.json_encoder.

Browse Source
This commit is contained in:
Davide Casale
2023-05-26 18:02:41 +02:00
parent dbc61ab275
commit bc0f83d408
3 changed files with 58 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
bfxapi/utils/json_encoder.py
View File

@@ -9,18 +9,20 @@ JSON = Union[Dict[str, "JSON"], List["JSON"], bool, int, float, str, Type[None]]
def _strip(dictionary: Dict) -> Dict:
return { key: value for key, value in dictionary.items() if value is not None }
def _convert_float_to_str(data: JSON) -> JSON:
def _convert_data_to_json(data: JSON) -> JSON:
if isinstance(data, bool):
return int(data)
if isinstance(data, float):
return format(Decimal(repr(data)), "f")
if isinstance(data, list):
return [ _convert_float_to_str(sub_data) for sub_data in data ]
return [ _convert_data_to_json(sub_data) for sub_data in data ]
if isinstance(data, dict):
return _strip({ key: _convert_float_to_str(value) for key, value in data.items() })
return _strip({ key: _convert_data_to_json(value) for key, value in data.items() })
return data
class JSONEncoder(json.JSONEncoder):
def encode(self, o: JSON) -> str:
return json.JSONEncoder.encode(self, _convert_float_to_str(o))
return json.JSONEncoder.encode(self, _convert_data_to_json(o))
def default(self, o: Any) -> Any:
if isinstance(o, Decimal):