xan/nexrad-archive
xan/nexrad-archive
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Indicate storm event type in output

Browse source
This commit is contained in:
XANTRONIX 2025-02-15 21:51:22 -05:00
parent 231ae2dcca
commit 1872d4a0ce
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
bin/nexrad-archive
View file

@ -13,8 +13,9 @@ parser = argparse.ArgumentParser(
parser.add_argument('--quiet', action='store_true', help='Suppress output') parser.add_argument('--quiet', action='store_true', help='Suppress output')
parser.add_argument('--dry-run', action='store_true', help='Do not actually archive data') parser.add_argument('--dry-run', action='store_true', help='Do not actually archive data')
parser.add_argument('--exclude', action='append', type=str, help='Exclude types of reports from ingest')
parser.add_argument('db', help='SQLite3 NEXRAD radar site database') parser.add_argument('db', help='SQLite3 NEXRAD radar site database')
parser.add_argument('csv-report-details', nargs='+', help='Compressed storm report details CSV file') parser.add_argument('csv-report-details', nargs='+', help='Compressed storm report details CSV file')
parser.add_argument('archive-dir', help='Target archive directory') parser.add_argument('archive-dir', help='Target archive directory')
args = parser.parse_args() args = parser.parse_args()
@ -22,9 +23,16 @@ args = parser.parse_args()
db = Database.connect(args.db) db = Database.connect(args.db)
bucket = S3Bucket() bucket = S3Bucket()
archive = Archive(getattr(args, 'archive-dir'), bucket) archive = Archive(getattr(args, 'archive-dir'), bucket)
exclude = dict()
for event_type in args.exclude:
exclude[event_type] = True
for path in getattr(args, 'csv-report-details'): for path in getattr(args, 'csv-report-details'):
for report in StormReport.each_from_csv_file(path): for report in StormReport.each_from_csv_file(path):
if report.event_type in exclude:
continue
if report.coord_start is None or report.coord_end is None: if report.coord_start is None or report.coord_end is None:
continue continue
@ -36,13 +44,13 @@ for path in getattr(args, 'csv-report-details'):
for key in bucket.each_matching_key(radars, report.timestamp_start, report.timestamp_end): for key in bucket.each_matching_key(radars, report.timestamp_start, report.timestamp_end):
if archive.is_downloaded(key): if archive.is_downloaded(key):
if not args.quiet: if not args.quiet:
print(f"event {report.id} key {key} already archived") print(f"event {report.id} key {key} type {report.event_type} already archived")
else: else:
if not args.quiet: if not args.quiet:
if args.dry_run: if args.dry_run:
print(f"event {report.id} key {key} would archive") print(f"event {report.id} key {key} type {report.event_type} would archive")
else: else:
print(f"event {report.id} key {key} archiving") print(f"event {report.id} key {key} type {report.event_type} archiving")
if not args.dry_run: if not args.dry_run:
archive.download(key) archive.download(key)