Загрузить файлы в «alert-processor/app»

2026-08-06 18:34:46 +03:00
parent fa0c42f0b1
commit f71f5f5fc1
2 changed files with 422 additions and 0 deletions
@@ -0,0 +1,152 @@
+from __future__ import annotations
+
+import itertools
+from typing import Any
+
+import httpx
+
+
+class ZabbixApiClient:
+    def __init__(
+        self,
+        api_url: str,
+        api_token: str,
+        timeout_seconds: float = 10,
+        verify_tls: bool = True,
+    ) -> None:
+        self.api_url = api_url
+        self.api_token = api_token
+        self._id_counter = itertools.count(1)
+        self.client = httpx.AsyncClient(
+            timeout=timeout_seconds,
+            verify=verify_tls,
+            headers={
+                "Authorization": f"Bearer {self.api_token}",
+                "Content-Type": "application/json-rpc",
+            },
+        )
+
+    async def close(self) -> None:
+        await self.client.aclose()
+
+    async def _call(self, method: str, params: dict[str, Any]) -> Any:
+        payload = {
+            "jsonrpc": "2.0",
+            "method": method,
+            "params": params,
+            "id": next(self._id_counter),
+        }
+
+        response = await self.client.post(self.api_url, json=payload)
+        response.raise_for_status()
+
+        data = response.json()
+        if "error" in data:
+            error = data["error"]
+            raise RuntimeError(
+                f"Zabbix API error: method={method} "
+                f"code={error.get('code')} message={error.get('message')} "
+                f"data={error.get('data')}"
+            )
+
+        return data.get("result")
+
+    async def get_trigger(self, trigger_id: str) -> dict[str, Any] | None:
+        result = await self._call(
+            "trigger.get",
+            {
+                "output": [
+                    "triggerid",
+                    "description",
+                    "comments",
+                    "opdata",
+                    "priority",
+                    "state",
+                    "status",
+                    "value",
+                ],
+                "triggerids": [trigger_id],
+                "selectHosts": ["hostid", "host", "name"],
+                "selectItems": ["itemid", "name", "key_", "lastvalue", "units", "value_type"],
+                "selectTags": "extend",
+                "limit": 1,
+            },
+        )
+        return result[0] if result else None
+
+    async def get_item(self, item_id: str) -> dict[str, Any] | None:
+        result = await self._call(
+            "item.get",
+            {
+                "output": [
+                    "itemid",
+                    "name",
+                    "key_",
+                    "lastvalue",
+                    "units",
+                    "value_type",
+                    "status",
+                ],
+                "itemids": [item_id],
+                "selectHosts": ["hostid", "host", "name"],
+                "selectTags": "extend",
+                "limit": 1,
+            },
+        )
+        return result[0] if result else None
+
+    async def get_event(self, event_id: str) -> dict[str, Any] | None:
+        result = await self._call(
+            "event.get",
+            {
+                "output": [
+                    "eventid",
+                    "objectid",
+                    "clock",
+                    "name",
+                    "severity",
+                    "value",
+                    "acknowledged",
+                ],
+                "eventids": [event_id],
+                "selectHosts": ["hostid", "host", "name"],
+                "selectTags": "extend",
+                "limit": 1,
+            },
+        )
+        return result[0] if result else None
+
+    async def get_host(self, host_id: str) -> dict[str, Any] | None:
+        result = await self._call(
+            "host.get",
+            {
+                "output": ["hostid", "host", "name"],
+                "hostids": [host_id],
+                "selectTags": "extend",
+                "limit": 1,
+            },
+        )
+        return result[0] if result else None
+
+    async def get_history(
+        self,
+        item_id: str,
+        value_type: int,
+        time_from: int,
+        time_till: int,
+        limit: int = 500,
+    ) -> list[dict[str, Any]]:
+        result = await self._call(
+            "history.get",
+            {
+                "output": ["clock", "value"],
+                "history": value_type,
+                "itemids": [item_id],
+                "time_from": time_from,
+                "time_till": time_till,
+                "sortfield": "clock",
+                "sortorder": "ASC",
+                "limit": limit,
+            },
+        )
+        return result or []