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 .. import common 

9 

10 

11def setup_subcommand(subparsers): 

12 return common.add_parser( 

13 subparsers, 'nuts', 

14 help='Search indexed contents.') 

15 

16 

17def setup(parser): 

18 common.add_selection_arguments(parser) 

19 common.add_query_arguments(parser) 

20 

21 parser.add_argument( 

22 '--contents', 

23 action='store_true', 

24 dest='print_contents', 

25 default=False, 

26 help='Print contents.') 

27 

28 

29def call(parser, args): 

30 d = common.squirrel_query_from_arguments(args) 

31 squirrel = common.squirrel_from_selection_arguments(args) 

32 for nut in squirrel.iter_nuts(**d): 

33 if args.print_contents: 

34 print('# %s' % nut.summary) 

35 print(squirrel.get_content(nut).dump()) 

36 else: 

37 print(nut.summary)