ó Ĥ–Ġ\c@`sŒddlmZmZmZddlmZddgZdefd„ƒYZd„Z ed d„ƒZ d„Z defd „ƒYZ d S( i(tabsolute_importtdivisiontprint_function(tcontextmanagertCallbackt add_callbackscB`sbeZdZeƒZdddddd„Zed„ƒZd„Z d„Z d„Z d„Z RS(sc 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 cC`s^|r||_n|r$||_n|r6||_n|rH||_n|rZ||_ndS(N(t_startt _start_statet_pretaskt _posttaskt_finish(tselftstartt start_statetpretasktposttasktfinish((s-lib/python2.7/site-packages/dask/callbacks.pyt__init__1s    c`s/dddddg}t‡fd†|DƒƒS(NRRRR R c3`s!|]}tˆ|dƒVqdS(N(tgetattrtNone(t.0ti(R (s-lib/python2.7/site-packages/dask/callbacks.pys @s(ttuple(R tfields((R s-lib/python2.7/site-packages/dask/callbacks.pyt _callback=scC`s t|ƒ|_|jjƒ|S(N(Rt_cmt __enter__(R ((s-lib/python2.7/site-packages/dask/callbacks.pyRBs cG`s|jj|ŒdS(N(Rt__exit__(R targs((s-lib/python2.7/site-packages/dask/callbacks.pyRGscC`stjj|jƒdS(N(RtactivetaddR(R ((s-lib/python2.7/site-packages/dask/callbacks.pytregisterJscC`stjj|jƒdS(N(RRtremoveR(R ((s-lib/python2.7/site-packages/dask/callbacks.pyt unregisterMsN( t__name__t __module__t__doc__tsetRRRtpropertyRRRRR!(((s-lib/python2.7/site-packages/dask/callbacks.pyRs&     cC`sS|r<gt|ŒD]%}g|D]}|r |^q ^qSdddddgSdS(s>Take an iterable of callbacks, return a list of each callback.N((((((tzip(tcbstfR((s-lib/python2.7/site-packages/dask/callbacks.pytunpack_callbacksQs6cc`sT|dk}|r+tjtƒ}t_nz|p7dVWd|rO|t_nXdS(sÌ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.N((RRRR%(t callbackstglobal_callbacks((s-lib/python2.7/site-packages/dask/callbacks.pytlocal_callbacksYs cC`s9t|tƒr|jSt|tƒr)|Stdƒ‚dS(s Normalizes a callback to a tuples.Callbacks must be either `Callback` or `tuple`N(t isinstanceRRRt TypeError(tcb((s-lib/python2.7/site-packages/dask/callbacks.pytnormalize_callbackis cB`s)eZdZd„Zd„Zd„ZRS(sÇ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() cG`s9g|D]}t|ƒ^q|_tjj|jƒdS(N(R1R+RRtupdate(R R+tc((s-lib/python2.7/site-packages/dask/callbacks.pyR‚s"cC`sdS(N((R ((s-lib/python2.7/site-packages/dask/callbacks.pyR†scC`s(x!|jD]}tjj|ƒq WdS(N(R+RRtdiscard(R ttypetvaluet tracebackR3((s-lib/python2.7/site-packages/dask/callbacks.pyR‰s(R"R#R$RRR(((s-lib/python2.7/site-packages/dask/callbacks.pyRss  N(t __future__RRRt contextlibRt__all__tobjectRR*RR-R1R(((s-lib/python2.7/site-packages/dask/callbacks.pyts I