B  ‹d%ã@s¦dZddlZddlZddlZddlZddlmZGdd„deƒZddd„Zdd d „Z d d „Z e  ¡e _ e  ¡e _d d„Zejddkrdd„Zndd„Zddd„ZdS)zSupport for Future chaining.éNé)ÚFuturec@s$eZdZdZddd„Zddd„ZdS) ÚThenableFuturez*Thenable* `Future` subclass.NcCs t|||ƒS)z.Return ``then(self, on_success, on_failure)``.)Úthen)ÚselfÚ on_successÚ on_failure©r úa/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/mpi4py/futures/aplus.pyrszThenableFuture.thencCs t||ƒS)z#Return ``catch(self, on_failure)``.)Úcatch)rrr r r r szThenableFuture.catch)NN)N)Ú__name__Ú __module__Ú __qualname__Ú__doc__rr r r r r rs rcCs(| ¡}tjt|||d}| |¡|S)aJavaScript-like (`Promises/A+`_) support for Future chaining. Args: future: Input future instance. on_success (optional): Function to be called when the input future is successfully resolved. Once the input future is resolved, its result value is the input for `on_success`. If `on_success` is ``None``, the output future (i.e., the one returned by this function) will be resolved directly with the result of the input future. If `on_success` returns a future instance, the result future is chained to the output future. Otherwise, the result of `on_success` is used to resolve the output future. on_failure (optional): Function to be called when the input future is rejected. The Exception instance picked up from the rejected future is the input value for `on_failure`. If `on_failure` is ``None``, the output future (i.e., the one returned by this function) will be rejected directly with the exception of the input future. Returns: Output future to be resolved once in input future is resolved and either `on_success` or `on_failure` completes. .. _Promises/A+: https://promisesaplus.com/ )rr)Ú __class__Ú functoolsÚpartialÚ_done_cbÚadd_done_callback)ÚfuturerrÚ new_futureÚdone_cbr r r rs rcCs$|dkrt|ddd„ƒSt|d|ƒS)aJClose equivalent to ``then(future, None, on_failure)``. Args: future: Input future instance. on_failure (optional): Function to be called when the input future is rejected. If `on_failure` is ``None``, the output future will be resolved with ``None`` thus ignoring the exception. NcSsdS)Nr )Úexcr r r ÚLózcatch..)r)rrr r r r As r c Csttjdtj}y ||}Wn$tk r>t ¡}|||<YnX||kr\td |t|ƒ¡ƒ‚|  |¡WdQRXdS)NzOCircular future chain detected: Future {0} is already in the resolved chain {1}) Ú _chain_logÚlockÚregistryÚKeyErrorÚweakrefÚWeakSetÚ RuntimeErrorÚformatÚsetÚadd)rrrÚlogr r r rPs rcCs$t||ƒt t|¡}| |¡dS)N)rrrrr)rrrr r r Ú _chain_futureds  r&écCst ¡d}|S)Nr)ÚsysÚexc_info)rr r r Ú_sys_exceptionks r*cCst ¡d}d|_|S)Nr)r(r)Ú __traceback__)rr r r r*os c Csâ| ¡s| ¡dS| ¡r(| ¡dSy8| ¡}|r>||ƒ}t|tƒrTt||ƒn | |¡Wn|tk rÜt ƒ}|s†|  |¡nRy,||ƒ}t|tƒr¦|  |¡n | |¡Wn$tk rÖt ƒ}|  |¡YnXYnXdS)N) ÚdoneÚcancelÚ cancelledÚresultÚ isinstancerr&Ú set_resultÚ BaseExceptionr*Ú set_exception)rrrrr/Ú exceptionr r r rus2     r)NN)N)NN)rr(rÚ threadingrZ_corerrrr rÚLockrÚWeakKeyDictionaryrr&Ú version_infor*rr r r r Ús   #