utils/test/dashboard/kibana_cleanup.py

   1 #! /usr/bin/env python
   2 import logging
   3 import urlparse
   4
   5 import argparse
   6
   7 from dashboard.common import elastic_access
   8
   9 logger = logging.getLogger('clear_kibana')
  10 logger.setLevel(logging.DEBUG)
  11 file_handler = logging.FileHandler('/var/log/{}.log'.format('clear_kibana'))
  12 file_handler.setFormatter(logging.Formatter('%(asctime)s %(levelname)s: '
  13                                             '%(message)s'))
  14 logger.addHandler(file_handler)
  15
  16
  17 def delete_all(url, es_creds):
  18     ids = elastic_access.get_docs(url, es_creds, body=None, field='_id')
  19     for id in ids:
  20         del_url = '/'.join([url, id])
  21         elastic_access.delete_docs(del_url, es_creds)
  22
  23
  24 if __name__ == '__main__':
  25     parser = argparse.ArgumentParser(
  26         description=('Delete saved kibana searches, '
  27                      'visualizations and dashboards'))
  28     parser.add_argument('-e', '--elasticsearch-url',
  29                         default='http://localhost:9200',
  30                         help=('the url of elasticsearch, '
  31                               'defaults to http://localhost:9200'))
  32
  33     parser.add_argument('-u', '--elasticsearch-username', default=None,
  34                         help=('The username with password for elasticsearch '
  35                               'in format username:password'))
  36
  37     args = parser.parse_args()
  38     base_elastic_url = args.elasticsearch_url
  39     es_creds = args.elasticsearch_username
  40
  41     urls = (urlparse.urljoin(base_elastic_url, '/.kibana/visualization'),
  42             urlparse.urljoin(base_elastic_url, '/.kibana/dashboard'),
  43             urlparse.urljoin(base_elastic_url, '/.kibana/search'))
  44
  45     for url in urls:
  46         delete_all(url, es_creds)