Added skip-blended.
This commit is contained in:
parent
79ee099809
commit
aa4535a600
1 changed files with 16 additions and 10 deletions
12
4-fusion.py
12
4-fusion.py
|
|
@ -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,8 +316,11 @@ 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 — "
|
||||||
|
|
@ -320,8 +328,6 @@ def main(argv: list[str] | None = None) -> int:
|
||||||
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
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue