Added write function

aviau · aviau · commit be476a4628d2 · 2015-02-17T00:00:38.000-05:00
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -173,6 +173,16 @@ def request(self, url, method='GET', params=None, data=None,
         else:
             raise InfluxDBClientError(response.content, response.status_code)
 
+    def write(self, data):
+        self.request(
+            url="write",
+            method='POST',
+            params=None,
+            data=data,
+            expected_response_code=200
+        )
+        return True
+
     # Writing Data
     #
     # Assuming you have a database named foo_production you can write data
diff --git a/tests/influxdb/client_test.py b/tests/influxdb/client_test.py
@@ -102,6 +102,35 @@ def test_switch_user(self):
         assert cli._username == 'another_username'
         assert cli._password == 'another_password'
 
+    def test_write(self):
+          with requests_mock.Mocker() as m:
+              m.register_uri(
+                  requests_mock.POST,
+                  "http://localhost:8086/write"
+              )
+
+              cli = InfluxDBClient(database='db')
+              cli.write(
+                  {"database": "mydb",
+                   "retentionPolicy": "mypolicy",
+                   "points": [{"name": "cpu_load_short",
+                               "tags": {"host": "server01",
+                                        "region": "us-west"},
+                              "timestamp": "2009-11-10T23:00:00Z",
+                              "values": {"value": 0.64}}]}
+              )
+
+              self.assertEqual(
+                  json.loads(m.last_request.body),
+                  {"database": "mydb",
+                   "retentionPolicy": "mypolicy",
+                   "points": [{"name": "cpu_load_short",
+                               "tags": {"host": "server01",
+                                        "region": "us-west"},
+                              "timestamp": "2009-11-10T23:00:00Z",
+                              "values": {"value": 0.64}}]}
+              )
+
     def test_write_points(self):
         with requests_mock.Mocker() as m:
             m.register_uri(
