Fix issue where parent is module level

* don't repeat print elements
* catch valueerror for linenos
* catch keyboard interupt

baroness/search.py

@@ -2,9 +2,9 @@
 
 
 from __future__ import print_function
-from itertools import repeat
 from multiprocessing import Pool
 import os
+import signal
 import sys
 
 from redbaron import RedBaron
@@ -29,8 +29,9 @@ def search(
     options = dict(locals())
     options.pop('files')
 
+    LOGGER.debug('DEBUG Using options: %s', options)
     LOGGER.debug('DEBUG Using %s processes', processes)
-    pool = Pool(processes)
+    pool = Pool(processes, _init_worker)
     tasks = ((filename, pattern, options) for filename in filenames(files))
     try:
         for result in pool.imap_unordered(_safe_search_file, tasks):
@@ -48,6 +49,11 @@ def search(
         pool.join()
 
 
+def _init_worker():
+    """Ensure that KeyboardInterrupt is ignored."""
+    signal.signal(signal.SIGINT, signal.SIG_IGN)
+
+
 def _safe_search_file(args):
     """A wrapper for multiprocessing around _search_file."""
     try:
@@ -74,12 +80,26 @@ def _search_file(filename, pattern, options):
 
         results = _search(root)
         output = []
+        seen = set()
         if results:
             output.append(filename)
             for result in results:
+                # Get the correct number of parents
                 for _ in range(options['parents']):
                     result = result.parent if result.parent else result
-                output.append(format_node(result, no_color=options['no_color'], no_linenos=options['no_linenos']))
+
+                # Ensure that we don't print the same code twice
+                if result in seen:
+                    continue
+                else:
+                    seen.add(result)
+
+                # format the output
+                output.append(format_node(
+                    result,
+                    no_color=options['no_color'],
+                    no_linenos=options['no_linenos']
+                ))
                 output.append(u'--')
             output.append(u'')
         return (filename, u'\n'.join(output))

baroness/utils.py

@@ -97,6 +97,8 @@ def format_lineno(lineno):
             linenos = count(node.absolute_bounding_box.top_left.line)
         except AttributeError:
             linenos = repeat(u'**')
+        except ValueError:
+            linenos = count()
         lines = [
             u'{}-{}'.format(format_lineno(lineno).strip(), line)
             for lineno, line in zip(linenos, lines)