4j%ddlZddlZddlZddlZddlmZmZmZmZm Z m Z ddl m Z m Z ejZGddZGddZGdd eZGd d eZe eeeefZy) N)AnyCallableDictListOptionalUnion) EXIT_FUNC INIT_FUNCceZdZdZ ddedeedeedeededee d dfd Z d efd Z d efd Z ddee d dfd Z ddee d efdZdeded dfdZy) AsyncResultz4 Adapted from ``multiprocessing.pool.ApplyResult``. Ncachecallbackerror_callbackjob_iddelete_from_cachetimeoutreturncH||_||_||_||_||_|t t n||_tj|_ d|_ d|_ |j|jvrtd|d||j|j<y)a :param cache: Cache for storing intermediate results :param callback: Callback function to call when the task is finished. The callback function receives the output of the function as its argument :param error_callback: Callback function to call when the task has failed. The callback function receives the exception as its argument :param job_id: Job ID of the task. If None, a new job ID is generated :param delete_from_cache: If True, the result is deleted from the cache when the task is finished :param timeout: Timeout in seconds for a single task. When the timeout is exceeded, MPIRE will raise a ``TimeoutError``. Use ``None`` to disable (default) NJob ID  already exists in cache)_cache _callback_error_callback_delete_from_cache_timeoutnext job_counterr threadingEvent _ready_event_success_value ValueError)selfr rrrrrs E/home/agent/.local/lib/python3.12/site-packages/mpire/async_result.py__init__zAsyncResult.__init__s !-"3 +1>d;'v %OO-  ;;$++ %wvh.FGH H#' DKK c6|jjS)z? :return: Returns True if the task is finished )r is_setr$s r%readyzAsyncResult.ready+s  ''))r'cj|jst|jd|jS)z :return: Returns True if the task has finished successfully :raises: ValueError if the task is not finished yet z is not ready)r+r#rr!r*s r% successfulzAsyncResult.successful1s- zz| }M:; ;}}r'c:|jj|y)zy Wait until the task is finished :param timeout: Timeout in seconds. If None, wait indefinitely N)r waitr$rs r%r/zAsyncResult.wait:s w'r'c|j||jst|jr |jS|j)am Wait until the task is finished and return the output of the function :param timeout: Timeout in seconds. If None, wait indefinitely :return: Output of the function :raises: TimeoutError if the task is not finished within the timeout. When the task has failed, the exception raised by the function is re-raised )r/r+ TimeoutErrorr!r"r0s r%getzAsyncResult.getBs: 'zz|  ==;; ++ r'successresultch||_||_|jr'|jr|j|j|jr'|js|j|j|jj |j r|j|j=yy)aw Set the result of the task and call any callbacks, when provided. This also removes the task from the cache, as it's no longer needed there. The user should store a reference to the result object :param success: True if the task has finished successfully :param result: Output of the function or the exception raised by the function N) r!r"rrr setrrrr$r4r5s r%_setzAsyncResult._setSs   >>dmm NN4;; '      -   " " DKK( #r')NTNN)__name__ __module__ __qualname____doc__rrrintboolfloatr&r+r-r/rr3r9r'r%r r s>ko(d(hx.@(RZ[cRd(!#(BF(X`afXg(sw(6*t* D(HUO(t(8E?c")D)#)$)r'r c eZdZdZ ddedeedeedeeddf dZdd Z dd e deede fd Z e Z dd Zd e de ddfdZdeddfdZdefdZddZy)UnorderedAsyncResultIteratorza Stores results of a task and provides an iterator to obtain the results in an unordered fashion Nr n_tasksrrrc||_d|_||_|ttn||_t j|_tjtj|_ d|_ d|_d|_tj |_|j |jvrt%d|d||j|j <||j'|yy)a :param cache: Cache for storing intermediate results :param n_tasks: Number of tasks that will be executed. If None, we don't know the lenght yet :param job_id: Job ID of the task. If None, a new job ID is generated :param timeout: Timeout in seconds for a single task. When the timeout is exceeded, MPIRE will raise a ``TimeoutError``. Use ``None`` to disable (default) N)lockrrr)r_n_tasksrrrr collectionsdeque_itemsr ConditionLock _condition _n_received _n_returned _exceptionr_got_exceptionr# set_length)r$r rErrs r%r&z%UnorderedAsyncResultIterator.__init__ms   +1>d;'v !'') #--9>>3CD'oo/ ;;$++ %wvh.FGH H#' DKK   OOG $ r'c|Sr:rBr*s r%__iter__z%UnorderedAsyncResultIterator.__iter__s r'blockcz|jr/|xjdz c_|jjS|j|j|jk(rt|st j |j5|jsg|jj| }|rt j |j|j|jk(rt|jsg|xjdz c_|jjcdddS#1swYyxYw)a2 Obtain the next unordered result for the task :param block: If True, wait until the next result is available. If False, raise queue.Empty if no result is available :param timeout: Timeout in seconds. If None, wait indefinitely :return: The next result N)r) rKrPpopleftrH StopIterationqueueEmptyrNr/)r$rVr timed_outs r%rz!UnorderedAsyncResultIterator.nexts ;;    ! ;;&&( ( == $)9)9T]])J ++ __ )kk $ 4 4W 4 EE ++%==,1A1AT]]1R'' kk    ! ;;&&( ) ) )sA3D19.D11D:c|j5|j|j|jkrA|jj|j'|j|jkrAdddy#1swYyxYw)z6 Wait until all results are available N)rNrHrOr/r*s r%r/z!UnorderedAsyncResultIterator.waitsh__ '--'4+;+;dmm+K$$&--'4+;+;dmm+K ' ' 'sA A=A==Br4r5c"|r`|xjdz c_|jj||j5|jj dddy||_|j jy#1swYyxYw)z Set the result of the task :param success: True if the task has finished successfully :param result: Output of the function or the exception raised by the function rXN)rOrKappendrNnotifyrQrRr7r8s r%r9z!UnorderedAsyncResultIterator._setsv     !  KK  v & )&&( ) )%DO    # # %  ) )s BBlengthc|j+|j|k7rtd|jd|y|j5||_|jjdddy#1swYyxYw)z_ Set the length of the iterator :param length: Length of the iterator Nz+Length of iterator has already been set to z, but is now set to )rHr#rNra)r$rbs r%rSz'UnorderedAsyncResultIterator.set_lengths| == $}}& #Nt}}o^66rrr?rAr&rUr@rr__next__r/r9rS ExceptionrfrhrBr'r%rDrDiskTX,0%d%Xc]%HSM%"5/%59%6)$))3)>H'&D&#&$&"%%%"y%r'rDc@eZdZdededdffd ZdefdZddZxZ S) AsyncResultWithExceptionGetterr rrNc0t||dd|ddy)NF)rrrrr)superr&)r$r r __class__s r%r&z'AsyncResultWithExceptionGetter.__init__s# d6ej!%  'r'c:|j|jSre)r/r"r*s r%rfz,AsyncResultWithExceptionGetter.get_exceptions {{r'cTd|_d|_|jjy)) Reset the result object N)r!r"r clearr*s r%resetz$AsyncResultWithExceptionGetter.resets$   !r'ri) r;r<r=rr?r&rkrfru __classcell__rps@r%rmrms.'d'C'D'y"r'rmcBeZdZdeddffd ZdeefdZddZxZ S) UnorderedAsyncExitResultIteratorr rNc4t||dtdy)N)rErr)ror&r )r$r rps r%r&z)UnorderedAsyncExitResultIterator.__init__s YMr'c,t|jS)z/ :return: List of exit results )listrKr*s r% get_resultsz,UnorderedAsyncExitResultIterator.get_resultssDKK  r'cd|_|jjd|_d|_d|_|j jy)rsNr)rHrKrtrOrPrQrRr*s r%ruz&UnorderedAsyncExitResultIterator.resetsD   !!#r'ri) r;r<r=rr&rrr}rurvrws@r%ryrys.NdNtN!T#Y! $r'ry)rI itertoolsr[rtypingrrrrrr mpire.commsr r countrr rDrmryAsyncResultTyperBr'r%rsw ==,ioo Z)Z)zx%x%v"[",$'C$. %CEa89:r'