226 lines
7.3 KiB
Python
226 lines
7.3 KiB
Python
"""Prometheus API client — fragt Host-Metriken ab für den Hausmeister-Bot."""
|
|
|
|
import requests
|
|
from datetime import datetime, timezone, timedelta
|
|
|
|
PROMETHEUS_URL = "http://10.10.10.1:9090"
|
|
|
|
WARN_CPU = 80
|
|
WARN_MEM = 85
|
|
WARN_DISK = 85
|
|
|
|
|
|
def _query(endpoint: str, params: dict) -> dict:
|
|
try:
|
|
r = requests.get(f"{PROMETHEUS_URL}{endpoint}", params=params, timeout=10)
|
|
r.raise_for_status()
|
|
return r.json()
|
|
except requests.RequestException as e:
|
|
return {"error": str(e), "status": "unavailable"}
|
|
|
|
|
|
def instant_query(query: str) -> dict:
|
|
return _query("/api/v1/query", {"query": query})
|
|
|
|
|
|
def range_query(query: str, hours: float = 1, step: str = "5m") -> dict:
|
|
now = datetime.now(timezone.utc)
|
|
start = now - timedelta(hours=hours)
|
|
return _query("/api/v1/query_range", {
|
|
"query": query,
|
|
"start": start.isoformat(),
|
|
"end": now.isoformat(),
|
|
"step": step,
|
|
})
|
|
|
|
|
|
def is_available() -> bool:
|
|
data = _query("/api/v1/query", {"query": "up"})
|
|
return data.get("status") == "success"
|
|
|
|
|
|
def get_targets() -> list[dict]:
|
|
data = _query("/api/v1/targets", {})
|
|
if "error" in data:
|
|
return [{"error": data["error"]}]
|
|
targets = []
|
|
for t in data.get("data", {}).get("activeTargets", []):
|
|
labels = t.get("labels", {})
|
|
targets.append({
|
|
"host": labels.get("host", labels.get("instance", "?")),
|
|
"location": labels.get("location", ""),
|
|
"health": t.get("health", "unknown"),
|
|
"job": labels.get("job", ""),
|
|
})
|
|
return targets
|
|
|
|
|
|
def _by_host(data: dict, metric: str) -> list[dict]:
|
|
if data.get("status") != "success":
|
|
return []
|
|
results = []
|
|
for r in data.get("data", {}).get("result", []):
|
|
m = r.get("metric", {})
|
|
host = m.get("host", m.get("instance", "?"))
|
|
val = float(r.get("value", [0, 0])[1])
|
|
results.append({"host": host, "value": val, "metric": metric})
|
|
return sorted(results, key=lambda x: x["host"])
|
|
|
|
|
|
def get_cpu(host: str = None) -> list[dict]:
|
|
filt = f', host="{host}"' if host else ""
|
|
q = f'100 - (avg by (host) (rate(node_cpu_seconds_total{{mode="idle"{filt}}}[5m])) * 100)'
|
|
return _by_host(instant_query(q), "cpu")
|
|
|
|
|
|
def get_memory(host: str = None) -> list[dict]:
|
|
filt = f', host="{host}"' if host else ""
|
|
q = f'(1 - node_memory_MemAvailable_bytes{{host!=""{filt}}} / node_memory_MemTotal_bytes{{host!=""{filt}}}) * 100'
|
|
return _by_host(instant_query(q), "mem")
|
|
|
|
|
|
def get_disk(host: str = None) -> list[dict]:
|
|
filt = f', host="{host}"' if host else ""
|
|
q = f'(1 - node_filesystem_avail_bytes{{mountpoint="/", host!=""{filt}}} / node_filesystem_size_bytes{{mountpoint="/", host!=""{filt}}}) * 100'
|
|
return _by_host(instant_query(q), "disk")
|
|
|
|
|
|
def get_disk_bytes(host: str = None) -> list[dict]:
|
|
"""Returns available and total bytes per host."""
|
|
filt = f', host="{host}"' if host else ""
|
|
avail = _by_host(instant_query(
|
|
f'node_filesystem_avail_bytes{{mountpoint="/", host!=""{filt}}}'), "avail_bytes")
|
|
total = _by_host(instant_query(
|
|
f'node_filesystem_size_bytes{{mountpoint="/", host!=""{filt}}}'), "total_bytes")
|
|
total_map = {r["host"]: r["value"] for r in total}
|
|
result = []
|
|
for a in avail:
|
|
t = total_map.get(a["host"], 0)
|
|
result.append({
|
|
"host": a["host"],
|
|
"avail_gb": a["value"] / (1024**3),
|
|
"total_gb": t / (1024**3),
|
|
})
|
|
return result
|
|
|
|
|
|
def get_uptime(host: str = None) -> list[dict]:
|
|
filt = f', host="{host}"' if host else ""
|
|
q = f'node_time_seconds{{host!=""{filt}}} - node_boot_time_seconds{{host!=""{filt}}}'
|
|
return _by_host(instant_query(q), "uptime_sec")
|
|
|
|
|
|
def get_load(host: str = None) -> list[dict]:
|
|
filt = f', host="{host}"' if host else ""
|
|
q = f'node_load5{{host!=""{filt}}}'
|
|
return _by_host(instant_query(q), "load5")
|
|
|
|
|
|
def get_warnings() -> list[str]:
|
|
"""Return list of warning strings for hosts exceeding thresholds."""
|
|
warnings = []
|
|
for r in get_cpu():
|
|
if r["value"] >= WARN_CPU:
|
|
warnings.append(f"🔴 {r['host']}: CPU {r['value']:.0f}%")
|
|
for r in get_memory():
|
|
if r["value"] >= WARN_MEM:
|
|
warnings.append(f"🔴 {r['host']}: RAM {r['value']:.0f}%")
|
|
for r in get_disk():
|
|
if r["value"] >= WARN_DISK:
|
|
warnings.append(f"🔴 {r['host']}: Disk {r['value']:.0f}%")
|
|
return warnings
|
|
|
|
|
|
def _fmt_uptime(seconds: float) -> str:
|
|
days = int(seconds // 86400)
|
|
hours = int((seconds % 86400) // 3600)
|
|
if days > 0:
|
|
return f"{days}d {hours}h"
|
|
return f"{hours}h"
|
|
|
|
|
|
def format_overview() -> str:
|
|
"""Kompakte Übersicht aller Hosts — für den Bot."""
|
|
if not is_available():
|
|
return "⚠️ Prometheus nicht erreichbar."
|
|
|
|
cpu = {r["host"]: r["value"] for r in get_cpu()}
|
|
mem = {r["host"]: r["value"] for r in get_memory()}
|
|
disk = {r["host"]: r["value"] for r in get_disk()}
|
|
disk_gb = {r["host"]: r for r in get_disk_bytes()}
|
|
uptime = {r["host"]: r["value"] for r in get_uptime()}
|
|
load = {r["host"]: r["value"] for r in get_load()}
|
|
|
|
hosts = sorted(set(list(cpu.keys()) + list(mem.keys()) + list(disk.keys())))
|
|
if not hosts:
|
|
return "Keine Prometheus-Daten verfügbar."
|
|
|
|
lines = [f"📊 Server-Metriken ({len(hosts)} Hosts)\n"]
|
|
for h in hosts:
|
|
c = cpu.get(h, -1)
|
|
m = mem.get(h, -1)
|
|
d = disk.get(h, -1)
|
|
dinfo = disk_gb.get(h, {})
|
|
u = uptime.get(h, 0)
|
|
l5 = load.get(h, -1)
|
|
|
|
warn = ""
|
|
if c >= WARN_CPU or m >= WARN_MEM or d >= WARN_DISK:
|
|
warn = " ⚠️"
|
|
|
|
disk_str = f"{d:.0f}%" if d >= 0 else "n/a"
|
|
if dinfo:
|
|
disk_str += f" ({dinfo.get('avail_gb', 0):.0f}/{dinfo.get('total_gb', 0):.0f} GB frei)"
|
|
|
|
emoji = "🟢" if c >= 0 else "🟡"
|
|
cpu_str = f"{c:.0f}%" if c >= 0 else "n/a"
|
|
mem_str = f"{m:.0f}%" if m >= 0 else "n/a"
|
|
load_str = f"{l5:.1f}" if l5 >= 0 else "n/a"
|
|
|
|
lines.append(
|
|
f"{emoji} {h}{warn}\n"
|
|
f" CPU: {cpu_str} RAM: {mem_str} Disk: {disk_str}\n"
|
|
f" Load5: {load_str} Uptime: {_fmt_uptime(u)}"
|
|
)
|
|
|
|
warnings = get_warnings()
|
|
if warnings:
|
|
lines.append("\n⚠️ WARNUNGEN:")
|
|
lines.extend(warnings)
|
|
else:
|
|
lines.append("\n✅ Alle Werte im Normalbereich.")
|
|
|
|
return "\n".join(lines)
|
|
|
|
|
|
def format_host_detail(host: str) -> str:
|
|
"""Detail-Metriken für einen einzelnen Host."""
|
|
if not is_available():
|
|
return "⚠️ Prometheus nicht erreichbar."
|
|
|
|
cpu = get_cpu(host)
|
|
mem = get_memory(host)
|
|
disk = get_disk(host)
|
|
disk_gb = get_disk_bytes(host)
|
|
uptime = get_uptime(host)
|
|
load = get_load(host)
|
|
|
|
if not cpu and not mem:
|
|
return f"Keine Metriken für '{host}' gefunden."
|
|
|
|
lines = [f"📊 {host} — Detail\n"]
|
|
if cpu:
|
|
lines.append(f"CPU: {cpu[0]['value']:.1f}%")
|
|
if mem:
|
|
lines.append(f"RAM: {mem[0]['value']:.1f}%")
|
|
if disk:
|
|
d_str = f"Disk: {disk[0]['value']:.1f}%"
|
|
if disk_gb:
|
|
d_str += f" ({disk_gb[0]['avail_gb']:.1f} / {disk_gb[0]['total_gb']:.1f} GB frei)"
|
|
lines.append(d_str)
|
|
if load:
|
|
lines.append(f"Load (5m): {load[0]['value']:.2f}")
|
|
if uptime:
|
|
lines.append(f"Uptime: {_fmt_uptime(uptime[0]['value'])}")
|
|
|
|
return "\n".join(lines)
|