1# http://pyrocko.org - GPLv3 

2# 

3# The Pyrocko Developers, 21st Century 

4# ---|P------/S----------~Lg---------- 

5 

6from __future__ import absolute_import, print_function 

7 

8from pyrocko.squirrel.model import to_kind 

9 

10 

11headline = 'Get summary of available data codes.' 

12 

13 

14def make_subparser(subparsers): 

15 return subparsers.add_parser( 

16 'codes', 

17 help=headline, 

18 description=headline) 

19 

20 

21def setup(parser): 

22 parser.add_squirrel_selection_arguments() 

23 parser.add_squirrel_query_arguments(without=['time', 'tmin', 'tmax']) 

24 

25 

26def run(parser, args): 

27 from pyrocko import squirrel as sq 

28 

29 squirrel = args.make_squirrel() 

30 

31 kwargs = args.squirrel_query 

32 kinds = kwargs.pop('kind', sq.supported_content_kinds()) 

33 codes_query = kwargs.pop('codes', None) 

34 

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)