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 

8headline = 'Search indexed contents.' 

9 

10 

11def make_subparser(subparsers): 

12 return subparsers.add_parser( 

13 'nuts', 

14 help=headline, 

15 description=headline) 

16 

17 

18def setup(parser): 

19 parser.add_squirrel_selection_arguments() 

20 parser.add_squirrel_query_arguments() 

21 

22 parser.add_argument( 

23 '--contents', 

24 action='store_true', 

25 dest='print_contents', 

26 default=False, 

27 help='Print contents.') 

28 

29 

30def run(parser, args): 

31 squirrel = args.make_squirrel() 

32 for nut in squirrel.iter_nuts(**args.squirrel_query): 

33 if args.print_contents: 

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

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

36 else: 

37 print(nut.summary)