IDE reformat
This commit is contained in:
parent
b0e599dc68
commit
762ef62098
GPG key ID:
AA742FDC5AFE2A72
70
dockstat.py
70
dockstat.py
|
|
@ -16,8 +16,8 @@ import numpy
|
|||
from prometheus_client import CollectorRegistry # type: ignore[import]
|
||||
from prometheus_client import Gauge, MetricsHandler, generate_latest
|
||||
|
||||
LISTEN_PORT = int(os.environ.get('DOCKSTAT_LISTEN_PORT', 8080))
|
||||
HEALTHY_STR = 'healthy'
|
||||
LISTEN_PORT = int(os.environ.get("DOCKSTAT_LISTEN_PORT", 8080))
|
||||
HEALTHY_STR = "healthy"
|
||||
|
||||
|
||||
class HTTPHandler(MetricsHandler): # type: ignore[misc]
|
||||
|
|
@ -36,18 +36,18 @@ class HTTPHandler(MetricsHandler): # type: ignore[misc]
|
|||
"""
|
||||
Handle GET requests
|
||||
"""
|
||||
if self.path == '/metrics':
|
||||
if self.path == "/metrics":
|
||||
try:
|
||||
self._metrics()
|
||||
except docker.errors.NotFound:
|
||||
pass
|
||||
|
||||
if self.path == '/healthcheck':
|
||||
if self.path == "/healthcheck":
|
||||
if not healthy():
|
||||
print('ERROR: Check requirements')
|
||||
self._respond(500, 'ERR')
|
||||
print("ERROR: Check requirements")
|
||||
self._respond(500, "ERR")
|
||||
|
||||
self._respond(200, 'OK')
|
||||
self._respond(200, "OK")
|
||||
|
||||
def _respond(self, status: int, message: str) -> None:
|
||||
"""
|
||||
|
|
@ -71,40 +71,40 @@ class HTTPHandler(MetricsHandler): # type: ignore[misc]
|
|||
Handle the request for metrics
|
||||
"""
|
||||
if not healthy:
|
||||
print('ERROR: Check requirements')
|
||||
self._respond(500, 'Server not configured correctly')
|
||||
print("ERROR: Check requirements")
|
||||
self._respond(500, "Server not configured correctly")
|
||||
return
|
||||
|
||||
registry = CollectorRegistry()
|
||||
|
||||
health_gauge = Gauge(
|
||||
'container_inspect_state_health_status',
|
||||
"container_inspect_state_health_status",
|
||||
"Container's healthcheck value (binary)",
|
||||
labelnames=['id', 'name', 'value'],
|
||||
labelnames=["id", "name", "value"],
|
||||
registry=registry,
|
||||
)
|
||||
status_gauge = Gauge(
|
||||
'container_inspect_state_running',
|
||||
"container_inspect_state_running",
|
||||
"Container's running state (binary)",
|
||||
labelnames=['id', 'name'],
|
||||
labelnames=["id", "name"],
|
||||
registry=registry,
|
||||
)
|
||||
started_at_gauge = Gauge(
|
||||
'container_inspect_state_started_at',
|
||||
"container_inspect_state_started_at",
|
||||
"Container's start time (int)",
|
||||
labelnames=['id', 'name'],
|
||||
labelnames=["id", "name"],
|
||||
registry=registry,
|
||||
)
|
||||
exit_code_gauge = Gauge(
|
||||
'container_inspect_state_exit_code',
|
||||
"container_inspect_state_exit_code",
|
||||
"Container's exit code (int)",
|
||||
labelnames=['id', 'name'],
|
||||
labelnames=["id", "name"],
|
||||
registry=registry,
|
||||
)
|
||||
alert_threshold_gauge = Gauge(
|
||||
'container_inspect_downtime_alert_threshold',
|
||||
"container_inspect_downtime_alert_threshold",
|
||||
"Container's downtime alert threshold in seconds (int)",
|
||||
labelnames=['id', 'name'],
|
||||
labelnames=["id", "name"],
|
||||
registry=registry,
|
||||
)
|
||||
|
||||
|
|
@ -112,16 +112,16 @@ class HTTPHandler(MetricsHandler): # type: ignore[misc]
|
|||
try:
|
||||
data = self.docker_api.inspect_container(container.id)
|
||||
except docker.errors.NotFound:
|
||||
print(f'WARNING: Container {container.id} does not exist. Skipping.')
|
||||
print(f"WARNING: Container {container.id} does not exist. Skipping.")
|
||||
continue
|
||||
|
||||
running = bool(data['State']['Running'])
|
||||
started_at = data['State']['StartedAt']
|
||||
exit_code = int(data['State']['ExitCode'])
|
||||
running = bool(data["State"]["Running"])
|
||||
started_at = data["State"]["StartedAt"]
|
||||
exit_code = int(data["State"]["ExitCode"])
|
||||
alert_threshold = int(
|
||||
data['Config']['Labels'].get('io.prometheus.alert.downtime', 3600)
|
||||
data["Config"]["Labels"].get("io.prometheus.alert.downtime", 3600)
|
||||
)
|
||||
starttime: numpy.longlong = numpy.datetime64(started_at, 's').astype('long')
|
||||
starttime: numpy.longlong = numpy.datetime64(started_at, "s").astype("long")
|
||||
|
||||
status_gauge.labels(
|
||||
container.id,
|
||||
|
|
@ -141,7 +141,7 @@ class HTTPHandler(MetricsHandler): # type: ignore[misc]
|
|||
).set(alert_threshold)
|
||||
|
||||
try:
|
||||
health_str = data['State']['Health']['Status']
|
||||
health_str = data["State"]["Health"]["Status"]
|
||||
health_gauge.labels(
|
||||
container.id,
|
||||
container.name,
|
||||
|
|
@ -162,12 +162,12 @@ def healthy() -> bool:
|
|||
"""
|
||||
return all(
|
||||
[
|
||||
os.path.exists('/var/run/docker.sock'),
|
||||
os.path.exists("/var/run/docker.sock"),
|
||||
]
|
||||
)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
if __name__ == "__main__":
|
||||
|
||||
def cli_parse() -> argparse.Namespace:
|
||||
"""
|
||||
|
|
@ -179,10 +179,10 @@ if __name__ == '__main__':
|
|||
parser = argparse.ArgumentParser()
|
||||
|
||||
parser.add_argument(
|
||||
'-H',
|
||||
'--healthcheck',
|
||||
action='store_true',
|
||||
help='Simply exit with 0 for healthy or 1 when unhealthy',
|
||||
"-H",
|
||||
"--healthcheck",
|
||||
action="store_true",
|
||||
help="Simply exit with 0 for healthy or 1 when unhealthy",
|
||||
)
|
||||
|
||||
return parser.parse_args()
|
||||
|
|
@ -197,11 +197,11 @@ if __name__ == '__main__':
|
|||
# Invert the sense of 'healthy' for Unix CLI usage
|
||||
return not healthy()
|
||||
|
||||
print(f'Starting web server on port {LISTEN_PORT}')
|
||||
print(f"Starting web server on port {LISTEN_PORT}")
|
||||
try:
|
||||
HTTPServer(('', LISTEN_PORT), HTTPHandler).serve_forever()
|
||||
HTTPServer(("", LISTEN_PORT), HTTPHandler).serve_forever()
|
||||
except KeyboardInterrupt:
|
||||
print('Exiting')
|
||||
print("Exiting")
|
||||
|
||||
return 0
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue