95 lines
2.4 KiB
Python
95 lines
2.4 KiB
Python
from __future__ import print_function
|
|
|
|
import logging
|
|
|
|
from fuel_ccp import config
|
|
from fuel_ccp import kubernetes
|
|
|
|
CONF = config.CONF
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
class State(object):
|
|
def __init__(self, name, total, running, urls):
|
|
self.name = name
|
|
self.total = total or 0
|
|
self.running = running or 0
|
|
self.urls = urls or []
|
|
|
|
def __repr__(self):
|
|
if self.ready:
|
|
return "ok"
|
|
else:
|
|
return "nok" # not ready
|
|
|
|
def __lt__(self, other):
|
|
return self.name.__lt__(other.name)
|
|
|
|
def __bool__(self):
|
|
return self.ready
|
|
|
|
def __nonzero__(self):
|
|
return self.ready
|
|
|
|
@property
|
|
def ready(self):
|
|
return self.total == self.running
|
|
|
|
|
|
def _get_pods_status(service, svc_map):
|
|
pods = kubernetes.list_cluster_pods(service=service)
|
|
total = running = 0
|
|
for pod in pods:
|
|
total += 1
|
|
if pod.ready:
|
|
running += 1
|
|
return State(
|
|
name=service,
|
|
total=total,
|
|
running=running,
|
|
urls=svc_map.get(service, []))
|
|
|
|
|
|
def get_pod_states(components=None):
|
|
ext_ip = CONF.configs.get("k8s_external_ip", "")
|
|
ext_link_template = "http://{ext_ip}:{port}"
|
|
states = []
|
|
svc_map = {}
|
|
for svc in kubernetes.list_cluster_services():
|
|
svc_name = svc.obj["metadata"]["name"]
|
|
svc_map.setdefault(svc_name, [])
|
|
for port in svc.obj["spec"]["ports"]:
|
|
svc_map[svc_name].append(ext_link_template.format(
|
|
ext_ip=ext_ip,
|
|
port=port["nodePort"]))
|
|
for dp in kubernetes.list_cluster_deployments():
|
|
if not components or dp.name in components:
|
|
states.append(_get_pods_status(dp.name, svc_map))
|
|
for ds in kubernetes.list_cluster_daemonsets():
|
|
if not components or ds.name in components:
|
|
states.append(_get_pods_status(ds.name, svc_map))
|
|
|
|
return states
|
|
|
|
|
|
def show_long_status(components=None):
|
|
states = get_pod_states(components)
|
|
columns = ("service", "pod", "ready", "links")
|
|
|
|
formatted_states = []
|
|
|
|
for state in sorted(states):
|
|
formatted_states.append((
|
|
state.name,
|
|
"/".join((str(state.running), str(state.total))),
|
|
state,
|
|
"\n".join(state.urls)))
|
|
|
|
return columns, formatted_states
|
|
|
|
|
|
def show_short_status():
|
|
status = "ok" if all(get_pod_states()) else "nok"
|
|
return ("status",), ((status,),)
|