From f3f4a772d4f1689e30d0cfdbe3072f0975231de4 Mon Sep 17 00:00:00 2001 From: Mohit Agrawal Date: Tue, 30 Oct 2018 08:34:48 +0530 Subject: [PATCH 1/2] update depth based on the directory being indexed --- blackarrow/blackarrow.py | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/blackarrow/blackarrow.py b/blackarrow/blackarrow.py index 13cbbe0..a5e8a79 100755 --- a/blackarrow/blackarrow.py +++ b/blackarrow/blackarrow.py @@ -61,13 +61,15 @@ def start_search(args: argparse.Namespace): final_queue = mp.Queue() # Use final queue for external output processes = [] - indexer = mp.Process( - name="indexer", - target=index_worker, - args=(args.directories, ignore_re, numworkers, search_queue, output, args.depth), - ) - indexer.start() - processes.append(indexer) + for i, directory in enumerate(args.directories): + depth = None if args.depth is None else args.depth + directory.count(os.sep) + indexer = mp.Process( + name="indexer-{}".format(i+1), + target=index_worker, + args=([directory], ignore_re, numworkers, search_queue, output, depth), + ) + indexer.start() + processes.append(indexer) for i in range(numworkers): worker = mp.Process( From 92aed094179fd1ade164e4f64417956bbc68293e Mon Sep 17 00:00:00 2001 From: Mohit Agrawal Date: Tue, 30 Oct 2018 09:28:19 +0530 Subject: [PATCH 2/2] formatting --- blackarrow/blackarrow.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/blackarrow/blackarrow.py b/blackarrow/blackarrow.py index c328b02..ae55beb 100755 --- a/blackarrow/blackarrow.py +++ b/blackarrow/blackarrow.py @@ -66,7 +66,14 @@ def start_search(args: argparse.Namespace): indexer = mp.Process( name="indexer-{}".format(i+1), target=index_worker, - args=([directory], ignore_re, filename_re, numworkers, search_queue, depth), + args=( + [directory], + ignore_re, + filename_re, + numworkers, + search_queue, + depth, + ), ) indexer.start() processes.append(indexer)