o {j6h@s|dZddlmZddlmZddlmZddlmZ ddl m Z dd giZ d d gZ edd dZddZdd Zdd ZdS)z, Thin wrappers around `concurrent.futures`. )contextmanager) length_hint) cpu_count)tqdm) TqdmWarningz github.com/ casperdcl thread_map process_mapccsTt|dd}|p |}t|||}|||V|dur#|`dS||dS)z>get (create if necessary) and then restore `tqdm_class`'s lock_lockN)getattrget_lockset_lockr ) tqdm_class lock_nameold_locklockrf/var/www/html/chefvision.cloud.itp360.com/venv/lib/python3.10/site-packages/tqdm/contrib/concurrent.py ensure_locks    rc Os|}d|vrt|d|d<|dt}|dtdtd}|dd}|d d }t||d =} |||j| fd #} t|| j |g|Rd|ifi|Wd Wd S1sdwYWd d S1stwYd S)a Implementation of `thread_map` and `process_map`. Parameters ---------- tqdm_class : [default: tqdm.auto.tqdm]. max_workers : [default: min(32, cpu_count() + 4)]. chunksize : [default: 1]. lock_name : [default: "":str]. totalrr max_workers  chunksizerr )r)r initializerinitargsN) copyrpop tqdm_autominrrrlistmap) PoolExecutorfn iterables tqdm_kwargskwargsrrrrlkexrrr _executor_maps"    &"r,cOs$ddlm}t||g|Ri|S)a Equivalent of `list(map(fn, *iterables))` driven by `concurrent.futures.ThreadPoolExecutor`. Parameters ---------- tqdm_class : optional `tqdm` class to use for bars [default: tqdm.auto.tqdm]. max_workers : int, optional Maximum number of workers to spawn; passed to `concurrent.futures.ThreadPoolExecutor.__init__`. [default: max(32, cpu_count() + 4)]. r)ThreadPoolExecutor)concurrent.futuresr-r,)r&r'r(r-rrrr 6s cOs|ddlm}|r&d|vr&ttt|}|dkr&ddlm}|d|tddd |vr2|}d |d <t ||g|Ri|S) a Equivalent of `list(map(fn, *iterables))` driven by `concurrent.futures.ProcessPoolExecutor`. Parameters ---------- tqdm_class : optional `tqdm` class to use for bars [default: tqdm.auto.tqdm]. max_workers : int, optional Maximum number of workers to spawn; passed to `concurrent.futures.ProcessPoolExecutor.__init__`. [default: min(32, cpu_count() + 4)]. chunksize : int, optional Size of chunks sent to worker processes; passed to `concurrent.futures.ProcessPoolExecutor.map`. [default: 1]. lock_name : str, optional Member of `tqdm_class.get_lock()` to use [default: mp_lock]. r)ProcessPoolExecutorri)warnzIterable length %d > 1000 but `chunksize` is not set. This may seriously degrade multiprocess performance. Set `chunksize=1` or more.r) stacklevelrmp_lock) r.r/maxr$rwarningsr0rrr,)r&r'r(r/longest_iterable_lenr0rrrr Hs   N)r )__doc__ contextlibroperatorrosrautorr!stdr __author____all__rr,r r rrrrs