o :ګec@sddlZddlmZddlmZddlmZmZeZdde de de fdd Z de d ej fd d Zde d ej dee fd dZde fddZdde de de fddZdedefddZde d ej dee fddZdefddZddZdS)N)time) MethodType)CallableOptionalqueuestimeoutreturncCsJtt}t||z t|||Wt|nt|wtdd|S)a/Waits for one or more *Queue* to be ready or until *timeout* expires. *queues* is a list containing one or more *Queue.Queue* objects. If *timeout* is not None the function will block for the specified amount of seconds. The function returns a list containing the ready *Queues*. cS | SNemptyqrg/vol/project/2023/60021/g236002117/DeepSeek-Coder/venv/lib/python3.10/site-packages/pebble/functions.py, zwaitforqueues..) threading ConditionLockprepare_queues wait_queues reset_queuesfilter)rrlockrrr waitforqueuess  rrc CsN|D]"}||_|j|j|_tt||_Wdn1swYqdS)zFReplaces queue._put() method in order to notify the waiting Condition.N) _pebble_lockmutex_put_pebble_old_methodr new_method)rrqueuerrrr/srcCsV|ttdd|s||WddSWddS1s$wYdS)NcSr r r r rrrr<rzwait_queues..)anymapwait)rrrrrrr8s  "rc CsP|D]#}|j |j|_Wdn1swYt|dt|dqdS)z$Resets original queue._put() method.Nrr)rrrdelattr)rr!rrrr@s   rthreadscsZdttfdd}t|z t||Wtntwtdd|S)aWaits for one or more *Thread* to exit or until *timeout* expires. .. note:: Expired *Threads* are not joined by *waitforthreads*. *threads* is a list containing one or more *threading.Thread* objects. If *timeout* is not None the function will block for the specified amount of seconds. The function returns a list containing the ready *Threads*. Ncs:| WddS1swYdSr ) notify_all)argsr old_functionrr new_functionZs "z$waitforthreads..new_functioncSr r is_alivetrrrrerz waitforthreads..)rrrprepare_threads wait_threads reset_threadsr)r&rr+rr)rwaitforthreadsIsr3r+cCs8ttj}|t_|WdS1swYdS)zVReplaces threading._get_ident() function in order to notify the waiting Condition.N_waitforthreads_lockr get_ident)r+r*rrrr0hs $r0cCst}|Ittdd|sF|dur|n|t|dkr,||t|n WddSttdd|rWddSWddS1sQwYdS)NcSr r r,r.rrrrxrzwait_threads..r)rr"r#r$)r&rr timestamprrrr1rs "r1r*cCs0t |t_WddS1swYdS)z/Resets original threading.get_ident() function.Nr4)r*rrrr2s"r2cGs@|j||j|jWddS1swYdSr )rrr')selfr(rrrr s  "r r )rrtypesrtypingrrrr5listfloatrrrrrrr3r0r1r2r rrrrs,