Added skip-blended.

This commit is contained in:
Felix Delattre 2026-06-11 00:42:02 +02:00
parent 79ee099809
commit aa4535a600

View file

@ -298,6 +298,11 @@ def main(argv: list[str] | None = None) -> int:
default=None, default=None,
help="Single sitename to fuse (default: all prepared sites)", help="Single sitename to fuse (default: all prepared sites)",
) )
parser.add_argument(
"--skip-blended",
action="store_true",
help="Skip sites whose fusion directory already exists under data/fusion/{year}/",
)
args = parser.parse_args(argv) args = parser.parse_args(argv)
year = args.evaluation_year year = args.evaluation_year
@ -311,17 +316,18 @@ def main(argv: list[str] | None = None) -> int:
print(f"[Fusion] Processing {len(sites)} site(s)") print(f"[Fusion] Processing {len(sites)} site(s)")
for i, sitename in enumerate(sites, 1): for i, sitename in enumerate(sites, 1):
fusion_dir = DATA_DIR / "fusion" / str(year) / sitename
if args.skip_blended and fusion_dir.exists():
print(f"[Fusion] ({i}/{len(sites)}) {sitename} — skipping (fusion directory exists)")
continue
print(f"[Fusion] ({i}/{len(sites)}) {sitename}") print(f"[Fusion] ({i}/{len(sites)}) {sitename}")
try: summary = fuse_site(sitename, year)
summary = fuse_site(sitename, year) print(
print( f"[Fusion] {sitename} done — "
f"[Fusion] {sitename} done — " f"{summary['fusion_dates']} dates, "
f"{summary['fusion_dates']} dates, " f"itb={summary['itb_fusion_files']} bti={summary['bti_fusion_files']} "
f"itb={summary['itb_fusion_files']} bti={summary['bti_fusion_files']} " f"bti_gcc={summary['bti_gcc_files']}"
f"bti_gcc={summary['bti_gcc_files']}" )
)
except Exception as exc:
print(f"[Fusion] {sitename} FAILED: {exc}")
return 0 return 0