mBNjs:zUddlmZddlZddlZddlZddlZddlmZddl m Z gdZ ejZ Gdd ejZGd d ZGd d ZGddZej$Zded<ejZded<ejZ ded<ej*Zded<ej*Zded<y)) annotationsN)_typing)wait_for) AsyncConditionAsyncCooperativeLockAsyncCooperativeRLock AsyncLock AsyncRLock ConditionCooperativeLockCooperativeRLockLockRLockceZdZdZej dZfdZfdZd dZ fdZ fdZ d dZ e Z fd Zd Zd ZxZS)r a Reentrant asyncio.lock. Inspired by Python's RLock implementation. .. warning:: In async Python there are no threads. This implementation uses :meth:`asyncio.current_task` to determine the owner of the lock. This means that the owner changes when using :meth:`asyncio.wait_for` or any other method that wraps the work in a new :class:`asyncio.Task`. z (?:\W|^)waiters[:=](\d+)(?:\W|$)c@t||i|d|_d|_yNr)super__init___owner_count)selfargskwargs __class__s p/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/neo4j/_async_compat/concurrency.pyrzAsyncRLock.__init__<s# $)&)  cPtj|}t|}|jdkDrdnd}|d|jz }|jj |}|r|d|j dz }|jr|d|jz }d|dd d |d S) Nrlockedunlockedz count=z waiters=rz owner=< []>)object__repr__rr _WAITERS_REsearchgroupr)rres lock_reprextra waiters_matchrs rr'zAsyncRLock.__repr__Asood#G$&  KK!O 74;;-((((// :  y!4!4Q!7 89 9E ;; wt{{m, ,E3q9+Rwb))rcL|tj}|j|k(SN)asyncio current_taskr)rtasks ris_ownerzAsyncRLock.is_ownerMs$ <'')D{{d""rcK|j|r|xjdz c_yt| }t j |} t j dd{|jr#|j}|||_ d|_y||jy7H#tj$r%t jjY}wxYww)Nr3rTrF) r4rracquirer1 ensure_futuresleepCancelledErrorr2canceldone exceptionr)rme acquire_coror3r=rs r_acquire_non_blockingz AsyncRLock._acquire_non_blockingRs ==b= ! KK1 Kw( $$\2 ,--" " " 99;(I     ! #%% ,  " ) ) +  ,s=A C.B3(B1)B3-AC.1B335C+(C.*C++C.cK|j|r|xjdz c_yt| d{||_d|_y7w)Nr6r)r4rrr7rrr>rs r_acquirezAsyncRLock._acquiremsI ==b= ! KK1 K go   s:AAAcFKtj}|dkr|dk7r td|s|dk7r td|s|j|d{S|r|dk(r|j |d{y tj |j |} t ||d{y7`7B7 #tj$r7|jr|j }|r|j|wxYw#tj$rYywxYww)zAcquire the lock.rr#ztimeout value must be positivez/can't specify a timeout for a non-blocking callNTF) r1r2 ValueErrorr@rCr8rr: cancelledr;_release TimeoutError)rblockingtimeoutr>futalready_finisheds rr7zAsyncRLock.acquireus  ! ! # Q;7b==> >GrMNO O33B77 7 2 --# # # '' b(9:C sG,,,%8 # -)) ==?'*zz|#3 #MM"% ##  smAD!B5D!4B75D!;$D B;/B90B;4D!7D!9B;;A DDDD!DD!c|j|s"|j tdtd|xjdzc_|jsd|_t|yy)Nr6z Cannot release un-acquired lock.zCannot release foreign lock.r)r4r RuntimeErrorrrreleaserBs rrGzAsyncRLock._releases^}}"}%{{""#EFF=> > q {{DK GO rcLtj}|j|S)zRelease the lock.)r1r2rGrr>s rrOzAsyncRLock.releases  ! ! #}}R  rc,K|jywr0rOrtvtbs r __aexit__zAsyncRLock.__aexit__ r0)Tr#)__name__ __module__ __qualname____doc__recompiler(rr'r4r@rCr7 __aenter__rGrOrX __classcell__rs@rr r -sO "**@AK *# 6<J! rr cPeZdZdZdZfdZdZdZdZeZ dZ dZ d Z xZ S) r a6 Lock placeholder for asyncio Python when working fully cooperatively. This lock doesn't do anything in async Python. Its threaded counterpart, however, is an ordinary :class:`threading.Lock`. The AsyncCooperativeLock only works if there is no await being used while the lock is held. cd|_y)NF_lockedrs rrzAsyncCooperativeLock.__init__s  rcXt|}|jrdnd}d|ddd|dS)Nr r!r"rr#r$r%)rr'rgrr+r-rs rr'zAsyncCooperativeLock.__repr__s8g  LLj3q9+Rwb))rc|jS) Return True if lock is acquired.rfrhs rr zAsyncCooperativeLock.lockeds ||rc@|jr tdd|_y)a Acquire a lock. This method will raise a RuntimeError where an ordinary (non-placeholder) lock would need to block. I.e., when the lock is already taken. Returns True if the lock was successfully acquired. z)Cannot acquire a locked cooperative lock.TrgrNrhs rr7zAsyncCooperativeLock.acquires! <<JK K rc@|jrd|_ytd)z Release a lock. When the lock is locked, reset it to unlocked, and return. When invoked on an unlocked lock, a RuntimeError is raised. There is no return value. FLock is not acquired.Nrnrhs rrOzAsyncCooperativeLock.releases << DL67 7rc$|jyr0rSrTs r__exit__zAsyncCooperativeLock.__exit__  rc*K|jSwr0) __enter__rhs rrazAsyncCooperativeLock.__aenter__s~~sc2K|j|||ywr0)rrrTs rrXzAsyncCooperativeLock.__aexit__s aBs)r[r\r]r^rr'r r7rOrurrrarXrbrcs@rr r s7*   8I  rr cDeZdZdZdZfdZdZdZdZeZ dZ xZ S)r a0 Reentrant lock placeholder for cooperative asyncio Python. This lock doesn't do anything in async Python. It's threaded counterpart, however, is an ordinary :class:`threading.Lock`. The AsyncCooperativeLock only works if there is no await being used while the lock is acquired. c d|_d|_yr)rrrhs rrzAsyncCooperativeRLock.__init__s  rct|}|jd|jd|j}nd}d|ddd|dS) Nzlocked z times by owner:r!r"rr#r$r%)rr'rrrjs rr'zAsyncCooperativeRLock.__repr__sUg  ;; "dkk]*:4;;-HEE3q9+Rwb))rc|jduS)rlN)rrhs rr zAsyncCooperativeRLock.lockeds{{$&&rctj}|j||_d|_y|j|ur|xjdz c_yt d)a Acquire a lock. This method will raise a RuntimeError where an ordinary (non-placeholder) lock would need to block. I.e., when the lock is already taken by another Task. Returns True if the lock was successfully acquired. rTz1Cannot acquire a foreign locked cooperative lock.)r1r2rrrNrQs rr7zAsyncCooperativeRLock.acquiresV ! ! # ;; DKDK ;;"  KK1 KNOOrctj}|j td|j|ur td|xjdzc_|jsd|_yy)z Release a lock. When the lock is locked, reset it to unlocked, and return. When invoked on an unlocked or foreign lock, a RuntimeError is raised. There is no return value. NrpzCannot release a foreign lock.r)r1r2rrNrrQs rrOzAsyncCooperativeRLock.releasesa ! ! # ;; 67 7 ;;b ?@ @ q {{DKrc$|jyr0rSrTs rrrzAsyncCooperativeRLock.__exit__+rsr) r[r\r]r^rr'r r7rOrurrrbrcs@rr r s.*'P(&Irr ceZdZdZd dZdZejZdZ dZ dZ fdZ ddZ d d Zd Zdd Zd ZxZS)ra$ Asynchronous equivalent to threading.Condition. This class implements condition variable objects. A condition variable allows one or more coroutines to wait until they are notified by another coroutine. A new Lock object is created and used as the underlying lock. Nc| t}||_|j|_|j|_|j|_t j |_yr0)r _lockr r7rO collectionsdeque_waiters)rlocks rrzAsyncCondition.__init__BsF <;D kk || || #))+ rc  tj}|j(|j5|j||_ddd||jurt|d|S#t$rd}Y_wxYw#1swY8xYw)Nz# is bound to a different event loop)r1get_running_looprN_loop _loop_lock)rloops r _get_loopzAsyncCondition._get_loopQs ++-D :: ::%!%DJ! tzz !$)LMN N  D !sA(A9( A65A69BcKt|jttfr|jj y|j d{y7wr0) isinstancerr r r7rhs rrazAsyncCondition.__aenter___s?  JJ-/DE  JJ   ,,. sAAAAc,K|jywr0rS)rexc_typeexcrWs rrXzAsyncCondition.__aexit__grYrZct|}|jrdnd}|jr|dt |j}d|ddd|dS) Nr r!z , waiters:r"rr#r$r%)rr'r rlenrjs rr'zAsyncCondition.__repr__js[g  KKMz ==gZDMM(:';rFrKs r_waitzAsyncCondition._waitqs {{}@A A djj* - JJ   # JJ    -.."002C MM  % *sG,,, $$S) 13HI ""$)%djj*="&**"5"5b"999"&**"4"4"666,,,5-(:6"11)$( )''   $$S) 13HI ""$)%djj*="&**"5"5b"999"&**"4"4"666"11)$( ),,,-))    $$S) 13HI ""$)%djj*="&**"5"5b"999"&**"4"4"666"11)$( ),,,s,A.N819K%+F.:F ;F.K%)r1r2r)rrJr>s rwaitzAsyncCondition.waits-  ! ! #ZZBZ7777s +424chK|}|s"|jd{|}|s"|S7w)z Wait until a predicate becomes true. The predicate should be a callable which result will be interpreted as a boolean value. The final predicate value is the return value. N)r)r predicateresults rrzAsyncCondition.wait_fors9))+  [F  s20 22c|js tdd}|jD]0}||k\ry|jr|dz }|j d2y)a_ Wake up a single threads waiting on this condition. By default, wake up one coroutine waiting on this condition, if any. If the calling coroutine has not acquired the lock when this method is called, a RuntimeError is raised. This method wakes up at most n of the coroutines waiting for the condition variable; it is a no-op if no coroutines are waiting. Note: an awakened coroutine does not actually return from its wait() call until it can reacquire the lock. Since notify() does not release the lock, its caller should. z!cannot notify on un-acquired lockrrFN)r rNrr< set_result)rnidxrKs rnotifyzAsyncCondition.notifysT{{}BC C==Cax88:qu% !rcL|jt|jy)a Wake up all threads waiting on this condition. This method acts like notify(), but wakes up all waiting threads instead of one. If the calling thread has not acquired the lock when this method is called, a RuntimeError is raised. N)rrrrhs r notify_allzAsyncCondition.notify_alls C &'rr0)NN)r)r[r\r]r^rr threadingrrrrarXr'rrrrrrbrcs@rrr/sP$ , E!J !*3-j8 &6(rrz t.TypeAliasr rrrr) __future__rr1rr_rrrUshimsr__all__rr r r r rr __annotations__rrrrrrs"#   LL xxv= = @DDNl(l(^#,, ;,(~~-NNk" )+/__{$r