1# http://pyrocko.org - GPLv3 

2# 

3# The Pyrocko Developers, 21st Century 

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

5 

6from pyrocko.squirrel.model import to_kind 

7 

8 

9headline = 'Get summary of available station/channel codes.' 

10 

11 

12def make_subparser(subparsers): 

13 return subparsers.add_parser( 

14 'codes', 

15 help=headline, 

16 description=headline) 

17 

18 

19def setup(parser): 

20 parser.add_squirrel_selection_arguments() 

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

22 

23 

24def run(parser, args): 

25 from pyrocko import squirrel as sq 

26 

27 squirrel = args.make_squirrel() 

28 

29 kwargs = args.squirrel_query 

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

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

32 

33 for kind in kinds: 

34 for kind_id, codes, deltat, _, count in sorted( 

35 squirrel._iter_codes_info(kind=kind, codes=codes_query)): 

36 print(to_kind(kind_id), codes, deltat, count)