diff --git a/truenas_tools.py b/truenas_tools.py new file mode 100644 index 0000000..84e5c9b --- /dev/null +++ b/truenas_tools.py @@ -0,0 +1,251 @@ +import json +import time +import requests +from typing import Dict +from pydantic import BaseModel, Field + + +class Tools: + """ + Мониторинг TrueNAS через Open WebUI. + REST с перебором альтернативных путей. + TrueNAS 25.10 (Goldeye). + """ + + class Valves(BaseModel): + TRUENAS_URL: str = Field( + default="http://192.168.31.100", + description="IP-адрес или домен TrueNAS (http://...)", + ) + TRUENAS_API_KEY: str = Field( + default="", + description="API-ключ для доступа к TrueNAS (Settings → API Keys)", + ) + CACHE_TTL: int = Field( + default=30, description="Время кэширования в секундах (0 = без кэша)" + ) + + def __init__(self): + self.valves = self.Valves() + self._cache: Dict[str, tuple[float, str]] = {} + self._base = self.valves.TRUENAS_URL.rstrip("/") + self._headers = { + "Authorization": f"Bearer {self.valves.TRUENAS_API_KEY}", + "Content-Type": "application/json", + } + + def clear_cache(self): + self._cache.clear() + + def _rest_get(self, cache_key: str, endpoint: str): + now = time.time() + if cache_key in self._cache: + t, d = self._cache[cache_key] + if now - t < self.valves.CACHE_TTL: + return d + try: + r = requests.get( + f"{self._base}/api/v2.0/{endpoint}", headers=self._headers, timeout=10 + ) + if r.status_code == 200: + data = r.json() + s = json.dumps(data, default=str) + if self.valves.CACHE_TTL > 0: + self._cache[cache_key] = (now, s) + return s + return None + except Exception: + return None + + def _try_endpoints(self, cache_key: str, endpoints: list) -> str: + for ep in endpoints: + result = self._rest_get(f"{cache_key}_{ep.replace('/', '_')}", ep) + if result is not None: + return result + return f"⚠️ Данные недоступны. Попробованы пути: {', '.join(endpoints)}" + + def get_system_status(self) -> str: + raw = self._rest_get("info", "system/info") + if raw is None: + return "⚠️ Ошибка получения информации о системе" + try: + d = json.loads(raw) + except Exception: + return "Ошибка парсинга" + ver = d.get("version", "N/A") + model = d.get("model", "N/A") + load = d.get("loadavg", []) + load_s = ( + f"1мин: {load[0]:.2f}, 5мин: {load[1]:.2f}, 15мин: {load[2]:.2f}" + if len(load) >= 3 + else str(load) if load else "Н/Д" + ) + up = d.get("uptime_seconds") + if up: + dd, r = divmod(int(up), 86400) + hh, r = divmod(r, 3600) + mm, _ = divmod(r, 60) + up_s = f"{dd}д {hh}ч {mm}м" + else: + up_s = d.get("uptime_string", "Н/Д") + return ( + f"🖥️ TrueNAS: {ver}\n" + f"⏱️ Uptime: {up_s}\n" + f"🔧 Процессор: {model}\n" + f"📊 Load Average: {load_s}" + ) + + def get_pools_status(self) -> str: + raw = self._try_endpoints("pools", ["pool/query", "pool", "storage/pool"]) + if raw.startswith("⚠️"): + return raw + try: + pools = json.loads(raw) + except Exception: + return "Ошибка парсинга" + if not pools: + return "Активные пулы не найдены." + rows = [] + for p in pools: + name = p.get("name", "?") + status = p.get("status", "?") + icon = "✅" if p.get("healthy", False) else "⚠️" + total = p.get("size", 0) + free = p.get("free", 0) + used_pct = ((total - free) / total * 100) if total > 0 else 0 + rows.append( + f"{icon} '{name}' | {status} | {total/1024**4:.2f} TiB, {used_pct:.1f}% занято" + ) + return "\n".join(rows) + + def get_disks_temp_and_info(self) -> str: + raw = self._try_endpoints("disks", ["disk/query", "disk", "storage/disk"]) + if raw.startswith("⚠️"): + return raw + try: + disks = json.loads(raw) + except Exception: + return "Ошибка парсинга" + if not disks: + return "Диски не обнаружены." + rows = [] + for d in disks: + name = d.get("name", "N/A") + dtype = d.get("type", "N/A") + serial = d.get("serial", "N/A") + size = f"{int(d['size']) / (1024**3):.1f} GB" if d.get("size") else "N/A" + model = d.get("model", "") + rows.append(f"💾 {name} ({dtype}) | {model} | {serial} | {size}") + return "\n".join(rows) + + def get_apps_status(self) -> str: + """Список установленных приложений (TrueNAS Apps, docker compose).""" + raw = self._try_endpoints("apps", ["app/query", "app", "chart/release/query"]) + if raw.startswith("⚠️"): + return raw + try: + apps = json.loads(raw) + except Exception: + return "Ошибка парсинга" + if not apps: + return "Приложения не установлены." + imap = { + "RUNNING": "🟢", + "STOPPED": "⚪", + "DEPLOYING": "🟡", + "CRASHED": "🔴", + "STOPPING": "🟠", + } + rows = [] + for a in apps: + name = a.get("name", a.get("id", "?")) + state = a.get("state", a.get("status", "N/A")) + version = a.get("human_version", a.get("version", "N/A")) + icon = imap.get(str(state).upper(), "⚪") + rows.append(f"{icon} {name} | статус: {state} | версия: {version}") + return "\n".join(rows) + + def get_vms_status(self) -> str: + """Список виртуальных машин (libvirt/KVM).""" + raw = self._try_endpoints("vms", ["vm/query", "vm"]) + if raw.startswith("⚠️"): + return raw + try: + vms = json.loads(raw) + except Exception: + return "Ошибка парсинга" + if not vms: + return "Виртуальные машины не найдены." + rows = [] + for vm in vms: + name = vm.get("name", "?") + status = vm.get("status", {}) + state = status.get("state", "N/A") if isinstance(status, dict) else status + vcpus = vm.get("vcpus", "N/A") + memory = vm.get("memory") + mem_s = f"{int(memory) / 1024:.0f} MB" if memory else "N/A" + autostart = "✅" if vm.get("autostart") else "❌" + icon = "🟢" if str(state).upper() == "RUNNING" else "⚪" + rows.append( + f"{icon} {name} | статус: {state} | vCPU: {vcpus} | RAM: {mem_s} | автозапуск: {autostart}" + ) + return "\n".join(rows) + + def get_containers_status(self) -> str: + """Лёгкие контейнеры / инстансы (incus/virt, LXC-подобные).""" + raw = self._try_endpoints( + "containers", ["virt/instance", "virt/instance/query", "container/query"] + ) + if raw.startswith("⚠️"): + return raw + try: + instances = json.loads(raw) + except Exception: + return "Ошибка парсинга" + if not instances: + return "Контейнеры не найдены." + rows = [] + for c in instances: + name = c.get("name", c.get("id", "?")) + status = c.get("status", c.get("state", "N/A")) + itype = c.get("type", "N/A") + image = c.get("image", {}) + image_s = image.get("name", "N/A") if isinstance(image, dict) else image + icon = "🟢" if str(status).upper() == "RUNNING" else "⚪" + rows.append(f"{icon} {name} | тип: {itype} | статус: {status} | образ: {image_s}") + return "\n".join(rows) + + def get_active_alerts(self) -> str: + raw = self._rest_get("alerts", "alert/list") + if raw is None: + return "⚠️ Ошибка получения алертов" + try: + alerts = json.loads(raw) + except Exception: + return "Ошибка парсинга" + active = [a for a in alerts if not a.get("dismissed", False)] + if not active: + return "✅ Все системы работают штатно." + rows = [] + imap = {"CRITICAL": "🔴", "WARNING": "🟡", "INFO": "🔵"} + for a in active: + level = a.get("level", "INFO") + klass = a.get("klass", "Alert") + msg = a.get("formatted", a.get("message", "?")) + rows.append(f"{imap.get(level, '⚪')} [{level}] {klass}: {msg}") + return "\n".join(rows) + + def get_full_report(self) -> str: + return ( + "══════════════════════════════════════════════════\n" + "📋 ПОЛНЫЙ ОТЧЁТ TRUENAS\n" + "(REST перебор путей)\n" + "══════════════════════════════════════════════════\n\n" + "--- СИСТЕМА ---\n" + self.get_system_status() + "\n\n" + "--- ПУЛЫ ZFS ---\n" + self.get_pools_status() + "\n\n" + "--- ДИСКИ ---\n" + self.get_disks_temp_and_info() + "\n\n" + "--- ПРИЛОЖЕНИЯ ---\n" + self.get_apps_status() + "\n\n" + "--- ВИРТУАЛЬНЫЕ МАШИНЫ ---\n" + self.get_vms_status() + "\n\n" + "--- КОНТЕЙНЕРЫ (virt/incus) ---\n" + self.get_containers_status() + "\n\n" + "--- АЛЕРТЫ ---\n" + self.get_active_alerts() + )