U Íø.eÁ ã@s˜dZddlmZddlZddlmZddlmZmZm Z ej dkrPddl m Z n ddlm Z erlddl mZd Ze d ƒZGd d „d ƒZGd d„dƒZdS)z&Semaphores and concurrency primitives.é)Ú annotationsN)Údeque)Ú TYPE_CHECKINGÚCallableÚDeque)éé )Ú ParamSpec)Ú TracebackType)Ú DummyLockÚLaxBoundedSemaphoreÚPc@s‚eZdZdZdddœdd„Zddd d d œd d „Zddœdd„Zddddœdd„Zddddœdd„Zddœdd„Z ddœdd„Z dS)r aûAsynchronous Bounded Semaphore. Lax means that the value will stay within the specified range even if released more times than it was acquired. Example: ------- >>> x = LaxBoundedSemaphore(2) >>> x.acquire(print, 'HELLO 1') HELLO 1 >>> x.acquire(print, 'HELLO 2') HELLO 2 >>> x.acquire(print, 'HELLO 3') >>> x._waiters # private, do not access directly [print, ('HELLO 3',)] >>> x.release() HELLO 3 ÚintÚNone)ÚvalueÚreturncCs,||_|_tƒ|_|jj|_|jj|_dS©N)Ú initial_valuerrÚ_waitingÚappendÚ _add_waiterÚpopleftÚ _pop_waiter)Úselfr©rú@/tmp/pip-unpacked-wheel-48hrr5dg/kombu/asynchronous/semaphore.pyÚ__init__.s  zLaxBoundedSemaphore.__init__zCallable[P, None]zP.argszP.kwargsÚbool)ÚcallbackÚ partial_argsÚpartial_kwargsrcOsD|j}|dkr"| |||f¡dSt|ddƒ|_|||ŽdSdS)a^Acquire semaphore. This will immediately apply ``callback`` if the resource is available, otherwise the callback is suspended until the semaphore is released. Arguments: --------- callback (Callable): The callback to apply. *partial_args (Any): partial arguments to callback. rFéTN)rrÚmax)rrrr rrrrÚacquire4s zLaxBoundedSemaphore.acquire©rcCsJz| ¡\}}}Wn(tk r:t|jd|jƒ|_Yn X|||ŽdS)z¸Release semaphore. Note: ---- If there are any waiters this will apply the first waiter that is waiting for the resource (FIFO order). r!N)rÚ IndexErrorÚminrr)rÚwaiterÚargsÚkwargsrrrÚreleaseNs zLaxBoundedSemaphore.releaser!)ÚnrcCs6|j|7_|j|7_t|ƒD] }| ¡q$dS)z6Change the size of the semaphore to accept more users.N)rrÚranger*)rr+Ú_rrrÚgrow]s zLaxBoundedSemaphore.growcCs(t|j|dƒ|_t|j|dƒ|_dS)z6Change the size of the semaphore to accept less users.rN)r"rr)rr+rrrÚshrinkdszLaxBoundedSemaphore.shrinkcCs|j ¡|j|_dS)z@Reset the semaphore, which also wipes out any waiting callbacks.N)rÚclearrr©rrrrr0is zLaxBoundedSemaphore.clearÚstrcCs d |jjt|ƒ|jt|jƒ¡S)Nz!<{} at {:#x} value:{} waiting:{}>)ÚformatÚ __class__Ú__name__ÚidrÚlenrr1rrrÚ__repr__ns ÿzLaxBoundedSemaphore.__repr__N)r!)r!) r5Ú __module__Ú __qualname__Ú__doc__rr#r*r.r/r0r8rrrrr sr c@s2eZdZdZddœdd„Zddddd œd d „Zd S) r zPretending to be a lock.r$cCs|Srrr1rrrÚ __enter__wszDummyLock.__enter__ztype[BaseException] | NonezBaseException | NonezTracebackType | Noner)Úexc_typeÚexc_valÚexc_tbrcCsdSrr)rr=r>r?rrrÚ__exit__zszDummyLock.__exit__N)r5r9r:r;r<r@rrrrr tsr )r;Ú __future__rÚsysÚ collectionsrÚtypingrrrÚ version_infoZtyping_extensionsr Útypesr Ú__all__r r r rrrrÚs     ^