mirror of
https://github.com/hansjone/netx.git
synced 2026-10-09 05:30:46 +08:00
oliver
042c015045
Introduce workbench operations for multi-vendor NE CRUD/connect-test and Netmiko batch collection with job lifecycle controls, log downloads, and paginated run filters. Co-authored-by: Cursor <cursoragent@cursor.com>
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
|
|
from sqlalchemy.orm import Session
|
|
|
|
from .collection_job_state import finalize_collection_job, reconcile_stale_collection_job
|
|
from .models import NeCollectionJob, NeCollectionRun
|
|
from .ne_collect_runner import schedule_collection_runs
|
|
|
|
_log = logging.getLogger("netx.collection.recovery")
|
|
|
|
|
|
def _parse_commands(text: str) -> list[str]:
|
|
lines: list[str] = []
|
|
for raw in str(text or "").splitlines():
|
|
line = raw.strip()
|
|
if not line or line.startswith("#"):
|
|
continue
|
|
lines.append(line)
|
|
return lines
|
|
|
|
|
|
def recover_collection_jobs_on_startup(db: Session) -> int:
|
|
"""Re-queue pending runs for running jobs after API restart; reconcile stale rows."""
|
|
jobs = db.query(NeCollectionJob).filter(NeCollectionJob.status == "running").all()
|
|
resumed = 0
|
|
for job in jobs:
|
|
job_id = str(job.id)
|
|
reconcile_stale_collection_job(db, job_id)
|
|
db.refresh(job)
|
|
if str(job.status or "") in ("done", "failed"):
|
|
continue
|
|
commands = _parse_commands(str(job.commands or ""))
|
|
if not commands:
|
|
job.status = "failed"
|
|
job.error_message = "commands_empty_on_recovery"
|
|
db.commit()
|
|
continue
|
|
pending = (
|
|
db.query(NeCollectionRun)
|
|
.filter(NeCollectionRun.job_id == job_id, NeCollectionRun.status == "pending")
|
|
.all()
|
|
)
|
|
if not pending:
|
|
finalize_collection_job(db, job_id)
|
|
continue
|
|
run_ids = [str(r.id) for r in pending]
|
|
schedule_collection_runs(job_id, run_ids, commands)
|
|
resumed += len(run_ids)
|
|
_log.info("resumed collection job=%s pending_runs=%s", job_id, len(run_ids))
|
|
return resumed
|