diff --git a/bin/xmet-afos-ingest b/bin/xmet-afos-ingest index f5462a9..24f7a05 100755 --- a/bin/xmet-afos-ingest +++ b/bin/xmet-afos-ingest @@ -1,6 +1,6 @@ #! /usr/bin/env python3 -import sys +import argparse from xmet.db import Database from xmet.afos import AFOSMessageParser @@ -27,12 +27,24 @@ def each_chunk(fh, sep: str): else: yield part.strip() -db = Database.connect(sys.argv[1]) +parser = argparse.ArgumentParser( + description = 'Ingest National Weather Service text bulletin products' +) + +parser.add_argument('--quiet', action='store_true', help='Suppress output') +parser.add_argument('--dry-run', action='store_true', help='Do not actually ingest products') + +parser.add_argument('db', help='XMET SQLite3 database') +parser.add_argument('afos-text-file', help='AFOS text bulletin product file') + +args = parser.parse_args() + +db = Database.connect(args.db) db.execute('begin transaction') parser = AFOSMessageParser() -for path in sys.argv[2:]: +for path in getattr(args, 'afos-text-file'): with open(path, 'r') as fh: for data in each_chunk(fh, '\x01'): if len(data) == 0: @@ -41,7 +53,11 @@ for path in sys.argv[2:]: try: message = parser.parse(data) - db.add(message) + if not args.quiet: + print(f"Ingesting AFOS file {path}") + + if not args.dry_run: + db.add(message) except: pass