o ^f%@sUdZddlmZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z ddl mZddlmZedZed ZejZejZejZejZejZejZd ed <d Zd ed <e rkejZd ed<ejZ d ed<nGddde!Z"Gdddeje"dZGdddeje"dZ d?ddZ#d?ddZ$GdddeeZGdddejZ%Gdd d ejZ&Gd!d"d"e eZ' # d@dAd*d+Z( # d@dBd.d/Z)dd0dCd6d7Z*dd0dDd8d9Z+dEd=d>Z,dS)Fz Utilities to ease the differences between async and sync code. These object offer a similar interface between sync and async versions; the script async_to_sync.py will replace the async names with the sync names when generating the sync version. ) annotationsN)AnyCallable Coroutine TYPE_CHECKING) TypeAlias)TypeVarz psycopg.poolTrWorkerasyncio.Task[None]AWorker_GQueue_AGQueuec@seZdZddZdS)FakeGenericMetacCs|SN)selfitemrrN/home/ertert/spirit/venv/lib/python3.10/site-packages/psycopg_pool/_acompat.py __getitem__.szFakeGenericMeta.__getitem__N)__name__ __module__ __qualname__rrrrrr-s rc@ eZdZdS)rNrrrrrrrr1) metaclassc@r)rNrrrrrr4rreturnstrcCs tjSr) threadingcurrent_threadnamerrrrcurrent_thread_name8s r#cCst}|r |SdS)Nz )asyncio current_taskget_name)trrrcurrent_task_name<sr(cs$eZdZdZd d fd d ZZS)Queuez> A Queue subclass with an interruptible get() method. TNblockbooltimeout float | Nonerr cs|durd}t||S)Ng@)superget)rr*r, __class__rrr/Fsz Queue.get)TN)r*r+r,r-rr )rrr__doc__r/ __classcell__rrr0rr)Asr)c@eZdZdZd ddZdS) AEventz Subclass of asyncio.Event adding a wait with timeout like threading.Event. wait_timeout() is converted to wait() by async_to_sync. r,floatrr+c6zt||IdHWdStjyYdSwNTFr$wait_forwait TimeoutErrorrr,rrr wait_timeoutTzAEvent.wait_timeoutNr,r6rr+rrrr2r>rrrrr5Mr5c@r4) AConditionz Subclass of asyncio.Condition adding a wait with timeout like threading.Condition. wait_timeout() is converted to wait() by async_to_sync. r,r6rr+cr7r8r9r=rrrr>cr?zACondition.wait_timeoutNr@rArrrrrC\rBrCc@r)AQueueNrrrrrrDkrrDrf(Callable[..., Coroutine[Any, Any, None]]argstuple[Any, ...]r" str | NonecCstj|||dS)z, Equivalent to asyncio.create_task. )r")r$ create_task)rErGr"rrraspawnosrKCallable[..., Any]threading.ThreadcCstj|||dd}||S)z= Equivalent to creating and running a daemon thread. T)targetrGr"daemon)r Threadstart)rErGr"r'rrrspawnzsrRr,tasksasyncio.Task[Any]r,r-Nonecstj|}z|durtjt||dIdHWdS|IdHWdStjy,Ynw|D]}|r6q/td||q/dS)7 Equivalent to asyncio.gather or Thread.join() NrSz'couldn't stop task %r within %s seconds) r$gatherr:shieldr<doneloggerwarningr&)r,rTr;r'rrragathers   r]cGs<|D]}|s q|||sqtd|j|qdS)rWz)couldn't stop thread %r within %s secondsN)is_alivejoinr[r\r")r,rTr'rrrrXs rXsecondsr6Coroutine[Any, Any, None]cCs t|S)zT Equivalent to asyncio.sleep(), converted to time.sleep() by async_to_sync. )r$sleep)r`rrrasleeps rc)rr)rN)rErFrGrHr"rIrr )rErLrGrHr"rIrrM)rTrUr,r-rrV)rTrMr,r-rrV)r`r6rra)-r2 __future__rtimequeuer$loggingr typingrrrrtyping_extensionsr_compatr getLoggerr[r Event ConditionRLockLockALockrbrPr __annotations__r r)rrtyperr#r(r5rCrDrKrRr]rXrcrrrrsP