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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
from __future__ import annotations
import logging
from pathlib import Path
from typing import Any
from typing import Callable
from typing import Optional
from typing import Union
from enum import StrEnum
from zabbix_cli.dirs import CONFIG_DIR
from zabbix_cli.dirs import DATA_DIR
from zabbix_cli.dirs import LOGS_DIR
from zabbix_cli.dirs import SITE_CONFIG_DIR
logger = logging.getLogger("zabbix_cli.config")
# Config file basename
CONFIG_FILENAME = "zabbix-cli.toml"
DEFAULT_CONFIG_FILE = CONFIG_DIR / CONFIG_FILENAME
CONFIG_PRIORITY = (
Path() / CONFIG_FILENAME, # current directory
DEFAULT_CONFIG_FILE, # local config directory
SITE_CONFIG_DIR / CONFIG_FILENAME, # system config directory
)
# File path defaults
AUTH_TOKEN_FILE = DATA_DIR / ".zabbix-cli_auth_token"
"""Path to file containing API session token."""
AUTH_FILE = DATA_DIR / ".zabbix-cli_auth"
"""Path to file containing user credentials."""
SESSION_FILE = DATA_DIR / ".zabbix-cli_session.json"
"""Path to JSON file containing API session IDs."""
HISTORY_FILE = DATA_DIR / "history"
"""Path to file containing REPL history."""
LOG_FILE = LOGS_DIR / "zabbix-cli.log"
# Environment variable names
class ConfigEnvVars:
API_TOKEN = "ZABBIX_API_TOKEN"
PASSWORD = "ZABBIX_PASSWORD"
URL = "ZABBIX_URL"
USERNAME = "ZABBIX_USERNAME"
class OutputFormat(StrEnum):
CSV = "csv"
JSON = "json"
TABLE = "table"
class SecretMode(StrEnum):
"""Mode for serializing secrets."""
HIDE = "hide"
MASK = "masked"
PLAIN = "plain"
_DEFAULT = MASK
@classmethod
def from_context(
cls, context: Optional[Union[dict[str, Any], Callable[..., dict[str, Any]]]]
) -> SecretMode:
"""Get the secret mode from a serialization context."""
if isinstance(context, dict) and (ctx := context.get("secrets")) is not None:
# Support for old-style context values (true/false)
# as well as new context values (enum values)
try:
if isinstance(ctx, SecretMode):
return ctx
elif ctx is True:
return cls.PLAIN
elif ctx is False:
return cls.MASK
else:
return cls(str(ctx).lower())
except ValueError:
logger.warning(
"Got invalid secret mode from context %s: %s", context, ctx
)
return cls._DEFAULT
|
