Merge pull request #7 from jacobsee/add-status-generation

Add scripts for status generation

Author: pabrahamsson

generate-status/collect_info.py

@@ -0,0 +1,35 @@
+from kubernetes import client, config
+import gitlab
+
+g = gitlab.Gitlab(context["gitlab_url"] , private_token=context["gitlab_private_key"])
+gitlab_group = g.groups.get(context["gitlab_group"])
+
+config.load_incluster_config()
+custom_object_api = client.CustomObjectsApi()
+anarchy_subject_list = custom_object_api.list_namespaced_custom_object(
+  group="anarchy.gpte.redhat.com",
+  version="v1",
+  namespace=context["babylon_namespace"],
+  plural="anarchysubjects",
+)
+
+def getAnarchySubject(project_id):
+  subject = list(filter(lambda x: project_id in x["metadata"]["annotations"]["poolboy.gpte.redhat.com/resource-claim-name"], anarchy_subject_list["items"]))
+  if len(subject) > 0:
+    return subject[0]
+  else:
+    return None
+
+context["repositories"] = [
+  {
+    "url": project.ssh_url_to_repo,
+    "directory": str(project.id),
+    "anarchy_subject": getAnarchySubject(str(project.id))
+  }
+for project in gitlab_group.projects.list(all=True, include_subgroups=True)]
+
+context["repositories"] = list(filter(lambda x: x["anarchy_subject"] is not None, context["repositories"]))
+
+need_to_process = [element["directory"] for element in context["repositories"]]
+
+print(f"Will update status for repositories: {need_to_process}")

generate-status/write_status.py

@@ -0,0 +1,52 @@
+import json
+import datetime
+
+subject = context["repositories"]
+
+status = {
+  "overall_status": "",
+  "messages": [],
+  "subsystems": []
+}
+ocp_subsystem = {
+  "name": "openshift",
+  "status": "",
+  "state": "",
+  "info": "",
+  "updated": "",
+  "access_urls": [],
+  "messages": []
+}
+
+with open(f"../../{subject['directory']}/engagement.json", "r") as read_file:
+  engagement = json.load(read_file)
+
+current_state = subject["anarchy_subject"]["spec"]["vars"]["current_state"]
+desired_state = subject["anarchy_subject"]["spec"]["vars"]["desired_state"]
+
+ocp_subsystem["state"] = current_state
+ocp_subsystem["updated"] = str(datetime.datetime.utcnow().replace(tzinfo=datetime.timezone.utc, microsecond=0).isoformat())
+ocp_subsystem["access_urls"] = [
+  {
+    "Web Console": f"https://console-openshift-console.apps.{engagement['ocp_sub_domain']}.{context["ocp_base_url"]}"
+  },
+  {
+    "API": f"https://api.{engagement['ocp_sub_domain']}.{context["ocp_base_url"]}:6443"
+  }
+]
+
+if current_state == "provisioning":
+  ocp_subsystem["status"] = "yellow"
+  ocp_subsystem["info"] = "Building cluster"
+elif current_state == desired_state :
+  ocp_subsystem["status"] = "green"
+  ocp_subsystem["info"] = "Working as expected"
+else:
+  ocp_subsystem["status"] = "yellow"
+  ocp_subsystem["info"] = "Contact SRE team"
+
+status["overall_status"] = ocp_subsystem["status"]
+status["subsystems"].append(ocp_subsystem)
+
+with open(f"../../{subject['directory']}/status.json", 'w') as fp:
+  json.dump(status, fp)