Add LandSAR cluster worker deployment

This commit is contained in:
2026-06-24 14:10:58 +08:00
parent 71c524967c
commit da09ba05cb
88 changed files with 9300 additions and 3011 deletions
+54
View File
@@ -5,6 +5,7 @@ import argparse
import asyncio
import json
import sys
from datetime import datetime
from pathlib import Path
from typing import Iterable, List, Optional
@@ -34,6 +35,55 @@ def _normalize_families(values: Optional[Iterable[str]]) -> List[str]:
async def _run(args: argparse.Namespace) -> dict:
families = _normalize_families(args.family)
database.init_db(settings.DATABASE_URL)
def _print_progress(payload: dict) -> None:
stamp = datetime.now().strftime("%H:%M:%S")
event = payload.get("event")
if event == "planned":
print(
"[{}] planned records_seen={} candidates={} skipped_ready={} families={}".format(
stamp,
payload.get("records_seen", 0),
payload.get("candidate_count", 0),
payload.get("skipped_ready", 0),
",".join(payload.get("families") or []),
),
flush=True,
)
return
if event == "item":
print(
(
"[{}] {}/{} ready={} failed={} missing={} raw_failed={} "
"skipped_ready={} status={} product={}"
).format(
stamp,
payload.get("processed", 0),
payload.get("total", 0),
payload.get("ready", 0),
payload.get("failed", 0),
payload.get("missing_source", 0),
payload.get("raw_failed", 0),
payload.get("skipped_ready", 0),
payload.get("status") or "",
payload.get("product_name") or "",
),
flush=True,
)
return
if event == "completed":
print(
"[{}] completed ready={} cached={} skipped_ready={} failed={} missing={}".format(
stamp,
payload.get("ready", 0),
payload.get("cached", 0),
payload.get("skipped_ready", 0),
payload.get("failed", 0),
payload.get("missing_source", 0),
),
flush=True,
)
summary = await asset_inventory_service.build_archive_preview_caches(
families=families,
limit=args.limit,
@@ -41,6 +91,8 @@ async def _run(args: argparse.Namespace) -> dict:
apply=bool(args.apply),
progress_start=0,
progress_end=100,
progress_callback=_print_progress if args.progress else None,
progress_interval=args.progress_interval,
)
return {"apply": bool(args.apply), **summary}
@@ -51,6 +103,8 @@ def main() -> int:
parser.add_argument("--family", action="append", help="family to process: LT1, S1, or comma-separated values")
parser.add_argument("--limit", type=int, default=0, help="maximum rows to build; 0 means all pending rows")
parser.add_argument("--force", action="store_true", help="rebuild even if preview_cache_status is READY")
parser.add_argument("--progress", action="store_true", help="print command-line progress while building")
parser.add_argument("--progress-interval", type=int, default=10, help="print every N processed candidates")
args = parser.parse_args()
payload = asyncio.run(_run(args))