Add get_list_measurements and drop_measurement (influxdata#402)

* 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 -> measurement * add docstring

Author: Vic020

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.

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":[{}]}'