mirror of
https://github.com/PlaneQuery/OpenAirframes.git
synced 2026-04-23 11:36:35 +02:00
ggman12
83 lines
2.8 KiB
Python
83 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Run src.adsb.main in an isolated git worktree so edits in the main
|
|
working tree won't affect subprocess imports during the run.
|
|
|
|
Usage:
|
|
python scripts/run_main_isolated.py 2026-01-01
|
|
python scripts/run_main_isolated.py --start_date 2026-01-01 --end_date 2026-01-03
|
|
"""
|
|
import argparse
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
from datetime import datetime, timezone
|
|
from pathlib import Path
|
|
|
|
|
|
def run(
|
|
cmd: list[str],
|
|
*,
|
|
cwd: Path | None = None,
|
|
check: bool = True,
|
|
) -> subprocess.CompletedProcess:
|
|
print(f"\n>>> {' '.join(cmd)}")
|
|
return subprocess.run(cmd, cwd=cwd, check=check)
|
|
|
|
|
|
def main() -> int:
|
|
parser = argparse.ArgumentParser(description="Run src.adsb.main in an isolated worktree")
|
|
parser.add_argument("date", nargs="?", help="Single date to process (YYYY-MM-DD)")
|
|
parser.add_argument("--start_date", help="Start date (inclusive, YYYY-MM-DD)")
|
|
parser.add_argument("--end_date", help="End date (exclusive, YYYY-MM-DD)")
|
|
parser.add_argument("--concat_with_latest_csv", action="store_true", help="Also concatenate with latest CSV from GitHub releases")
|
|
args = parser.parse_args()
|
|
|
|
if args.date and (args.start_date or args.end_date):
|
|
raise SystemExit("Use a single date or --start_date/--end_date, not both.")
|
|
|
|
if args.date:
|
|
datetime.strptime(args.date, "%Y-%m-%d")
|
|
main_args = ["--date", args.date]
|
|
else:
|
|
if not args.start_date or not args.end_date:
|
|
raise SystemExit("Provide --start_date and --end_date, or a single date.")
|
|
datetime.strptime(args.start_date, "%Y-%m-%d")
|
|
datetime.strptime(args.end_date, "%Y-%m-%d")
|
|
main_args = ["--start_date", args.start_date, "--end_date", args.end_date]
|
|
|
|
if args.concat_with_latest_csv:
|
|
main_args.append("--concat_with_latest_csv")
|
|
|
|
repo_root = Path(__file__).resolve().parents[1]
|
|
snapshots_root = repo_root / ".snapshots"
|
|
snapshots_root.mkdir(exist_ok=True)
|
|
|
|
timestamp = datetime.now(timezone.utc).strftime("%Y%m%d_%H%M%S")
|
|
snapshot_root = snapshots_root / f"run_{timestamp}"
|
|
snapshot_src = snapshot_root / "src"
|
|
|
|
exit_code = 0
|
|
try:
|
|
shutil.copytree(repo_root / "src", snapshot_src)
|
|
|
|
runner = (
|
|
"import sys, runpy; "
|
|
f"sys.path.insert(0, {repr(str(snapshot_root))}); "
|
|
f"sys.argv = ['src.adsb.main'] + {main_args!r}; "
|
|
"runpy.run_module('src.adsb.main', run_name='__main__')"
|
|
)
|
|
cmd = [sys.executable, "-c", runner]
|
|
run(cmd, cwd=repo_root)
|
|
except subprocess.CalledProcessError as exc:
|
|
exit_code = exc.returncode
|
|
finally:
|
|
shutil.rmtree(snapshot_root, ignore_errors=True)
|
|
|
|
return exit_code
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|