mirror of
https://github.com/PlaneQuery/OpenAirframes.git
synced 2026-09-14 01:48:55 +02:00
feat: build each registry source on its own thread and join them
- add a reusable registry-source workflow so every source runs in parallel rather than as a hand-written job; adding a registry becomes one matrix entry plus one script - add build_registry.py to align the per-source CSVs on the union of columns and emit a single table discriminated by the source column - reindex each frame to the union before concatenating, so a source missing a column yields an empty cell rather than a shifted row Generated-by: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,70 @@
|
||||
name: registry-source
|
||||
|
||||
# One registry source, on its own thread. Called once per entry in the caller's matrix, so
|
||||
# adding a registry is a matrix entry plus src/create_daily_<source>_release.py — no new job.
|
||||
|
||||
on:
|
||||
workflow_call:
|
||||
inputs:
|
||||
source:
|
||||
description: 'Source id; must match src/create_daily_<source>_release.py'
|
||||
required: true
|
||||
type: string
|
||||
date:
|
||||
description: 'Date to process (YYYY-MM-DD, default: today UTC)'
|
||||
required: false
|
||||
type: string
|
||||
python-version:
|
||||
required: false
|
||||
type: string
|
||||
default: '3.14'
|
||||
required:
|
||||
description: 'Fail the thread when this source cannot be built'
|
||||
required: false
|
||||
type: boolean
|
||||
default: false
|
||||
|
||||
jobs:
|
||||
build:
|
||||
runs-on: ubuntu-24.04-arm
|
||||
timeout-minutes: 30
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v6
|
||||
with:
|
||||
fetch-depth: 0
|
||||
|
||||
- name: Setup Python
|
||||
uses: actions/setup-python@v6
|
||||
with:
|
||||
python-version: ${{ inputs.python-version }}
|
||||
|
||||
- name: Install dependencies
|
||||
run: |
|
||||
python -m pip install --upgrade pip
|
||||
pip install -r requirements.txt
|
||||
|
||||
- name: Build ${{ inputs.source }} registry
|
||||
continue-on-error: ${{ !inputs.required }}
|
||||
env:
|
||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||
SOURCE: ${{ inputs.source }}
|
||||
RUN_DATE: ${{ inputs.date }}
|
||||
run: |
|
||||
script="src/create_daily_${SOURCE}_release.py"
|
||||
if [ ! -f "$script" ]; then
|
||||
echo "::error title=Unknown registry source::$script does not exist"
|
||||
exit 1
|
||||
fi
|
||||
python "$script" ${RUN_DATE:+--date "$RUN_DATE"}
|
||||
ls -lah data/openairframes
|
||||
|
||||
- name: Upload ${{ inputs.source }} registry
|
||||
uses: actions/upload-artifact@v4
|
||||
with:
|
||||
name: registry-${{ inputs.source }}
|
||||
path: |
|
||||
data/openairframes/openairframes_${{ inputs.source }}_*.csv
|
||||
data/faa_releasable/ReleasableAircraft_*.zip
|
||||
retention-days: 1
|
||||
if-no-files-found: ignore
|
||||
@@ -0,0 +1,102 @@
|
||||
"""Join the per-source registry CSVs into one union table.
|
||||
|
||||
Every source publishes its own `openairframes_<source>_{start}_{end}.csv` on its own thread.
|
||||
This reads whatever landed, aligns them on the union of columns, and writes a single
|
||||
`openairframes_registry_{start}_{end}.csv` discriminated by the `source` column.
|
||||
|
||||
Adding a registry means adding a source to the workflow matrix; nothing here changes.
|
||||
|
||||
Usage:
|
||||
python src/build_registry.py --input-dir artifacts/registry --date 2026-08-31
|
||||
"""
|
||||
from datetime import datetime, timezone
|
||||
from pathlib import Path
|
||||
import argparse
|
||||
import re
|
||||
import sys
|
||||
|
||||
import pandas as pd
|
||||
|
||||
# Sources that are registries. Community and ADS-B are published separately: they are
|
||||
# observations and contributions, not registration records, and do not share this schema.
|
||||
FILENAME_RE = re.compile(r"openairframes_(?P<source>[a-z]+)_(?P<start>\d{4}-\d{2}-\d{2})_(?P<end>\d{4}-\d{2}-\d{2})\.csv$")
|
||||
EXCLUDED_SOURCES = {"community", "adsb", "registry"}
|
||||
|
||||
# Identifier columns lead the union so the table is usable without reading 70 headers.
|
||||
LEADING_COLUMNS = [
|
||||
"download_date",
|
||||
"source",
|
||||
"transponder_code_hex",
|
||||
"registration_number",
|
||||
"openairframes_id",
|
||||
]
|
||||
|
||||
|
||||
def discover(input_dir: Path) -> list[tuple[str, str, str, Path]]:
|
||||
"""Return (source, start, end, path) for each per-source registry CSV found."""
|
||||
found = []
|
||||
for path in sorted(input_dir.rglob("openairframes_*.csv")):
|
||||
match = FILENAME_RE.search(path.name)
|
||||
if not match:
|
||||
continue
|
||||
source = match.group("source")
|
||||
if source in EXCLUDED_SOURCES:
|
||||
continue
|
||||
found.append((source, match.group("start"), match.group("end"), path))
|
||||
return found
|
||||
|
||||
|
||||
def build(input_dir: Path, date_str: str) -> tuple[pd.DataFrame, str, str]:
|
||||
parts = discover(input_dir)
|
||||
if not parts:
|
||||
raise SystemExit(f"No per-source registry CSVs found under {input_dir}")
|
||||
|
||||
frames = []
|
||||
for source, _, _, path in parts:
|
||||
# keep_default_na=False so a literal "NA" survives the round trip unchanged.
|
||||
df = pd.read_csv(path, dtype=str, keep_default_na=False)
|
||||
if "source" not in df.columns:
|
||||
raise SystemExit(f"{path.name}: no source column; cannot discriminate rows")
|
||||
actual = set(df["source"].unique())
|
||||
if len(actual) != 1:
|
||||
raise SystemExit(f"{path.name}: expected one source value, found {sorted(actual)}")
|
||||
print(f" {source}: {len(df)} rows, {len(df.columns)} columns from {path.name}")
|
||||
frames.append(df)
|
||||
|
||||
columns = list(dict.fromkeys(c for df in frames for c in df.columns))
|
||||
ordered = [c for c in LEADING_COLUMNS if c in columns]
|
||||
ordered += [c for c in columns if c not in ordered]
|
||||
|
||||
# reindex rather than concat directly: a source missing a column must yield an empty
|
||||
# cell, never a shifted row.
|
||||
df_union = pd.concat([df.reindex(columns=ordered) for df in frames], ignore_index=True)
|
||||
df_union = df_union.fillna("")
|
||||
|
||||
start = min(p[1] for p in parts)
|
||||
end = max(p[2] for p in parts + [("", "", date_str, Path())])
|
||||
return df_union, start, end
|
||||
|
||||
|
||||
def main() -> None:
|
||||
parser = argparse.ArgumentParser(description="Join per-source registry CSVs into one table")
|
||||
parser.add_argument("--input-dir", default="artifacts/registry", help="Directory to search")
|
||||
parser.add_argument("--output-dir", default="data/openairframes", help="Where to write")
|
||||
parser.add_argument("--date", help="Run date (YYYY-MM-DD, default: today UTC)")
|
||||
args = parser.parse_args()
|
||||
|
||||
date_str = args.date or datetime.now(timezone.utc).strftime("%Y-%m-%d")
|
||||
df, start, end = build(Path(args.input_dir), date_str)
|
||||
|
||||
out_dir = Path(args.output_dir)
|
||||
out_dir.mkdir(parents=True, exist_ok=True)
|
||||
out_path = out_dir / f"openairframes_registry_{start}_{end}.csv"
|
||||
df.to_csv(out_path, index=False)
|
||||
|
||||
print(f"Wrote {out_path}: {len(df)} rows, {len(df.columns)} columns")
|
||||
print(" rows per source:")
|
||||
for source, count in df["source"].value_counts().items():
|
||||
print(f" {source}: {count}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user