1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
|
from __future__ import annotations
import random
import re
from typing import TYPE_CHECKING
from typing import Optional
from zabbix_cli.exceptions import ZabbixAPICallError
from zabbix_cli.exceptions import ZabbixNotFoundError
from zabbix_cli.pyzabbix.client import ZabbixAPI
if TYPE_CHECKING:
from zabbix_cli.pyzabbix.types import Proxy
def get_random_proxy(client: ZabbixAPI, pattern: Optional[str] = None) -> Proxy:
"""Fetch a random proxy, optionally matching a regex pattern."""
proxies = client.get_proxies()
if not proxies:
raise ZabbixNotFoundError("No proxies found")
if pattern:
try:
re_pattern = re.compile(pattern)
except re.error as e:
raise ZabbixAPICallError(f"Invalid proxy regex pattern: {pattern!r}") from e
proxies = [proxy for proxy in proxies if re_pattern.match(proxy.name)]
if not proxies:
raise ZabbixNotFoundError(f"No proxies matching pattern {pattern!r}")
return random.choice(proxies)
def get_proxy_map(client: ZabbixAPI) -> dict[str, Proxy]:
"""Fetch all proxies and return a mapping of proxy IDs to Proxy objects."""
proxies = client.get_proxies()
return {proxy.proxyid: proxy for proxy in proxies}
|
