New kibana config load/restore

Change-Id: I9c26e891c171d8f84bb9f4e03319f63599e00298
Issue-ID: CLAMP-226
Signed-off-by: ac2550 <ac2550@intl.att.com>

1 files changed, 89 insertions, 0 deletions

diff --git a/src/main/docker/kibana/backup.py b/src/main/docker/kibana/backup.py
new file mode 100755
index 000000000..4127eb23c
--- /dev/null
+++ b/src/main/docker/kibana/backup.py
@@ -0,0 +1,89 @@
+#!/usr/bin/env python
+###
+# ============LICENSE_START=======================================================
+# ONAP CLAMP
+# ================================================================================
+# Copyright (C) 2018 AT&T Intellectual Property. All rights
+#                             reserved.
+# ================================================================================
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============LICENSE_END============================================
+# ===================================================================
+#
+###
+
+import json
+import logging
+import os
+import sys
+
+import requests
+
+PER_PAGE = 1000
+
+def parse_args(args):
+    """ Parse arguments given to this script""" 
+    import argparse
+    parser = argparse.ArgumentParser(
+        description=('Description of the script'))
+    parser.add_argument('-v', '--verbose', dest='log_level', action='store_const',
+                        const=logging.DEBUG, default=logging.INFO,
+                        help='Use verbose logging')
+    parser.add_argument('-C', '--configuration_path',
+                        default='./default',
+                        help='Path of the configuration to be backed up.')
+    parser.add_argument('-f', '--force', action='store_const',
+                        const=True, default=False,
+                        help=('If the save folder already exists, overwrite files'
+                              ' matching a configuration item that should be written.'
+                              ' Files already in the folder that do not match are'
+                              ' left as-is.'))
+    parser.add_argument('-H', '--kibana-host', default='http://localhost:5601',
+                        help='Kibana endpoint.')
+
+    return parser.parse_args(args)
+
+def get_logger(args):
+    """Creates the logger based on the provided arguments"""
+    logging.basicConfig()
+    logger = logging.getLogger(__name__)
+    logger.setLevel(args.log_level)
+    return logger
+
+def main():
+    """ This script dumps the kibana configuration from Kibana"""
+    args = parse_args(sys.argv[1:])
+
+    base_config_path = args.configuration_path
+
+    # get list of all objects available
+    url = "%s/api/saved_objects/" % (args.kibana_host.rstrip("/"),)
+    saved_objects_req = requests.get(url,
+                                     params={'per_page': PER_PAGE})
+
+    saved_objects = saved_objects_req.json()['saved_objects']
+
+    for obj in saved_objects:
+
+        obj_folder = os.path.sep.join((base_config_path, obj['type']))
+
+        if not os.path.exists(obj_folder):
+            os.makedirs(obj_folder)
+
+        filename = "%s/%s-%s.json" % (obj_folder, obj['type'], obj['id'])
+        with open(filename, 'w') as file:
+            json.dump(obj, fp=file)
+
+
+if __name__ == "__main__":
+    main()