mirror of
https://github.com/hansjone/netx.git
synced 2026-10-09 04:20:45 +08:00
oliver
17d136cf6b
AuxCommand is key+profile_id only; EnrichJoin copies fields after primary parse; ARP no longer hardcodes if_intf join. Co-authored-by: Cursor <cursoragent@cursor.com>
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
"""Declarative cross-command enrich (equal join) for biz-state metrics."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Any, Mapping, Sequence
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class EnrichJoin:
|
|
"""Copy fields from an aux metric table onto primary rows by equal key.
|
|
|
|
Example (ARP ← if_intf)::
|
|
|
|
EnrichJoin(from_aux="if_intf", on="interface", take=("vrf",))
|
|
"""
|
|
|
|
from_aux: str
|
|
take: tuple[str, ...]
|
|
on: str = ""
|
|
left_on: str = ""
|
|
right_on: str = ""
|
|
fill_missing: bool = True # on miss, set take fields to ""
|
|
|
|
|
|
def _sides(join: EnrichJoin) -> tuple[str, str]:
|
|
left = str(join.left_on or join.on or "").strip()
|
|
right = str(join.right_on or join.on or "").strip()
|
|
if not left or not right:
|
|
raise ValueError(f"EnrichJoin {join.from_aux!r} needs on= or left_on/right_on")
|
|
return left, right
|
|
|
|
|
|
def apply_enrich_joins(
|
|
records: list[dict[str, Any]],
|
|
aux_records: Mapping[str, list[dict[str, Any]]] | None,
|
|
joins: Sequence[EnrichJoin] | None,
|
|
) -> list[dict[str, Any]]:
|
|
"""In-place enrich of ``records``; returns the same list."""
|
|
if not records or not joins:
|
|
return records
|
|
aux_map = aux_records or {}
|
|
for join in joins:
|
|
left, right = _sides(join)
|
|
take = [str(t).strip() for t in (join.take or ()) if str(t).strip()]
|
|
if not take:
|
|
continue
|
|
index: dict[str, dict[str, Any]] = {}
|
|
for row in aux_map.get(str(join.from_aux or "").strip()) or []:
|
|
if not isinstance(row, dict):
|
|
continue
|
|
key = str(row.get(right) or "").strip()
|
|
if key and key not in index:
|
|
index[key] = row
|
|
for rec in records:
|
|
if not isinstance(rec, dict):
|
|
continue
|
|
hit = index.get(str(rec.get(left) or "").strip())
|
|
for field in take:
|
|
if hit is not None:
|
|
rec[field] = str(hit.get(field) or "").strip()
|
|
elif join.fill_missing:
|
|
rec[field] = ""
|
|
return records
|