summaryrefslogtreecommitdiff
path: root/libs/service_status_client.py
blob: bac0ed788f585af754693dcfe2fe631e211c9d72 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
import requests
from requests.packages.urllib3.exceptions import InsecureRequestWarning
import json


class ServiceStatusClient(object):
    def __init__(self, host=None):
        requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
        self.session = requests.Session()
        self.host = host
        self.base_url = "%s/api/v2.0" % self.host

    def store_result(self, resp):
        self.status_code = resp.status_code
        self.body = resp.text
        try:
            self.json = resp.json
        except Exception:
            self.json = None

    def get(self, url=None, params=None, headers={}, timeout=60):
        resp = self.session.get(url=url,
                                params=params,
                                headers=headers,
                                timeout=timeout,
                                verify=False)
        self.store_result(resp)
        return resp

    def post(self, url=None, data=None, headers={}, timeout=60):
        resp = self.session.post(url=url,
                                 data=data,
                                 headers=headers,
                                 timeout=timeout,
                                 verify=False)
        self.store_result(resp)
        return resp

    def get_service_managers(self):
        regions = []
        self.get(url="%s/" % self.base_url)
        for project in [p for p in self.json()["projects"]]:
            self.get(url="%s/%s/project-status" % (self.base_url, project["project"]),
                     params={"depth": "2"})
            for site in [r for r in self.json()["sites"]]:
                for region in site["regions"]:
                    if region["attributes"]["live-date"] != "TBD":
                        regions.append({"name": "%s-%s" % (region["region"].replace("r", "s"), region["site"]),
                                        "connectionUrl": "%s:13000/v3" % region["attributes"]["dashboard-url"].replace("/dashboard", "")})
        return regions