j5jdZddlmZddlZddlmZmZddlmZm Z ddl m Z m Z ddl mZmZe dZdd Zdd ZGd d eZdd ZGddeZy)z"OCRmyPDF concurrency abstractions.) annotationsN)ABCabstractmethod)CallableIterable)AnyTypeVar)NullProgressBar ProgressBarTcyN)_args_kwargss J/var/www/html/qr/venv/lib/python3.12/site-packages/ocrmypdf/_concurrent.py _task_nooprs c$|jyr)update)_resultpbars r_task_finished_nooprs KKMrceZdZdZej ZeZdddZ ddddd ddZ e d dZ y) ExecutorzAbstract concurrent executor.N pbar_classc|r||_yyrr)selfrs r__init__zExecutor.__init__ s (DO r)worker_initializertasktask_arguments task_finishedc |sy|st}|st}|st}|j5|j|||||||dddy#1swYyxYw)aSet up parallel execution and progress reporting. Args: use_threads: If ``False``, the workload is the sort that will benefit from running in a multiprocessing context (for example, it uses Python heavily, and parallelizing it with threads is not expected to be performant). max_workers: The maximum number of workers that should be run. progress_kwargs: Arguments to set up the progress bar. worker_initializer: Called when a worker is initialized, in the worker's execution context. If the child workers are processes, it must be possible to marshall/pickle the worker initializer. ``functools.partial`` can be used to bind parameters. task: Called when the worker starts a new task, in the worker's execution context. Must be possible to marshall to the worker. task_finished: Called when a worker finishes a task, in the parent's context. task_arguments: An iterable that generates a group of parameters for each task. This runs in the parent's context, but the parameters must be marshallable to the worker. N) use_threads max_workersprogress_kwargsr!r"r#r$)rr pool_lock_executerr&r'r(r!r"r#r$s r__call__zExecutor.__call__$sf@ !!+ /MD ^^  MM'' /#5-+     s A  Acy)z-Custom executors should override this method.Nrr+s rr*zExecutor._executeXsr)r&boolr'intr(dictr!zCallable | Noner"zCallable[..., T] | Noner#zIterable | Noner$z'Callable[[T, ProgressBar], None] | NonereturnNoner&r.r'r/r(r0r!rr"rr#rr$r) __name__ __module__ __qualname____doc__ threadingLockr)r rr r,rr*rrrrrs'  I J%))/3(,*.AE22 2  2 , 2&2(2?2 2h < < <  < % < <! <  < rJsP)"#.> CL J