mirror of
https://github.com/hansjone/netx.git
synced 2026-10-09 04:20:45 +08:00
Dedupe if-intf/config_interface collects and allow idle manual collect.
Remap legacy if_intf items, skip duplicate CLI+metric work, hide aux command duplicates in the workbook, use profile titles for sheet tabs, sticky table headers, and let scheduled tasks trigger collect-now when idle. Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
parent
18a0c41196
commit
99cec1c324
7 changed files with 241 additions and 42 deletions
|
|
@ -77,6 +77,25 @@ def _bindings_for_item(db, item_id: str) -> list[dict[str, str]]:
|
|||
]
|
||||
|
||||
|
||||
def _resolve_collect_profile(profile_id: str):
|
||||
"""Resolve task-item profile; remap disabled if_intf → config_interface."""
|
||||
pid = str(profile_id or "").strip()
|
||||
if not pid:
|
||||
return None
|
||||
profile = get_profile(pid)
|
||||
if profile is None:
|
||||
return None
|
||||
if profile.enabled:
|
||||
return profile
|
||||
# Legacy IF VRF check merged into Config Interface Intent — VRF is a subset.
|
||||
if profile.metric_id == "if_intf" or pid.endswith(".if_intf"):
|
||||
vk = str(profile.vendor_key or "zte").strip() or "zte"
|
||||
remapped = get_profile(f"{vk}.config_interface") or get_profile("zte.config_interface")
|
||||
if remapped and remapped.enabled:
|
||||
return remapped
|
||||
return None
|
||||
|
||||
|
||||
def _persist_lldp_rows(
|
||||
db,
|
||||
*,
|
||||
|
|
@ -199,8 +218,12 @@ def _finish_task(task_id: str, *, error: str = "") -> None:
|
|||
db.close()
|
||||
|
||||
|
||||
def dispatch_collect(task_id: str) -> None:
|
||||
"""Claim and run one collect round."""
|
||||
def dispatch_collect(task_id: str, *, manual: bool = False) -> None:
|
||||
"""Claim and run one collect round.
|
||||
|
||||
Scheduler calls with ``manual=False`` (only when task status is ``running``).
|
||||
Collect-now calls with ``manual=True`` (any status, as long as not already collecting).
|
||||
"""
|
||||
db = SessionLocal()
|
||||
batch_id = ""
|
||||
try:
|
||||
|
|
@ -209,8 +232,14 @@ def dispatch_collect(task_id: str) -> None:
|
|||
return
|
||||
if task.collect_running:
|
||||
return
|
||||
if str(task.status or "") not in ("running", "draft", "paused"):
|
||||
return
|
||||
st = str(task.status or "").strip()
|
||||
if manual:
|
||||
# Idle manual trigger: allow scheduled / paused / draft / stopped
|
||||
if st in ("", "deleted"):
|
||||
return
|
||||
else:
|
||||
if st != "running":
|
||||
return
|
||||
|
||||
items = (
|
||||
db.query(BizStateTaskItem)
|
||||
|
|
@ -338,19 +367,25 @@ def _run_collect_session(
|
|||
# Build work list before opening session
|
||||
work: list[tuple[str, dict[str, str], str, str, str]] = []
|
||||
# concrete, params, profile_id, item_id, mode
|
||||
# Dedupe same CLI → same metric (e.g. legacy if_intf + config_interface).
|
||||
seen_work: set[tuple[str, str]] = set()
|
||||
for item in items:
|
||||
if item.kind == "custom_raw":
|
||||
cmd = normalize_command(item.command_override)
|
||||
if cmd:
|
||||
key = (cmd, "__custom__")
|
||||
if key in seen_work:
|
||||
continue
|
||||
seen_work.add(key)
|
||||
work.append((cmd, {}, "", item.id, "custom"))
|
||||
continue
|
||||
profile = get_profile(item.source_profile_id)
|
||||
profile = _resolve_collect_profile(item.source_profile_id)
|
||||
if profile is None:
|
||||
_append_event(
|
||||
db,
|
||||
task_id=task_id,
|
||||
message=f"unknown profile {item.source_profile_id}",
|
||||
level="error",
|
||||
message=f"skip profile {item.source_profile_id} (missing or disabled)",
|
||||
level="info",
|
||||
)
|
||||
continue
|
||||
binds = _bindings_for_item(db, item.id)
|
||||
|
|
@ -364,7 +399,16 @@ def _run_collect_session(
|
|||
_append_event(db, task_id=task_id, message=str(exc), level="error")
|
||||
continue
|
||||
for concrete, params in pairs:
|
||||
work.append((concrete, params, profile.profile_id, item.id, "normal"))
|
||||
cmd = normalize_command(concrete)
|
||||
# Prefer match_command metric so remapped if_intf shares key with config_interface
|
||||
hit = match_command(vendor_key=vendor_key, command=cmd)
|
||||
mid = str((hit.profile.metric_id if hit else profile.metric_id) or "").strip()
|
||||
pid = str((hit.profile.profile_id if hit else profile.profile_id) or "").strip()
|
||||
key = (cmd, mid or pid)
|
||||
if key in seen_work:
|
||||
continue
|
||||
seen_work.add(key)
|
||||
work.append((cmd, params, pid or profile.profile_id, item.id, "normal"))
|
||||
|
||||
if not work:
|
||||
batch.status = "failed"
|
||||
|
|
@ -424,9 +468,15 @@ def _run_collect_session(
|
|||
params_json=dict(params or {}),
|
||||
created_at=_utcnow(),
|
||||
)
|
||||
cache_hit_primary = False
|
||||
try:
|
||||
raw = send_show_command(conn, concrete, read_timeout=per_cmd)
|
||||
cmd_row.raw_text = str(raw or "")
|
||||
cached = session.get_cached(concrete)
|
||||
if cached is not None and str(cached.raw or "").strip():
|
||||
cmd_row.raw_text = str(cached.raw or "")
|
||||
cache_hit_primary = True
|
||||
else:
|
||||
raw = send_show_command(conn, concrete, read_timeout=per_cmd)
|
||||
cmd_row.raw_text = str(raw or "")
|
||||
except Exception as exc:
|
||||
any_fail = True
|
||||
cmd_row.parse_status = "failed"
|
||||
|
|
@ -466,12 +516,16 @@ def _run_collect_session(
|
|||
sdb.commit()
|
||||
continue
|
||||
|
||||
session.remember(
|
||||
concrete,
|
||||
raw=cmd_row.raw_text or "",
|
||||
ok=True,
|
||||
cmd_row_id=cmd_row.id,
|
||||
)
|
||||
if cache_hit_primary:
|
||||
# Keep existing cache entry; primary will re-parse from shared raw.
|
||||
pass
|
||||
else:
|
||||
session.remember(
|
||||
concrete,
|
||||
raw=cmd_row.raw_text or "",
|
||||
ok=True,
|
||||
cmd_row_id=cmd_row.id,
|
||||
)
|
||||
|
||||
resolved_aux = []
|
||||
aux_results: dict[str, Any] = {}
|
||||
|
|
@ -684,5 +738,6 @@ def trigger_collect_now(task_id: str) -> dict[str, Any]:
|
|||
raise HTTPException(status_code=409, detail="collect already running")
|
||||
finally:
|
||||
db.close()
|
||||
dispatch_collect(task_id)
|
||||
# Manual: allow even when schedule is on (status=running) or paused/stopped.
|
||||
dispatch_collect(task_id, manual=True)
|
||||
return {"ok": True, "task_id": task_id}
|
||||
|
|
|
|||
|
|
@ -26,7 +26,7 @@ from ..models import (
|
|||
ManagedNE,
|
||||
)
|
||||
from ..timeutil import utcnow_naive
|
||||
from .command_match import preview_task_item
|
||||
from .command_match import normalize_command, preview_task_item
|
||||
from .profiles import (
|
||||
all_profiles,
|
||||
get_profile,
|
||||
|
|
@ -583,18 +583,25 @@ def get_batch(db: Session, batch_id: str) -> dict[str, Any]:
|
|||
cmd_payload: list[dict[str, Any]] = []
|
||||
sheets_order: list[str] = []
|
||||
sheet_cmds: dict[str, list[dict[str, Any]]] = {}
|
||||
sheet_titles: dict[str, str] = {}
|
||||
|
||||
def _push_sheet(mid: str, cmd_info: dict[str, Any] | None = None) -> None:
|
||||
def _push_sheet(mid: str, cmd_info: dict[str, Any] | None = None, title: str = "") -> None:
|
||||
id_ = str(mid or "").strip()
|
||||
if not id_ or id_ in ("vrf_list", "commands"):
|
||||
return
|
||||
if id_ not in sheets_order:
|
||||
sheets_order.append(id_)
|
||||
sheet_cmds.setdefault(id_, [])
|
||||
if title:
|
||||
sheet_titles[id_] = title
|
||||
if cmd_info is not None:
|
||||
# Prefer primary collect rows over aux / aux_cached for the same CLI
|
||||
sheet_cmds[id_].append(cmd_info)
|
||||
|
||||
primary_cmds_by_cli: dict[str, dict[str, Any]] = {}
|
||||
for c in cmds:
|
||||
status = str(c.parse_status or "").strip().lower()
|
||||
is_aux = status.startswith("aux")
|
||||
info = {
|
||||
"id": c.id,
|
||||
"profile_id": c.profile_id,
|
||||
|
|
@ -606,10 +613,28 @@ def get_batch(db: Session, batch_id: str) -> dict[str, Any]:
|
|||
"row_count": c.row_count,
|
||||
"message": c.message,
|
||||
"has_raw": bool(str(c.raw_text or "").strip()),
|
||||
"is_aux": is_aux,
|
||||
}
|
||||
cmd_n = normalize_command(str(c.raw_command or ""))
|
||||
if not is_aux and cmd_n:
|
||||
primary_cmds_by_cli.setdefault(cmd_n, info)
|
||||
# Commands sheet: hide aux when the same CLI already has a primary row
|
||||
if is_aux and cmd_n and cmd_n in primary_cmds_by_cli:
|
||||
continue
|
||||
if is_aux and cmd_n:
|
||||
# aux may appear before primary in list — defer; second pass below
|
||||
continue
|
||||
cmd_payload.append(info)
|
||||
mid = str(c.metric_id or "").strip()
|
||||
if mid and mid not in ("", "vrf_list"):
|
||||
title = ""
|
||||
pid = str(c.profile_id or "").strip()
|
||||
if pid:
|
||||
prof = get_profile(pid)
|
||||
if prof:
|
||||
title = str(prof.title or "")
|
||||
if not mid:
|
||||
mid = str(prof.metric_id or "").strip()
|
||||
if mid and mid not in ("", "vrf_list") and not is_aux:
|
||||
_push_sheet(
|
||||
mid,
|
||||
{
|
||||
|
|
@ -621,17 +646,49 @@ def get_batch(db: Session, batch_id: str) -> dict[str, Any]:
|
|||
"has_raw": info["has_raw"],
|
||||
"profile_id": c.profile_id,
|
||||
},
|
||||
title=title,
|
||||
)
|
||||
# Aux command rows may have empty metric_id — still attach by profile if needed later
|
||||
|
||||
# Include aux-only CLIs that had no primary counterpart
|
||||
for c in cmds:
|
||||
status = str(c.parse_status or "").strip().lower()
|
||||
if not status.startswith("aux"):
|
||||
continue
|
||||
cmd_n = normalize_command(str(c.raw_command or ""))
|
||||
if cmd_n and cmd_n in primary_cmds_by_cli:
|
||||
continue
|
||||
cmd_payload.append(
|
||||
{
|
||||
"id": c.id,
|
||||
"profile_id": c.profile_id,
|
||||
"parser_id": c.parser_id,
|
||||
"metric_id": c.metric_id,
|
||||
"raw_command": c.raw_command,
|
||||
"params": c.params_json or {},
|
||||
"parse_status": c.parse_status,
|
||||
"row_count": c.row_count,
|
||||
"message": c.message,
|
||||
"has_raw": bool(str(c.raw_text or "").strip()),
|
||||
"is_aux": True,
|
||||
}
|
||||
)
|
||||
|
||||
for mid in metric_counts:
|
||||
if mid not in sheets_order:
|
||||
sheets_order.append(mid)
|
||||
sheet_cmds.setdefault(mid, [])
|
||||
if mid not in sheet_titles:
|
||||
# Best-effort title from any profile with this metric_id
|
||||
for p in all_profiles():
|
||||
if p.metric_id == mid and p.enabled:
|
||||
sheet_titles[mid] = str(p.title or mid)
|
||||
break
|
||||
sheet_titles.setdefault(mid, mid)
|
||||
|
||||
sheets = [
|
||||
{
|
||||
"metric_id": mid,
|
||||
"title": sheet_titles.get(mid) or mid,
|
||||
"row_count": int(metric_counts.get(mid) or 0),
|
||||
"commands": list(sheet_cmds.get(mid) or []),
|
||||
}
|
||||
|
|
|
|||
|
|
@ -193,10 +193,11 @@ def api_collect_now(
|
|||
raise HTTPException(status_code=404, detail="task not found")
|
||||
if bool(task.collect_running):
|
||||
return {"ok": True, "started": False, "reason": "already_collecting", "task_id": task_id}
|
||||
# Allow one-shot from draft/paused
|
||||
# Allow one-shot even when schedule is enabled (idle only).
|
||||
task.last_collect_ended_at = None
|
||||
db.commit()
|
||||
background_tasks.add_task(dispatch_collect, task_id)
|
||||
tid = task_id
|
||||
background_tasks.add_task(lambda: dispatch_collect(tid, manual=True))
|
||||
return {"ok": True, "started": True, "task_id": task_id}
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue