scott/dockstat
1
0
Fork 0
Code Issues Pull requests Packages Releases Activity

Add missing content-type header

Browse source
This commit is contained in:
Scott Wallace 2024-11-15 09:18:51 +00:00
parent 1999536b50
commit 23ccb252da
Signed by: scott
SSH key fingerprint: SHA256:+LJug6Dj01Jdg86CILGng9r0lJseUrpI0xfRqdW9Uws
1 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
dockstat.py
View file

@ -11,18 +11,17 @@ import sys
from http.server import HTTPServer
from typing import Any
import docker # type: ignore[import]
import docker.errors # type: ignore[import]
import docker
import docker.errors
import numpy
from docker.models.containers import Container # type: ignore[import]
from prometheus_client import CollectorRegistry # type: ignore[import]
from prometheus_client import Gauge, MetricsHandler, generate_latest
from docker.models.containers import Container
from prometheus_client import CollectorRegistry, Gauge, MetricsHandler, generate_latest
LISTEN_PORT = int(os.environ.get("DOCKSTAT_LISTEN_PORT", 8080))
HEALTHY_STR = "healthy"
class HTTPHandler(MetricsHandler): # type: ignore[misc]
class HTTPHandler(MetricsHandler):
"""
Class to encompass the requirements of a Prometheus Exporter
for Docker containers with a healthcheck configured
@ -62,6 +61,7 @@ class HTTPHandler(MetricsHandler): # type: ignore[misc]
message (str): String to output
"""
self.send_response(int(status) or 500)
self.send_header("content-type", "text/plain")
self.end_headers()
try:
self.wfile.write(bytes(str(message).encode()))
@ -111,9 +111,9 @@ class HTTPHandler(MetricsHandler): # type: ignore[misc]
)
container: Container
for container in self.docker_client.containers.list(all=True): # type: ignore
for container in self.docker_client.containers.list(all=True):
try:
data = self.docker_api.inspect_container(container.id)
data = self.docker_api.inspect_container(getattr(container, "id"))
except docker.errors.NotFound:
print(f"WARNING: Container {container.id} does not exist. Skipping.")
continue