added delete_series method w/ tests

jjmalina · jjmalina · commit 3205172b73e0 · 2014-02-12T10:49:50.000-05:00
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -243,6 +243,28 @@ def delete_database(self, database):
             raise Exception(
                 "{0}: {1}".format(response.status_code, response.content))
 
+    def delete_series(self, series):
+        """
+        Drop a series
+
+        Parameters
+        ----------
+        series: string
+            series name
+        """
+        response = session.delete("{0}/db/{1}/series/{2}?u={3}&p={4}".format(
+            self._baseurl,
+            self._database,
+            series,
+            self._username,
+            self._password))
+
+        if response.status_code == 204:
+            return True
+        else:
+            raise Exception(
+                "{0}: {1}".format(response.status_code, response.content))
+
     # Security
     # get list of cluster admins
     # curl http://localhost:8086/cluster_admins?u=root&p=root
diff --git a/tests/influxdb/client_test.py b/tests/influxdb/client_test.py
@@ -142,6 +142,19 @@ def test_delete_database_fails(self):
             cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
             cli.delete_database('old_db')
 
+    def test_delete_series(self):
+        with patch.object(session, 'delete') as mocked_delete:
+            mocked_delete.return_value = _build_response_object(status_code=204)
+            cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
+            cli.delete_series('old_series')
+
+    @raises(Exception)
+    def test_delete_series_fails(self):
+        with patch.object(session, 'delete') as mocked_delete:
+            mocked_delete.return_value = _build_response_object(status_code=401)
+            cli = InfluxDBClient('host', 8086, 'username', 'password', 'db')
+            cli.delete_series('old_series')
+
     def test_get_list_cluster_admins(self):
         pass
 
