o e@s|dZddlmZddlZddlmZmZddlmZm Z ddZ Gdd d Z Gd d d eZ dd dZ Gddde ZdS)z"OCRmyPDF concurrency abstractions.) annotationsN)ABCabstractmethod)CallableIterablecOdSN)_args_kwargsr r DD:\Projects\ConvertPro\env\Lib\site-packages\ocrmypdf/_concurrent.py _task_noop r c@s2eZdZdZddZddZddZd d d ZdS) NullProgressBarz'Progress bar API that takes no actions.cKrrr )selfkwargsr r r __init__rzNullProgressBar.__init__cCs|Srr )rr r r __enter__rzNullProgressBar.__enter__cCdS)NFr )rexc_type exc_value tracebackr r r __exit__rzNullProgressBar.__exit__NcCrrr )rZ_argr r r updaterzNullProgressBar.updater)__name__ __module__ __qualname____doc__rrrrr r r r rs rc@sNeZdZdZeZeZddddZ ddddddddZ e dddZ dS)ExecutorzAbstract concurrent executor.N pbar_classcCs|r||_dSdSrr)rr r r r r's zExecutor.__init__)worker_initializertasktask_arguments task_finished use_threadsbool max_workersint tqdm_kwargsdictr!Callable | Noner"r#Iterable | Noner$returnNonec Csd|sdS|st}|s t}|st}|j|j|||||||dWddS1s+wYdS)a Set 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. tdqm_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)r%r'r)r!r"r#r$)r pool_lock_executerr%r'r)r!r"r#r$r r r __call__+s&""zExecutor.__call__rrcCr)z-Custom executors should override this method.Nr r1r r r r0aszExecutor._execute)r%r&r'r(r)r*r!r+r"r+r#r,r$r+r-r.r%r&r'r(r)r*r!rr"rr#rr$r) rrrr threadingLockr/rr rr2rr0r r r r r!s 6rr-cCs|j}|jj|dS)N)Zprogressbar_class)hookZget_progressbar_classZ get_executor)Zplugin_managerr r r r setup_executorps r7c@seZdZdZdddZdS)SerialExecutorzImplements a purely sequential executor using the parallel protocol. The current process/thread will be the worker that executes all tasks in order. As such, ``worker_initializer`` will never be called. r%r&r'r(r)r*r!rr"r#rr$c CsR|jdi|}|D] } || } || |q WddS1s"wYdS)Nr r) rr%r'r)r!r"r#r$Zpbarargsresultr r r r0|s   "zSerialExecutor._executeNr3)rrrrr0r r r r r8usr8)r-r)r __future__rr4abcrrtypingrrr rrr7r8r r r r s  O