80 lines
2.6 KiB
Python
80 lines
2.6 KiB
Python
"""Parser for Params/DBNames files."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import asdict, dataclass
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from .payload import parse_brace_text, payload_to_text, scalar
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class DBNamesRecord:
|
|
guid: str
|
|
storage_role: str
|
|
sql_number: int
|
|
index: int
|
|
source: str
|
|
|
|
def to_dict(self) -> dict[str, Any]:
|
|
return asdict(self)
|
|
|
|
|
|
def _unwrap_bom_sequence(value: Any) -> Any:
|
|
if (
|
|
isinstance(value, dict)
|
|
and value.get("type") == "sequence"
|
|
and len(value.get("items") or []) == 2
|
|
and isinstance(value["items"][0], dict)
|
|
and value["items"][0].get("type") == "atom"
|
|
and str(value["items"][0].get("value") or "").strip("\ufeff") == ""
|
|
):
|
|
return value["items"][1]
|
|
return value
|
|
|
|
|
|
def parse_dbnames_bytes(data: bytes, *, source: str = "DBNames") -> dict[str, Any]:
|
|
decoded = payload_to_text(data)
|
|
text = decoded.get("text")
|
|
if text is None:
|
|
raise ValueError(f"{source}: cannot decode DBNames text")
|
|
parsed = _unwrap_bom_sequence(parse_brace_text(text))
|
|
if not (isinstance(parsed, dict) and parsed.get("type") == "list"):
|
|
raise ValueError(f"{source}: expected root list")
|
|
items = parsed.get("items") or []
|
|
if len(items) != 2:
|
|
raise ValueError(f"{source}: expected 2 root items, got {len(items)}")
|
|
records_node = items[1]
|
|
if not (isinstance(records_node, dict) and records_node.get("type") == "list"):
|
|
raise ValueError(f"{source}: expected records list")
|
|
record_items = records_node.get("items") or []
|
|
records: list[DBNamesRecord] = []
|
|
for index, node in enumerate(record_items[1:], start=1):
|
|
if not (isinstance(node, dict) and node.get("type") == "list"):
|
|
continue
|
|
fields = node.get("items") or []
|
|
if len(fields) != 3:
|
|
continue
|
|
records.append(
|
|
DBNamesRecord(
|
|
guid=scalar(fields[0]).lower(),
|
|
storage_role=scalar(fields[1]),
|
|
sql_number=int(scalar(fields[2])),
|
|
index=index,
|
|
source=source,
|
|
)
|
|
)
|
|
return {
|
|
"source": source,
|
|
"compression": decoded["compression"],
|
|
"encoding": decoded["encoding"],
|
|
"root_number": int(scalar(items[0])),
|
|
"declared_count": int(scalar(record_items[0]) or "0") if record_items else 0,
|
|
"records": records,
|
|
}
|
|
|
|
|
|
def parse_dbnames_file(path: Path) -> dict[str, Any]:
|
|
return parse_dbnames_bytes(path.read_bytes(), source=path.name)
|