Add get_list_measurements and drop_measurement (influxdata#402)

Vic020 · sebito91 · commit df6e2fe2f9f5 · 2017-09-07T20:58:17.000-05:00
* Add get_list_measurements and drop_measurement

Dear maintainer,

Look likes we are missing the get list measurements and drop measurement feature for client.

pls accept this pr.

* add test case

* fix: dbname -&gt; measurement

* add docstring
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -547,6 +547,32 @@ def drop_database(self, dbname):
         """
         self.query("DROP DATABASE {0}".format(quote_ident(dbname)))
 
+    def get_list_measurements(self):
+        """Get the list of measurements in InfluxDB.
+
+        :returns: all measurements in InfluxDB
+        :rtype: list of dictionaries
+
+        :Example:
+
+        ::
+
+            >> dbs = client.get_list_measurements()
+            >> dbs
+            [{u'name': u'measurements1'},
+             {u'name': u'measurements2'},
+             {u'name': u'measurements3'}]
+        """
+        return list(self.query("SHOW MEASUREMENTS").get_points())
+
+    def drop_measurement(self, measurement):
+        """Drop a measurement from InfluxDB.
+
+        :param measurement: the name of the measurement to drop
+        :type measurement: str
+        """
+        self.query("DROP MEASUREMENT {0}".format(quote_ident(measurement)))
+
     def create_retention_policy(self, name, duration, replication,
                                 database=None, default=False):
         """Create a retention policy for a database.
diff --git a/influxdb/tests/client_test.py b/influxdb/tests/client_test.py
@@ -466,6 +466,20 @@ def test_drop_database(self):
                 'drop database "new_db"'
             )
 
+    def test_drop_measurement(self):
+        """Test drop measurement for TestInfluxDBClient object."""
+        with requests_mock.Mocker() as m:
+            m.register_uri(
+                requests_mock.GET,
+                "http://localhost:8086/query",
+                text='{"results":[{}]}'
+            )
+            self.cli.drop_measurement('new_measurement')
+            self.assertEqual(
+                m.last_request.qs['q'][0],
+                'drop measurement "new_measurement"'
+            )
+
     def test_drop_numeric_named_database(self):
         """Test drop numeric db for TestInfluxDBClient object."""
         with requests_mock.Mocker() as m:
@@ -504,6 +518,24 @@ def test_get_list_database_fails(self):
         with _mocked_session(cli, 'get', 401):
             cli.get_list_database()
 
+    def test_get_list_measurements(self):
+        """Test get list of measurements for TestInfluxDBClient object."""
+        data = {
+            "results": [{
+                "series": [
+                    {"name": "measurements",
+                     "columns": ["name"],
+                     "values": [["cpu"], ["disk"]
+                                ]}]}
+            ]
+        }
+
+        with _mocked_session(self.cli, 'get', 200, json.dumps(data)):
+            self.assertListEqual(
+                self.cli.get_list_measurements(),
+                [{'name': 'cpu'}, {'name': 'disk'}]
+            )
+
     def test_create_retention_policy_default(self):
         """Test create default ret policy for TestInfluxDBClient object."""
         example_response = '{"results":[{}]}'
