ó P'—^c@@s dZddlmZmZddlZddlZddlZddlZddlZddl m Z ddl m Z ddl mZmZddd d d d gZejjd ƒZejddkZe d„ƒZdejfd„ƒYZyddlmZWn™ek r¢yddlmZWq£ek ržyddlmZWqŸek ršyddlmZWq›ek r–ddl mZq›XqŸXq£XnXd„Z!de"fd„ƒYZ#de"fd„ƒYZ$d e"fd„ƒYZ%dej&fd„ƒYZ'ere'Z(ne$Z(dS(sN celery.utils.threads ~~~~~~~~~~~~~~~~~~~~ Threading utilities. i(tabsolute_importtprint_functionN(tcontextmanager(tProxy(tTHREAD_TIMEOUT_MAXtitemstbgThreadtLocalt LocalStackt LocalManagert get_identtdefault_socket_timeouttUSE_FAST_LOCALSicc@s/tjƒ}tj|ƒdVtj|ƒdS(N(tsockettgetdefaulttimeouttsetdefaulttimeout(ttimeouttprev((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyR s  cB@sAeZdd„Zd„Zd„Zd„Zd„Zd„ZRS(cK@sStt|ƒjƒtjƒ|_tjƒ|_t|_|pI|j j |_ dS(N( tsuperRt__init__t threadingtEventt _is_shutdownt _is_stoppedtTruetdaemont __class__t__name__tname(tselfRtkwargs((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyR's  cC@stdƒ‚dS(Nssubclass responsibility(tNotImplementedError(R((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pytbody.scO@s_t|j|Œdtjƒtjƒ}z,tj|d|d|ddtjƒWd~XdS(Ntfileiii(tprinttformattsyststderrtexc_infot tracebacktprint_exceptiontNone(RtmsgtfmtRR&((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyton_crash1s  cC@s‘|j}|jj}zjxc|ƒs}y |ƒWqtk ry}z$|jd|j|ƒ|jƒWdtjdƒXqXqWWd|jƒXdS(Ns{0!r} crashed: {1!r}i( R Rtis_sett ExceptionR,Rt _set_stoppedtost_exit(RR t shutdown_settexc((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pytrun:s    cC@s)y|jjƒWntk r$nXdS(N(Rtsett TypeError(R((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyR/Js cC@s:|jjƒ|jjƒ|jƒr6|jtƒndS(sGraceful shutdown.N(RR5Rtwaittis_alivetjoinR(R((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pytstopRs   N( Rt __module__R)RR R,R4R/R:(((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyR%s     (t getcurrent(R cC@s|jƒdS(sMReleases the contents of the local for the current context. This makes it possible to use locals without a manager. Example:: >>> loc = Local() >>> loc.foo = 42 >>> release_local(loc) >>> hasattr(loc, 'foo') False With this function one can release :class:`Local` objects as well as :class:`StackLocal` objects. However it is not possible to release data held by proxies that way, one always has to retain a reference to the underlying local object in order to be able to release it. .. versionadded:: 0.6.1 N(t__release_local__(tlocal((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyt release_localhscB@sMeZd Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z RS( t __storage__t__ident_func__cC@s*tj|diƒtj|dtƒdS(NR@RA(tobjectt __setattr__R (R((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyR‚scC@stt|jƒƒS(N(titerRR@(R((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyt__iter__†scC@s t||ƒS(sCreate a proxy for a name.(R(Rtproxy((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyt__call__‰scC@s|jj|jƒdƒdS(N(R@tpopRAR)(R((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyR=scC@s=y|j|jƒ|SWntk r8t|ƒ‚nXdS(N(R@RAtKeyErrortAttributeError(RR((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyt __getattr__s cC@sP|jƒ}|j}y||||>> ls = LocalStack() >>> ls.push(42) >>> ls.top 42 >>> ls.push(23) >>> ls.top 23 >>> ls.pop() 23 >>> ls.top 42 They can be force released by using a :class:`LocalManager` or with the :func:`release_local` function but the correct way is to pop the item from the stack after using. When the stack is empty it will no longer be bound to the current context (and as such released). By calling the stack without arguments it will return a proxy that resolves to the topmost item on the stack. cC@stƒ|_dS(N(Rt_local(R((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyR¿scC@s|jjƒdS(N(RRR=(R((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyR=ÂscC@s |jjS(N(RRRA(R((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyt_get__ident_func__ÅscC@stj|jd|ƒdS(NRA(RBRCRR(RRL((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyt_set__ident_func__Èsc@s‡fd†}t|ƒS(Nc@s(ˆj}|dkr$tdƒ‚n|S(Nsobject unbound(ttopR)t RuntimeError(trv(R(s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyt_lookupÎs  (R(RRX((Rs6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyRGÍscC@sEt|jddƒ}|dkr4g|j_}n|j|ƒ|S(sPushes a new item to the stacktstackN(tgetattrRRR)RYtappend(RtobjRW((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pytpushÕs   cC@sZt|jddƒ}|dkr%dSt|ƒdkrLt|jƒ|dS|jƒSdS(s|Remove the topmost item from the stack, will return the old value or `None` if the stack was already empty. RYiiÿÿÿÿN(RZRRR)tlenR?RH(RRY((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyRHÝs  cC@s)t|jddƒ}|r%t|ƒSdS(NRYi(RZRRR)R^(RRY((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyt__len__êscC@s)t|jddƒ}|dk r%|SgS(sdget_current_worker_task uses this to find the original task that was executed by the worker.RYN(RZRRR)(RRY((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyRYîs cC@s1y|jjdSWnttfk r,dSXdS(s[The topmost item on the stack. If the stack is empty, `None` is returned. iÿÿÿÿN(RRRYRJt IndexErrorR)(R((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyRU÷s(RR;t__doc__RR=RSRTtpropertyRARGR]RHR_RYRU(((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyRQ¥s        cB@s8eZdZddd„Zd„Zd„Zd„ZRS(sœLocal objects cannot manage themselves. For that you need a local manager. You can pass a local manager multiple locals or add them later by appending them to `manager.locals`. Everytime the manager cleans up it, will clean up all the data left in the locals for this context. The `ident_func` parameter can be added to override the default ident function for the wrapped locals. cC@s‘|dkrg|_n-t|tƒr6|g|_nt|ƒ|_|dk r„||_x0|jD]}tj|d|ƒqdWn t|_dS(NRA( R)tlocalst isinstanceRtlistt ident_funcRBRCR (RRcRfR>((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyRs    cC@s |jƒS(sReturn the context identifier the local objects use internally for this context. You cannot override this method to change the behavior but use it to link other context local objects (such as SQLAlchemy's scoped sessions) to the Werkzeug locals.(Rf(R((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyR scC@s"x|jD]}t|ƒq WdS(sManually clean up the data in the locals for this context. Call this at the end of the request or use `make_middleware()`. N(RcR?(RR>((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pytcleanup#scC@sdj|jjt|jƒƒS(Ns<{0} storages: {1}>(R#RRR^Rc(R((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyt__repr__,sN(RR;RaR)RR RgRh(((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyR s    t_FastLocalStackcB@s)eZd„Zed„ƒZd„ZRS(cC@s+g|_|jj|_|jj|_dS(N(RYR[R]RH(R((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyR3s cC@s.y|jdSWnttfk r)dSXdS(Niÿÿÿÿ(RYRJR`R)(R((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyRU8scC@s t|jƒS(N(R^RY(R((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyR_?s(RR;RRbRUR_(((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pyRi1s ()Rat __future__RRR0R R$RR't contextlibRt celery.localRt celery.fiveRRt__all__tenvirontgetR t version_infotPY3R tThreadRtgreenletR<R t ImportErrort_threadtthreadt _dummy_threadt dummy_threadR?RBRRQR R>RiR(((s6/tmp/pip-unpacked-wheel-gV1wwp/celery/utils/threads.pytsH       4      &]/