92 lines
2.8 KiB
Python
92 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
try:
|
|
from .lt1_input_resolver import repair_dem_sidecar_paths
|
|
except ImportError:
|
|
SCRIPT_DIR = Path(__file__).resolve().parent
|
|
if str(SCRIPT_DIR) not in sys.path:
|
|
sys.path.insert(0, str(SCRIPT_DIR))
|
|
from lt1_input_resolver import repair_dem_sidecar_paths # type: ignore
|
|
|
|
|
|
def parse_args() -> argparse.Namespace:
|
|
parser = argparse.ArgumentParser(
|
|
description="Audit and optionally repair moved ISCE DEM XML sidecars."
|
|
)
|
|
parser.add_argument(
|
|
"--root",
|
|
type=Path,
|
|
required=True,
|
|
help="Directory containing DEM files and sidecars",
|
|
)
|
|
parser.add_argument(
|
|
"--repair",
|
|
action="store_true",
|
|
help="Write repaired file_name / metadata_location / extra_file_name values back to XML",
|
|
)
|
|
return parser.parse_args()
|
|
|
|
|
|
def iter_dem_sidecars(root: Path) -> list[Path]:
|
|
sidecars: list[Path] = []
|
|
for xml_path in sorted(root.rglob("*.xml")):
|
|
if xml_path.name.lower().endswith(".aux.xml"):
|
|
continue
|
|
dem_path = Path(str(xml_path)[:-4])
|
|
if dem_path.exists():
|
|
sidecars.append(dem_path)
|
|
return sidecars
|
|
|
|
|
|
def main() -> int:
|
|
args = parse_args()
|
|
root = args.root.resolve()
|
|
if not root.exists() or not root.is_dir():
|
|
raise FileNotFoundError(f"DEM root directory not found: {root}")
|
|
|
|
sidecars = iter_dem_sidecars(root)
|
|
changed_count = 0
|
|
mismatch_count = 0
|
|
|
|
print(f"DEM root: {root}")
|
|
print(f"Sidecars: {len(sidecars)}")
|
|
for dem_path in sidecars:
|
|
report = repair_dem_sidecar_paths(dem_path, write_changes=bool(args.repair))
|
|
updated_fields = list(report.get("updated_fields") or [])
|
|
if updated_fields:
|
|
changed_count += 1
|
|
mismatch_count += 1
|
|
print(
|
|
f"[fixed] {report['xml_path']} -> {', '.join(updated_fields)}"
|
|
if args.repair
|
|
else f"[mismatch] {report['xml_path']} -> {', '.join(updated_fields)}"
|
|
)
|
|
continue
|
|
|
|
current = report.get("current") or {}
|
|
expected = report.get("expected") or {}
|
|
mismatched = [
|
|
key
|
|
for key, expected_value in expected.items()
|
|
if expected_value and str(current.get(key) or "").strip() != str(expected_value).strip()
|
|
]
|
|
if mismatched:
|
|
mismatch_count += 1
|
|
print(f"[mismatch] {report['xml_path']} -> {', '.join(mismatched)}")
|
|
else:
|
|
print(f"[ok] {report['xml_path']}")
|
|
|
|
print(f"Mismatched: {mismatch_count}")
|
|
if args.repair:
|
|
print(f"Repaired: {changed_count}")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|