diff --git a/GrafanaSnapshot/feature/base.py b/GrafanaSnapshot/feature/base.py index c59f49f..1a38e35 100644 --- a/GrafanaSnapshot/feature/base.py +++ b/GrafanaSnapshot/feature/base.py @@ -1,5 +1,6 @@ class Base(object): - def __init__(self, api, host, protocol): + def __init__(self, api, host, protocol, port=3000): self.api = api self.host = host self.protocol = protocol + self.port = port diff --git a/GrafanaSnapshot/feature/snapshots.py b/GrafanaSnapshot/feature/snapshots.py index 6abba88..f1525e7 100644 --- a/GrafanaSnapshot/feature/snapshots.py +++ b/GrafanaSnapshot/feature/snapshots.py @@ -51,7 +51,7 @@ def get_snapshot_by_key(self, key): snapshot_list = [] for snapshot in snapshot: if key in snapshot["name"]: - url=snapshot["name"]+" : "+self.protocol+"://"+self.host+":3000/dashboard/snapshot/"+snapshot["key"] + url=snapshot["name"]+" : "+self.protocol+"://"+self.host+":"+str(self.port)+"/dashboard/snapshot/"+snapshot["key"] snapshot_list.append(url) return snapshot_list diff --git a/GrafanaSnapshot/snapshot_face.py b/GrafanaSnapshot/snapshot_face.py index 5370923..69ed978 100644 --- a/GrafanaSnapshot/snapshot_face.py +++ b/GrafanaSnapshot/snapshot_face.py @@ -32,4 +32,4 @@ def __init__( protocol=protocol, verify=verify, ) - self.snapshots = Snapshots(self.api, host, protocol) + self.snapshots = Snapshots(self.api, host, protocol, port)