#!/usr/bin/env python
# -*- coding: utf8 -*-
# SPDX-License-Identifier: GPL-3.0-or-later
"""
Main script for seiscat.
:copyright:
2022-2023 Claudio Satriano <satriano@ipgp.fr>
:license:
GNU General Public License v3.0 or later
(https://www.gnu.org/licenses/gpl-3.0-standalone.html)
"""
import sys
import argparse
import argcomplete
from .._version import get_versions
# NOTE: most modules are lazy-imported to speed up startup time
[docs]def parse_arguments():
"""Parse command line arguments."""
parser = argparse.ArgumentParser(
description='Keep a local seismic catalog.')
subparser = parser.add_subparsers(dest='action')
subparser.add_parser('initdb', help='initialize database')
subparser.add_parser('updatedb', help='update database')
versions_parser = argparse.ArgumentParser(add_help=False)
versions_parser.add_argument(
'-a',
'--allversions',
action='store_true',
default=False,
help='show all versions of each event (default: %(default)s)'
)
print_parser = subparser.add_parser(
'print', parents=[versions_parser], help='print catalog')
print_parser.add_argument(
'-f',
'--format',
type=str,
default='table',
choices=['table', 'stats'],
help='output format (default: %(default)s)'
)
print_parser.add_argument(
'-r',
'--reverse',
action='store_true',
default=False,
help='print catalog in reverse order (default: %(default)s)'
)
plot_parser = subparser.add_parser(
'plot', parents=[versions_parser], help='plot catalog map')
plot_parser.add_argument(
'-s',
'--scale',
type=float,
default=10,
help='scale factor for marker size (default: %(default)s)'
)
subparser.add_parser('sampleconfig', help='write sample config file')
parser.add_argument(
'-c',
'--configfile',
type=str,
default='seiscat.conf',
help='config file for data sources and processing params'
)
parser.add_argument(
'-v',
'--version',
action='version',
version=f"%(prog)s {get_versions()['version']}",
)
argcomplete.autocomplete(parser)
args = parser.parse_args()
if args.action is None:
parser.print_help()
sys.exit(0)
return args
[docs]def download_and_store(config, initdb):
"""
Download events and store them in the database.
:param config: config object
:param initdb: if True, create new database file
"""
from ..db import check_db_exists, write_catalog_to_db
from ..fdsnws import open_fdsn_connection, query_events
from ..utils import err_exit
check_db_exists(config, initdb)
try:
client = open_fdsn_connection(config)
except Exception as e:
err_exit(e)
cat = query_events(client, config, first_query=initdb)
write_catalog_to_db(cat, config, initdb)
[docs]def run():
"""Run seiscat."""
args = parse_arguments()
from ..utils import parse_configspec, read_config, write_sample_config
configspec = parse_configspec()
if args.action == 'sampleconfig':
write_sample_config(configspec, 'seiscat')
sys.exit(0)
config = read_config(args.configfile, configspec)
config['args'] = args
if args.action == 'initdb':
download_and_store(config, initdb=True)
elif args.action == 'updatedb':
download_and_store(config, initdb=False)
elif args.action == 'print':
from ..print import print_catalog
print_catalog(config)
elif args.action == 'plot':
from ..plot import plot_catalog_map
plot_catalog_map(config)
[docs]def main():
"""Main function. Catch KeyboardInterrupt."""
try:
run()
except KeyboardInterrupt:
sys.exit(1)