Download/restore of dashboards after some basic testing.

Author: Adityashankar Kini

examples/download_dashboards.py

@@ -0,0 +1,89 @@
+#!/usr/bin/env python
+#
+# Save/restore dashboards
+#
+
+import os
+import sys
+import zipfile
+import json
+sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(sys.argv[0])), '..'))
+from sdcclient import SdcClient
+
+def zipdir(path, ziph):
+    # ziph is zipfile handle
+    for root, dirs, files in os.walk(path):
+        for file in files:
+            ziph.write(os.path.join(root, file))
+
+def cleanup_dir(path):
+    if os.path.exists(path) == False:
+        return
+    if os.path.isdir(path) == False:
+        print 'Provided path is not a directory'
+        sys.exit(-1)
+
+    for file in os.listdir(path):
+        file_path = os.path.join(path, file)
+        try:
+            if os.path.isfile(file_path):
+                os.unlink(file_path)
+            else:
+                print 'Cannot clean the provided directory due to delete failure on %s' % file_path
+        except Exception as e:
+            print(e)
+    os.rmdir(path)
+
+#
+# Parse arguments
+#
+if len(sys.argv) != 3:
+    print 'usage: %s <sysdig-token> <file-name>' % sys.argv[0]
+    print 'You can find your token at https://app.sysdigcloud.com/#/settings/user'
+    sys.exit(1)
+
+sdc_token = sys.argv[1]
+dashboard_state_file = sys.argv[2]
+sysdig_dashboard_dir = 'sysdig-dashboard-dir'
+
+#
+# Instantiate the SDC client
+#
+sdclient = SdcClient(sdc_token, sdc_url='https://app-staging.sysdigcloud.com')
+
+#
+# Fire the request.
+#
+res = sdclient.get_dashboards()
+
+#
+# Show the list of dashboards
+#
+if res[0]:
+    data = res[1]
+else:
+    print res[1]
+    sys.exit(1)
+
+
+# Clean up any state in the tmp directory
+cleanup_dir(sysdig_dashboard_dir)
+
+# Creating sysdig dashboard directory to store dashboards
+if not os.path.exists(sysdig_dashboard_dir):
+    os.makedirs(sysdig_dashboard_dir)
+
+
+for db in data['dashboards']:
+    file_path = os.path.join(sysdig_dashboard_dir, str(db['id']))
+    f = open(file_path, 'w')
+    f.write(json.dumps(db))
+    print "Name: %s, # Charts: %d" % (db['name'], len(db['items']))
+
+
+zipf = zipfile.ZipFile(dashboard_state_file, 'w', zipfile.ZIP_DEFLATED)
+zipdir(sysdig_dashboard_dir, zipf)
+zipf.close()
+
+# Clean up any state in the directory
+cleanup_dir(sysdig_dashboard_dir)

examples/restore_dashboards.py

@@ -0,0 +1,50 @@
+#!/usr/bin/env python
+#
+# Save/restore dashboards
+#
+
+import os
+import sys
+import zipfile
+import json
+sys.path.insert(0, os.path.join(os.path.dirname(os.path.realpath(sys.argv[0])), '..'))
+from sdcclient import SdcClient
+
+#
+# Parse arguments
+#
+if len(sys.argv) != 3:
+    print 'usage: %s <sysdig-token> <file-name>' % sys.argv[0]
+    print 'You can find your token at https://app.sysdigcloud.com/#/settings/user'
+    sys.exit(1)
+
+sdc_token = sys.argv[1]
+dashboard_state_file = sys.argv[2]
+
+#
+# Instantiate the SDC client
+#
+sdclient = SdcClient(sdc_token)
+
+zipf = zipfile.ZipFile(dashboard_state_file, 'r')
+
+
+dashboard_conf_items = ['showAsType', 'filterRoot', 'linkMetrics',
+                        'singleTimeNavigation', 'gridConfiguration', 'responsive',
+                        'nodesNoiseFilter', 'compareWith', 'format', 'linksNoiseFilter',
+                        'filterProcesses', 'isLegendExpanded', 'inhertitTimeNavigation',
+                        'schema', 'sortAscending', 'mapDataLimit', 'metrics', 'filterExtNodes',
+                        'sorting', 'name', 'sourceExploreView', 'items', 'showAs', 'eventsFilter',
+                        'timeMode', 'isShared', 'sourceDrilldownView']
+
+for info in zipf.infolist():
+    data = zipf.read(info.filename)
+    j = json.loads(data)
+    k = {}
+    for item in j.keys():
+        if item in dashboard_conf_items:
+            k[item] = j[item]
+     
+    res = sdclient.create_dashboard_with_configuration(k)
+    if res[0] == False:
+        print "Dashboard creation failed for dashboard name %s with error %s" % (j['name'], res[1])

sdcclient/_client.py

@@ -406,6 +406,13 @@ def create_item(configuration):
             dashboards = map(create_item, filter(filter_fn, res[1]['dashboards']))
             return [True, dashboards]
 
+    def create_dashboard_with_configuration(self, configuration):
+        res = requests.post(self.url + '/ui/dashboards', headers=self.hdrs, data=json.dumps({'dashboard': configuration}))
+        if not self.__checkResponse(res):
+            return [False, self.lasterr]
+        else:
+            return [True, res.json()]
+
     def create_dashboard(self, name):
         dashboard_configuration = {
             'name':   name,