ResultSet: added as_dict()

aviau · aviau · commit 75b6e480a38c · 2015-04-08T16:28:14.000-04:00
diff --git a/influxdb/point.py b/influxdb/point.py
@@ -69,3 +69,9 @@ def __eq__(self, other):
 
     def __ne__(self, other):
         return not self.__eq__(other)
+
+    def as_dict(self):
+        return {
+            "serie": self.serie,
+            "point": [{col: getattr(self.values, col)} for col in self.columns]
+        }
diff --git a/tests/influxdb/point_test.py b/tests/influxdb/point_test.py
@@ -6,9 +6,8 @@
 
 
 class TestPoint(unittest.TestCase):
-
-    def test_point(self):
-        point = Point(
+    def setUp(self):
+        self.point = Point(
             "serie_name",
             ['col1', 'col2'],
             [1, '2'],
@@ -18,12 +17,13 @@ def test_point(self):
             }
         )
 
-        self.assertEqual(point.columns, ['col1', 'col2'])
-        self.assertEqual(point.tags, {"SWAG": True, "ALLO": "BYE"})
-        self.assertEqual(point.values.col1, 1)
-        self.assertEqual(point.values.col2, '2')
+    def test_point(self):
+        self.assertEqual(self.point.columns, ['col1', 'col2'])
+        self.assertEqual(self.point.tags, {"SWAG": True, "ALLO": "BYE"})
+        self.assertEqual(self.point.values.col1, 1)
+        self.assertEqual(self.point.values.col2, '2')
         self.assertEqual(
-            str(point),
+            str(self.point),
             "Point(values=(col1=1, col2='2'),"
             " tags={'ALLO': 'BYE', 'SWAG': True})"
         )
@@ -36,3 +36,9 @@ def test_point_eq(self):
                        tags={"SWAG": True, "ALLO": "BYE"})
 
         self.assertEqual(point1, point2)
+
+    def test_as_dict(self):
+        self.assertEqual(
+            self.point.as_dict(),
+            {'point': [{'col1': 1}, {'col2': '2'}], 'serie': 'serie_name'}
+        )
