File size: 2,314 Bytes
8ede856 | 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 | from __future__ import annotations
from collections.abc import Mapping
from dataclasses import dataclass
from typing import Any
_DEFAULT_MAX_COMPONENT_CHAIN_DEPTH = 4
_DEFAULT_MAX_FORWARD_NODE_DEPTH = 6
_DEFAULT_MAX_FORWARD_FETCH = 32
def _read_int_mapping(
mapping: Mapping[str, Any],
key: str,
default: int,
) -> int:
raw = mapping.get(key)
if raw is None:
return default
try:
value = int(raw)
except (TypeError, ValueError):
return default
if value <= 0:
return default
return value
def _read_bool_mapping(
mapping: Mapping[str, Any],
key: str,
default: bool,
) -> bool:
raw = mapping.get(key)
if raw is None:
return default
if isinstance(raw, bool):
return raw
if isinstance(raw, str):
lowered = raw.strip().lower()
if lowered in {"1", "true", "yes", "on"}:
return True
if lowered in {"0", "false", "no", "off"}:
return False
return default
@dataclass(frozen=True)
class QuotedMessageParserSettings:
max_component_chain_depth: int = _DEFAULT_MAX_COMPONENT_CHAIN_DEPTH
max_forward_node_depth: int = _DEFAULT_MAX_FORWARD_NODE_DEPTH
max_forward_fetch: int = _DEFAULT_MAX_FORWARD_FETCH
warn_on_action_failure: bool = False
def with_overrides(
self,
overrides: Mapping[str, Any] | None = None,
) -> QuotedMessageParserSettings:
if not overrides:
return self
return QuotedMessageParserSettings(
max_component_chain_depth=_read_int_mapping(
overrides,
"max_component_chain_depth",
self.max_component_chain_depth,
),
max_forward_node_depth=_read_int_mapping(
overrides,
"max_forward_node_depth",
self.max_forward_node_depth,
),
max_forward_fetch=_read_int_mapping(
overrides,
"max_forward_fetch",
self.max_forward_fetch,
),
warn_on_action_failure=_read_bool_mapping(
overrides,
"warn_on_action_failure",
self.warn_on_action_failure,
),
)
SETTINGS = QuotedMessageParserSettings()
|