diff options
| author | Adam Janovsky | 2021-05-19 18:20:53 +0200 |
|---|---|---|
| committer | Adam Janovsky | 2021-05-19 18:20:53 +0200 |
| commit | a6b5a8bea7677cc7c70f8b77571ebac5c7718bcd (patch) | |
| tree | 6e016f84636f5b23dfd1676acca5c990fc2423b0 /sec_certs/parallel_processing.py | |
| parent | 4e479828a7c683b9b8301db0335384a848391c47 (diff) | |
| download | sec-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.py | 7 |
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) |
