o Df)@sPdZddlmZddlZddlZddlZddlmZmZddl m Z m Z ddl m Z mZddlmZmZddlmZd d lmZd ZeZGd d d ZGdddeZddefddZGdddZddZe efe ffddZe effddZ d2ddZ!ddZ"d3d d!Z#d4d#d$Z$  % d5d'd(Z%d6d+d,Z&d7d.d/Z'd0d1Z(eZ)eZ*dS)8zFunctional Utilities.) annotationsN) OrderedDictUserDict)IterableMapping)countrepeat)sleeptime)wraps) safe_repr)LRUCachememoizelazymaybe_evaluateis_list maybe_list dictfilterretry_over_timec@s$eZdZddZddZddZdS)ChannelPromisecCs ||_dSN) __contract__)selfcontractrO/home/ubuntu/webapp/venv/lib/python3.10/site-packages/kombu/utils/functional.py__init__ zChannelPromise.__init__cCs.z|jWSty|}|_|YSwr) __value__AttributeErrorr)rvaluerrr__call__s  zChannelPromise.__call__cCs4zt|jWStydt|jddYSw)Nz )reprrr idrrrrr__repr__%s   zChannelPromise.__repr__N)__name__ __module__ __qualname__rr"r(rrrrrs rc@seZdZdZdddZddZddZdd d Zd d ZddZ ddZ e Z ddZ e Z ddZeZdddZddZddZeZe Ze ZdS) raMLRU Cache implementation using a doubly linked list to track access. Arguments: --------- limit (int): The maximum number of keys to keep in the cache. When a new key is inserted and the limit has been exceeded, the *Least Recently Used* key will be discarded from the cache. NcCs||_t|_t|_dSr)limit threadingRLockmutexrdata)rr,rrrr7s  zLRUCache.__init__cCsB|j|j|}||<|WdS1swYdSr)r/r0poprkeyr!rrr __getitem__<s$zLRUCache.__getitem__cOs|jA|j|j}}|j|i||r,t||kr4tt||D]}|jddq#WddSWddSWddS1sGwYdS)NF)last)r/r0r,updatelenrangepopitem)rargskwargsr0r,_rrrr6As"zLRUCache.updateTcCs6|j|j|WdS1swYdSr)r/r0r9)rr5rrrr9Js $zLRUCache.popitemcCsb|j$|jrt|j|jkr|jtt|j||j|<WddS1s*wYdSr)r/r,r7r0r1nextiterr2rrr __setitem__Ns  "zLRUCache.__setitem__cCs t|jSr)r>r0r'rrr__iter__UrzLRUCache.__iter__c cs^|j!|D]}z ||j|fVWqtyYqwWddS1s(wYdSrr/r0KeyErrorrkrrr_iterate_itemsXs "zLRUCache._iterate_itemsc csZ|j|D]}z|j|VWqtyYqwWddS1s&wYdSrrArCrrr_iterate_valuesas "zLRUCache._iterate_valuescCs4|j |jWdS1swYdSr)r/r0keysr'rrr _iterate_keysks$zLRUCache._iterate_keysr cCsN|jt|j||}t|||<|WdS1s wYdSr)r/intr0r1str)rr3deltanewvalrrrincrqs  $z LRUCache.incrcCstt|}|d|S)Nr/)dictvarsr1)rdrrr __getstate__ys  zLRUCache.__getstate__cCs||_t|_dSr)__dict__r-r.r/)rstaterrr __setstate__~szLRUCache.__setstate__r)T)r )r)r*r+__doc__rr4r6r9r?r@rE iteritemsrF itervaluesrHiterkeysrMrQrTrGvaluesitemsrrrrr,s(   rcsfdd}|S)z)Decorator to cache function return value.csXtdtfddfdd}d__|__S)N)r,c sr||}n |tftt|}z |}Wdn1s'wYWn0ty]|i|}jd7_||<WdY|S1sUwYY|Swjd7_|S)Nr ) KEYWORD_MARKtuplesortedrZrBmisseshits)r:r;r3r!)_Mcachefunkeyfunr/rrr`s(     z%memoize.._memoize.._Mcsd__dS)z+Clear the cache and reset cache statistics.rN)clearr_r^r)r`rarrrdsz(memoize.._memoize..clearr)r-Lockr r_r^rd original_func)rbrdCachercmaxsize)r`rarbr/r_memoizes  zmemoize.._memoizer)rircrhrjrrgrrs rc@sXeZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ dS)ra Holds lazy evaluation. Evaluated when called or if the :meth:`evaluate` method is called. The function is re-evaluated on every call. Overloaded operations that will evaluate the promise: :meth:`__str__`, :meth:`__repr__`, :meth:`__cmp__`. cOs||_||_||_dSr_fun_args_kwargs)rrbr:r;rrrrs z lazy.__init__cCs|Sr)evaluater'rrrr"sz lazy.__call__cCs|j|ji|jSrrkr'rrrrosz lazy.evaluatecC t|Sr)rJr'rrr__str__rz lazy.__str__cCrpr)r%r'rrrr(rz lazy.__repr__cCs ||kSrrrrhsrrr__eq__rz lazy.__eq__cCs ||kSrrrrrrr__ne__rz lazy.__ne__cCs||t|<|Sr)r&)rmemorrr __deepcopy__s zlazy.__deepcopy__cCs|j|jf|j|jdfS)N)rmrn) __class__rlrmrnr'rrr __reduce__szlazy.__reduce__N) r)r*r+rUrr"rorqr(rtrurwryrrrrrs  rcCst|tr |S|S)z9Evaluate value only if value is a :class:`lazy` instance.) isinstancerro)r!rrrrs rcCst||o t||p d S)zwReturn true if the object is iterable. Note: ---- Returns false if object is a mapping or string. r)rz)objscalarsitersrrrrsrcCs|dus t||r |S|gS)z0Return list of one element if ``l`` is a scalar.N)r)r{r|rrrrsrcKs6|dur|n |rt|fi|n|}dd|DS)z=Remove all keys from dict ``d`` whose value is :const:`None`.NcSsi|] \}}|dur||qSrr.0rDvrrr szdictfilter..)rNrZ)rPkwrrrrs$rccs4t|}tj}tdD] }|||dVq dS)Nr)listrandomshuffler)itrr<rrr shufflecycles  r?Fccs<|d} |r ||kr|V||7}n |sdS||Vq)Nrr)startstopstep repeatlastcurrrrfxranges   rY@ccsJd|d}} ||krdS|V|rt|||}n||7}||7}q )Nrr)min)rrrmaxsum_rrrr fxrangemaxsrc  Cs |sin|}|s gn|}t||||dd} | rt| nd} tD]`} z ||i|WS|y}zH|dur?| |kr?| rGt| krG| rL| t|rU||| | nt| }|rxtt|D] }| ri| tdqbttt||WYd}~q"d}~wwdS)aRetry the function over and over until max retries is exceeded. For each retry we sleep a for a while before we try again, this interval is increased for every retry until the max seconds is reached. Arguments: --------- fun (Callable): The function to try catch (Tuple[BaseException]): Exceptions to catch, can be either tuple or a single exception class. Keyword Arguments: ----------------- args (Tuple): Positional arguments passed on to the function. kwargs (Dict): Keyword arguments passed on to the function. errback (Callable): Callback for when an exception in ``catch`` is raised. The callback must take three arguments: ``exc``, ``interval_range`` and ``retries``, where ``exc`` is the exception instance, ``interval_range`` is an iterator which return the time in seconds to sleep next, and ``retries`` is the number of previous retries. max_retries (int): Maximum number of retries before we give up. If neither of this and timeout is set, we will retry forever. If one of this and timeout is reached, stop. interval_start (float): How long (in seconds) we start sleeping between retries. interval_step (float): By how much the interval is increased for each retry. interval_max (float): Maximum number of seconds to sleep between retries. timeout (int): Maximum seconds waiting before we give up. T)rNr) rr rfloatr=r8rIr abs)rbcatchr:r;errback max_retriesinterval_start interval_step interval_maxcallbacktimeoutinterval_rangeendretriesexcttsr<rrrrs: #   r, {0}={1}cs|fdd|DS)Nc3s$|] \}}|t|VqdSr)format _safe_reprr~fmtrr Rs"zreprkwargs..)joinrZ)r;seprrrr reprkwargsQsrrcCs>|sin|}d||tt|pd|r|r|pdt||S)Nz {}({}{}{})r)rrmaprr)namer:r;rrrrreprcallUs rcCst|}||jvp||jvSr)inspectgetfullargspecr: kwonlyargs)func argument_name argument_specrrraccepts_argument^s  rr)rNrF)rNrr) NNNNrrrNN)rr)rNr)+rU __future__rrrr- collectionsrrcollections.abcrr itertoolsrrr r vine.utilsr encodingr r__all__objectr[rrrrrrJrrrrrrrrrrpromise maybe_promiserrrrsB   [%*     >