#! /usr/bin/env python3

import argparse

from xmet.db    import Database
from xmet.radar import Radar
from xmet.wfo   import WFO

parser = argparse.ArgumentParser(
    description = 'Initialize NEXRAD radar site database table'
)

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('wfo-tsv',    help='Path to forecast office TSV file')

args = parser.parse_args()

db = Database.connect(args.db)

db.execute('begin transaction')

for radar in Radar.each_from_tsv(getattr(args, 'radars-tsv')):
    db.add(radar)

for wfo in WFO.each_from_tsv(getattr(args, 'wfo-tsv')):
    db.add(wfo)

db.commit()