Added support for period index

timtroendle · timtroendle · commit bc25d2f42c09 · 2014-11-16T20:21:49.000+01:00
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -742,7 +742,7 @@ def send_packet(self, packet):
         self.udp_socket.sendto(byte, (self._host, self.udp_port))
 
     def write_points_from_dataframe(self, dataframe, name):
-        dataframe = dataframe.copy()
+        dataframe.index = dataframe.index.to_datetime()
         dataframe['time'] = [time.mktime(dt.timetuple()) for dt in dataframe.index]
         data = dict()
         data['name'] = name
diff --git a/tests/influxdb/client_test.py b/tests/influxdb/client_test.py
@@ -629,6 +629,39 @@ def test_write_points_from_dataframe(self):
                 self.dummy_points
             )
 
+    def test_write_points_from_dataframe_with_period_index(self):
+        now = datetime(2014, 11, 16)
+        self.dummy_points = [
+            {
+                "points": [
+                    ["1", 1, 1.0, time.mktime(now.timetuple())],
+                    ["2", 2, 2.0, time.mktime((now + timedelta(hours=24)).timetuple())]
+                ],
+                "name": "foo",
+                "columns": ["column_one", "column_two", "column_three", "time"]
+            }
+        ]
+        self.dummy_dataframe = pd.DataFrame(data=[["1", 1, 1.0], ["2", 2, 2.0]],
+                                            index = [pd.Period('2014-11-16'), pd.Period('2014-11-17')],
+                                            columns=["column_one", "column_two", "column_three"])
+
+        with requests_mock.Mocker() as m:
+            m.register_uri(
+                requests_mock.POST,
+                "http://localhost:8086/db/db/series"
+            )
+
+            cli = InfluxDBClient(database='db')
+            cli.write_points_from_dataframe(
+                self.dummy_dataframe, name="foo"
+            )
+
+            self.assertListEqual(
+                json.loads(m.last_request.body),
+                self.dummy_points
+            )
+
+
     def test_query_into_dataframe(self):
         data = [
             {
