GithubStar/RedisDesktopManager
1
0
Fork 0
mirror of https://github.com/redis/RedisDesktopManager.git synced 2026-10-17 04:31:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix std Python objects formatting

Browse Source
This commit is contained in:
Mrn Om
2021-03-26 17:01:15 +02:00
committed by Igor Malinovskiy
parent 9fa53f9303
commit e753e9e425
2 changed files with 21 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/py/formatters/pickle.py
+9 -3
View File
@@ -16,6 +16,11 @@ class PickleFormatter(BaseFormatter):
decode_format = "json"
def decode(self, value):
def get_json_output(deserialized_object):
return json.dumps(deserialized_object,
default=self.default,
ensure_ascii=False)
read_only = self.read_only
decode_format = self.decode_format
deserialized = ''
@@ -43,10 +48,11 @@ class PickleFormatter(BaseFormatter):
border=0)
output = self.format_html_output(deserialized, html)
decode_format = 'html'
else:
output = get_json_output(deserialized)
else:
output = json.dumps(deserialized,
default=self.default,
ensure_ascii=False)
output = get_json_output(deserialized)
return {
'output': output,
'decode-format': decode_format,
tests/py_tests/test_formatters/test_pickle_formatter.py
+12 -3
View File
@@ -30,10 +30,19 @@ class TestPickleFormatter(unittest.TestCase):
columns=list('ABCD')),
)
def test_decode(self, val):
pickled_val = pickle.dumps(val)
expected_output = json.dumps(val, default=self.formatter.default,
ensure_ascii=False)
if isinstance(val, pd.Series):
expected_output = f'{str(type(val))[1:-1]}\n{val.to_string()}'
elif isinstance(val, pd.DataFrame):
html = val.to_html(render_links=True, border=0)
expected_output = self.formatter.format_html_output(val, html)
elif isinstance(val, np.ndarray):
html = pd.DataFrame(val).to_html(render_links=True, border=0)
expected_output = self.formatter.format_html_output(val, html)
else:
expected_output = json.dumps(val, default=self.formatter.default,
ensure_ascii=False)
pickled_val = pickle.dumps(val)
formatter_response_dict = self.formatter.decode(pickled_val)
self.assertIn('output', formatter_response_dict)