Skip to content
This repository was archived by the owner on Jan 24, 2023. It is now read-only.

Commit 2dff151

Browse files
Александр Игоревич КорочкинAlKorochkin
Александр Игоревич Корочкин
authored andcommitted
change output format for mget to dict
1 parent e51df3e commit 2dff151

File tree

2 files changed

+10
-11
lines changed

2 files changed

+10
-11
lines changed

redistimeseries/client.py

+4-5
Original file line numberDiff line numberDiff line change
@@ -48,14 +48,13 @@ def parse_get(response):
4848
return (int(response[0]), float(response[1]))
4949

5050
def parse_m_get(response):
51-
res = []
51+
res = dict()
5252
for item in response:
5353
if item[2] == []:
54-
res.append({ nativestr(item[0]) : [list_to_dict(item[1]), None, None]})
54+
res[nativestr(item[0])] = (list_to_dict(item[1]), None, None)
5555
else:
56-
res.append({ nativestr(item[0]) : [list_to_dict(item[1]),
57-
int(item[2][0]), float(item[2][1])]})
58-
56+
res[nativestr(item[0])] = (list_to_dict(item[1]),
57+
int(item[2][0]), float(item[2][1]))
5958
return res
6059

6160
def parseToList(response):

test_commands.py

+6-6
Original file line numberDiff line numberDiff line change
@@ -154,20 +154,20 @@ def testMGet(self):
154154
rts.create(1, labels={'Test':'This'})
155155
rts.create(2, labels={'Test':'This', 'Taste':'That'})
156156
act_res = rts.mget(['Test=This'])
157-
exp_res = [{'1': [{}, None, None]}, {'2': [{}, None, None]}]
157+
exp_res = {'1': ({}, None, None), '2': ({}, None, None)}
158158
self.assertEqual(act_res, exp_res)
159159
rts.add(1, '*', 15)
160160
rts.add(2, '*', 25)
161161
res = rts.mget(['Test=This'])
162-
self.assertEqual(15, res[0]['1'][2])
163-
self.assertEqual(25, res[1]['2'][2])
162+
self.assertEqual(15, res['1'][2])
163+
self.assertEqual(25, res['2'][2])
164164
res = rts.mget(['Taste=That'])
165-
self.assertEqual(25, res[0]['2'][2])
165+
self.assertEqual(25, res['2'][2])
166166

167167
# test with_labels
168-
self.assertEqual({}, res[0]['2'][0])
168+
self.assertEqual({}, res['2'][0])
169169
res = rts.mget(['Taste=That'], with_labels=True)
170-
self.assertEqual({'Taste': 'That', 'Test': 'This'}, res[0]['2'][0])
170+
self.assertEqual({'Taste': 'That', 'Test': 'This'}, res['2'][0])
171171

172172
def testInfo(self):
173173
'''Test TS.INFO calls'''

0 commit comments

Comments
 (0)