XANTRONIX Industrial
6d7b8023c2
Other changes:
* Refactor Database.query() to accept a list of clauses, rather than a
dict of key-value pairs to build '{k} = :{k}' clauses from
46 lines
1.2 KiB
Python
Executable file
46 lines
1.2 KiB
Python
Executable file
#! /usr/bin/env python3
|
|
|
|
import argparse
|
|
|
|
from xmet.db import Database
|
|
from xmet.raob import RAOBReader
|
|
from xmet.igra import IGRAStation
|
|
|
|
parser = argparse.ArgumentParser(
|
|
description = 'Ingest RAOB soundings'
|
|
)
|
|
|
|
parser.add_argument('--quiet', action='store_true', help='Suppress output')
|
|
parser.add_argument('--dry-run', action='store_true', help='Do not actually ingest data')
|
|
|
|
parser.add_argument('db', help='XMET SQLite3 database')
|
|
parser.add_argument('raob-sounding-file', nargs='+', help='RAOB sounding file')
|
|
|
|
args = parser.parse_args()
|
|
|
|
db = Database.connect(args.db)
|
|
|
|
if not args.dry_run:
|
|
db.execute('begin transaction')
|
|
|
|
for path in getattr(args, 'raob-sounding-file'):
|
|
if not args.quiet:
|
|
print(f"Ingesting sounding file {path}")
|
|
|
|
for sounding in RAOBReader.each_sounding_from_file(path):
|
|
station = IGRAStation.find_station(db, sounding.station)
|
|
|
|
if station is not None:
|
|
sounding.station = station.code
|
|
|
|
if args.dry_run:
|
|
continue
|
|
|
|
db.add(sounding)
|
|
|
|
for sample in sounding.samples:
|
|
sample.sounding_id = sounding.id
|
|
db.add(sample)
|
|
|
|
if not args.dry_run:
|
|
db.commit()
|