Update nexrad-archive-init to populate nexrad_wfo table

This commit is contained in:
XANTRONIX Industrial 2025-02-21 16:30:19 -05:00
parent 3fa26aca83
commit 8fb9f67a25
2 changed files with 8 additions and 2 deletions

View file

@ -9,7 +9,7 @@ RUN mkdir -p /var/opt/nexrad-archive/lib/nexrad
RUN mkdir -p /var/opt/nexrad-archive/bin RUN mkdir -p /var/opt/nexrad-archive/bin
RUN mkdir -p /var/lib/nexrad-archive RUN mkdir -p /var/lib/nexrad-archive
COPY db/nexrad.sql doc/radars.tsv /tmp COPY db/nexrad.sql doc/radars.tsv doc/wfo.tsv /tmp
COPY lib/nexrad/*.py /var/opt/nexrad-archive/lib/nexrad COPY lib/nexrad/*.py /var/opt/nexrad-archive/lib/nexrad
COPY bin/nexrad-archive bin/nexrad-archive-init /var/opt/nexrad-archive/bin COPY bin/nexrad-archive bin/nexrad-archive-init /var/opt/nexrad-archive/bin
@ -19,6 +19,7 @@ RUN sqlite3 -init /tmp/nexrad.sql /var/lib/nexrad-archive/nexrad.db .quit
RUN /var/opt/nexrad-archive/bin/nexrad-archive-init \ RUN /var/opt/nexrad-archive/bin/nexrad-archive-init \
/var/lib/nexrad-archive/nexrad.db \ /var/lib/nexrad-archive/nexrad.db \
/tmp/radars.tsv /tmp/radars.tsv \
/tmp/wfo.tsv
ENTRYPOINT ["/var/opt/nexrad-archive/bin/nexrad-archive", "/var/lib/nexrad-archive/nexrad.db"] ENTRYPOINT ["/var/opt/nexrad-archive/bin/nexrad-archive", "/var/lib/nexrad-archive/nexrad.db"]

View file

@ -4,6 +4,7 @@ import argparse
from nexrad.db import Database from nexrad.db import Database
from nexrad.radar import Radar from nexrad.radar import Radar
from nexrad.wfo import WFO
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
description = 'Initialize NEXRAD radar site database' description = 'Initialize NEXRAD radar site database'
@ -11,6 +12,7 @@ parser = argparse.ArgumentParser(
parser.add_argument('db', help='Path to SQLite3 database') parser.add_argument('db', help='Path to SQLite3 database')
parser.add_argument('radars-tsv', help='Path to NEXRAD radar station TSV file') parser.add_argument('radars-tsv', help='Path to NEXRAD radar station TSV file')
parser.add_argument('wfo-tsv', help='Path to forecast office TSV file')
args = parser.parse_args() args = parser.parse_args()
@ -21,4 +23,7 @@ db.execute('begin transaction')
for radar in Radar.each_from_tsv(getattr(args, 'radars-tsv')): for radar in Radar.each_from_tsv(getattr(args, 'radars-tsv')):
db.add(radar) db.add(radar)
for wfo in WFO.each_from_tsv(getattr(args, 'wfo-tsv')):
db.add(wfo)
db.commit() db.commit()