Include embedded dashboard in ghost page

Signed-off-by: Matthew Stidham <stidmatt@gmail.com>

py-scripts/ghost_profile.py

@@ -33,7 +33,7 @@ if 'py-json' not in sys.path:
 from GhostRequest import GhostRequest
 
 
-class UseGhost():
+class UseGhost:
     def __init__(self,
                  _ghost_token=None,
                  host="localhost",
@@ -79,7 +79,12 @@ class UseGhost():
                       password_push,
                       customer,
                       testbed,
-                      test_run):
+                      test_run,
+                      grafana_dashboard,
+                      grafana_token,
+                      grafana_host,
+                      grafana_port):
+        target_folders = list()
         return self.GP.wifi_capacity_to_ghost(authors,
                                               folders,
                                               title,
@@ -92,7 +97,12 @@ class UseGhost():
                                               password_push,
                                               customer,
                                               testbed,
-                                              test_run)
+                                              test_run,
+                                              target_folders,
+                                              grafana_dashboard,
+                                              grafana_token,
+                                              grafana_host,
+                                              grafana_port)
 
 
 def main():
@@ -132,6 +142,10 @@ def main():
     optional.add_argument('--customer')
     optional.add_argument('--testbed')
     optional.add_argument('--test_run', default=None)
+    optional.add_argument('--grafana_dashboard')
+    optional.add_argument('--grafana_token', default=None)
+    optional.add_argument('--grafana_host', default=None)
+    optional.add_argument('--grafana_port', default=3000)
     optional.add_argument('--debug')
     args = parser.parse_args()
 
@@ -170,7 +184,11 @@ def main():
                             args.password_push,
                             args.customer,
                             args.testbed,
-                            args.test_run)
+                            args.test_run,
+                            args.grafana_dashboard,
+                            args.grafana_token,
+                            args.grafana_host,
+                            args.grafana_port)
 
 
 if __name__ == "__main__":

py-scripts/grafana_profile.py

@@ -24,36 +24,37 @@ import string
 import random
 
 
-class UseGrafana(LFCliBase):
-    def __init__(self,
-                 _grafana_token,
-                 host="localhost",
-                 _grafana_host="localhost",
-                 port=8080,
-                 _debug_on=False,
-                 _exit_on_fail=False,
-                 _grafana_port=3000):
-        super().__init__(host, port, _debug=_debug_on, _exit_on_fail=_exit_on_fail)
-        self.grafana_token = _grafana_token
-        self.grafana_port = _grafana_port
-        self.grafana_host = _grafana_host
-        self.GR = GrafanaRequest(self.grafana_host, str(self.grafana_port), _folderID=0, _api_token=self.grafana_token)
+#!/usr/bin/env python3
 
-    def create_dashboard(self,
-                         dashboard_name):
-        return self.GR.create_dashboard(dashboard_name)
+# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+# Class holds default settings for json requests to Grafana     -
+# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+import sys
 
-    def delete_dashboard(self,
-                         dashboard_uid):
-        return self.GR.delete_dashboard(dashboard_uid)
+if sys.version_info[0] != 3:
+    print("This script requires Python 3")
+    exit()
 
-    def list_dashboards(self):
-        return self.GR.list_dashboards()
+import requests
 
-    def create_dashboard_from_data(self,
-                                   json_file):
-        return self.GR.create_dashboard_from_data(json_file=json_file)
+import json
 
+#!/usr/bin/env python3
+
+# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+# Class holds default settings for json requests to Grafana     -
+# - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+import sys
+
+if sys.version_info[0] != 3:
+    print("This script requires Python 3")
+    exit()
+
+import requests
+
+import json
+
+class UseGrafana(GrafanaRequest):
     def groupby(self, params, grouptype):
         dic = dict()
         dic['params'] = list()
@@ -301,7 +302,6 @@ class UseGrafana(LFCliBase):
         return dict(zip(graph_group, units))
 
 
-
 def main():
     parser = LFCliBase.create_basic_argparse(
         prog='grafana_profile.py',
@@ -353,11 +353,13 @@ def main():
     optional.add_argument('--from_date', help='Date you want to start your Grafana dashboard from', default='now-1y')
     optional.add_argument('--graph_height', help='Custom height for the graph on grafana dashboard', default=8)
     optional.add_argument('--graph_width', help='Custom width for the graph on grafana dashboard', default=12)
+    optional.add_argument('--create_snapshot', action='store_true')
+    optional.add_argument('--list_snapshots', action='store_true')
     args = parser.parse_args()
 
     Grafana = UseGrafana(args.grafana_token,
-                         args.grafana_port,
-                         args.grafana_host
+                         args.grafana_host,
+                         grafanajson_port=args.grafana_port
                          )
     if args.dashboard_name is not None:
         Grafana.create_dashboard(args.dashboard_name)
@@ -386,6 +388,13 @@ def main():
                                         graph_height=args.graph_height,
                                         graph__width=args.graph_width)
 
+    if args.create_snapshot:
+        Grafana.create_snapshot(args.title)
+
+    if args.list_snapshots:
+        Grafana.list_snapshots()
+
+
 
 if __name__ == "__main__":
     main()