1# http://pyrocko.org - GPLv3
2#
3# The Pyrocko Developers, 21st Century
4# ---|P------/S----------~Lg----------
6from __future__ import absolute_import, print_function
8from pyrocko.squirrel.model import to_kind
11headline = 'Get summary of available data codes.'
14def make_subparser(subparsers):
15 return subparsers.add_parser(
16 'codes',
17 help=headline,
18 description=headline)
21def setup(parser):
22 parser.add_squirrel_selection_arguments()
23 parser.add_squirrel_query_arguments(without=['time', 'tmin', 'tmax'])
26def run(parser, args):
27 from pyrocko import squirrel as sq
29 squirrel = args.make_squirrel()
31 kwargs = args.squirrel_query
32 kinds = kwargs.pop('kind', sq.supported_content_kinds())
33 codes_query = kwargs.pop('codes', None)
35 for kind in kinds:
36 for kind_id, codes, deltat, _, count in sorted(
37 squirrel._iter_codes_info(kind=kind, codes=codes_query)):
38 print(to_kind(kind_id), codes, deltat, count)