aboutsummaryrefslogtreecommitdiffhomepage
path: root/sec_certs/parallel_processing.py
diff options
context:
space:
mode:
authorAdam Janovsky2021-05-19 18:20:53 +0200
committerAdam Janovsky2021-05-19 18:20:53 +0200
commita6b5a8bea7677cc7c70f8b77571ebac5c7718bcd (patch)
tree6e016f84636f5b23dfd1676acca5c990fc2423b0 /sec_certs/parallel_processing.py
parent4e479828a7c683b9b8301db0335384a848391c47 (diff)
downloadsec-certs-a6b5a8bea7677cc7c70f8b77571ebac5c7718bcd.tar.gz
sec-certs-a6b5a8bea7677cc7c70f8b77571ebac5c7718bcd.tar.zst
sec-certs-a6b5a8bea7677cc7c70f8b77571ebac5c7718bcd.zip
add progress bar description
Diffstat (limited to 'sec_certs/parallel_processing.py')
-rw-r--r--sec_certs/parallel_processing.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/sec_certs/parallel_processing.py b/sec_certs/parallel_processing.py
index b5519aab..20a82f32 100644
--- a/sec_certs/parallel_processing.py
+++ b/sec_certs/parallel_processing.py
@@ -5,7 +5,8 @@ import time
def process_parallel(func: Callable, items: Iterable, max_workers: int, callback: Optional[Callable] = None,
- use_threading: bool = True, progress_bar: bool = True, unpack: bool = False):
+ use_threading: bool = True, progress_bar: bool = True, unpack: bool = False,
+ progress_bar_desc: Optional[str] = None):
if use_threading is True:
pool = ThreadPool(max_workers)
else:
@@ -16,8 +17,8 @@ def process_parallel(func: Callable, items: Iterable, max_workers: int, callback
else:
results = [pool.apply_async(func, (*i, ), callback=callback) for i in items]
- if progress_bar is True:
- bar = tqdm(total=len(results))
+ if progress_bar is True and items:
+ bar = tqdm(total=len(results), desc=progress_bar_desc)
while not all([x.ready() for x in results]):
done_count = len([x.ready() for x in results if x.ready()])
bar.update(done_count - bar.n)