B ÛT•\µã@sjddlmZmZmZddlmZddgZGdd„deƒZdd„Z edd d „ƒZ d d „Z Gd d„deƒZ dS)é)Úabsolute_importÚdivisionÚprint_function)ÚcontextmanagerÚCallbackÚ add_callbacksc@sLeZdZdZeƒZddd„Zedd„ƒZdd„Z d d „Z d d „Z d d„Z dS)rac Base class for using the callback mechanism Create a callback with functions of the following signatures: >>> def start(dsk): ... pass >>> def start_state(dsk, state): ... pass >>> def pretask(key, dsk, state): ... pass >>> def posttask(key, result, dsk, state, worker_id): ... pass >>> def finish(dsk, state, failed): ... pass You may then construct a callback object with any number of them >>> cb = Callback(pretask=pretask, finish=finish) # doctest: +SKIP And use it either as a context manager over a compute/get call >>> with cb: # doctest: +SKIP ... x.compute() # doctest: +SKIP Or globally with the ``register`` method >>> cb.register() # doctest: +SKIP >>> cb.unregister() # doctest: +SKIP Alternatively subclass the ``Callback`` class with your own methods. >>> class PrintKeys(Callback): ... def _pretask(self, key, dask, state): ... print("Computing: {0}!".format(repr(key))) >>> with PrintKeys(): # doctest: +SKIP ... x.compute() # doctest: +SKIP NcCs6|r ||_|r||_|r||_|r(||_|r2||_dS)N)Ú_startÚ _start_stateÚ_pretaskÚ _posttaskÚ_finish)ÚselfÚstartZ start_stateZpretaskZposttaskZfinish©rú-lib/python3.7/site-packages/dask/callbacks.pyÚ__init__1szCallback.__init__cs$dddddg}t‡fdd„|DƒƒS)Nrr r r r c3s|]}tˆ|dƒVqdS)N)Úgetattr)Ú.0Úi)r rrú @sz%Callback._callback..)Útuple)r Zfieldsr)r rÚ _callback=szCallback._callbackcCst|ƒ|_|j ¡|S)N)rÚ_cmÚ __enter__)r rrrrBs  zCallback.__enter__cGs|jj|ŽdS)N)rÚ__exit__)r ÚargsrrrrGszCallback.__exit__cCstj |j¡dS)N)rÚactiveÚaddr)r rrrÚregisterJszCallback.registercCstj |j¡dS)N)rrÚremover)r rrrÚ unregisterMszCallback.unregister)NNNNN) Ú__name__Ú __module__Ú __qualname__Ú__doc__ÚsetrrÚpropertyrrrrr rrrrrs& cCs(|rdd„t|ŽDƒSdddddgSdS)z>Take an iterable of callbacks, return a list of each callback.cSsg|]}dd„|Dƒ‘qS)cSsg|] }|r|‘qSrr)rrrrrú Tsz/unpack_callbacks...r)rÚfrrrr'Tsz$unpack_callbacks..rN)Úzip)ZcbsrrrÚunpack_callbacksQsr*Nccs<|dk}|rtjtƒ}t_z|p$dVWd|r6|t_XdS)zÌAllows callbacks to work with nested schedulers. Callbacks will only be used by the first started scheduler they encounter. This means that only the outermost scheduler will use global callbacks.Nr)rrr%)Ú callbacksZglobal_callbacksrrrÚlocal_callbacksYsr,cCs*t|tƒr|jSt|tƒr|Stdƒ‚dS)z Normalizes a callback to a tuplez.Callbacks must be either `Callback` or `tuple`N)Ú isinstancerrrÚ TypeError)ÚcbrrrÚnormalize_callbackis   r0c@s(eZdZdZdd„Zdd„Zdd„ZdS) raÇContext manager for callbacks. Takes several callbacks and applies them only in the enclosed context. Callbacks can either be represented as a ``Callback`` object, or as a tuple of length 4. Examples -------- >>> def pretask(key, dsk, state): ... print("Now running {0}").format(key) >>> callbacks = (None, pretask, None, None) >>> with add_callbacks(callbacks): # doctest: +SKIP ... res.compute() cGs"dd„|Dƒ|_tj |j¡dS)NcSsg|] }t|ƒ‘qSr)r0)rÚcrrrr'ƒsz*add_callbacks.__init__..)r+rrÚupdate)r r+rrrr‚szadd_callbacks.__init__cCsdS)Nr)r rrrr†szadd_callbacks.__enter__cCs x|jD]}tj |¡qWdS)N)r+rrÚdiscard)r ÚtypeÚvalueÚ tracebackr1rrrr‰s zadd_callbacks.__exit__N)r!r"r#r$rrrrrrrrss)N) Z __future__rrrÚ contextlibrÚ__all__Úobjectrr*r,r0rrrrrÚs I