o Â¿iRã@s<ddlmZddlZddlZddlZddlmZddlZddlm Z ddlm Z ddlmZddlmZddlm Z dd lmZdd lmZddlmZdd lmZddlmZddlmZddlmZddlmZddlmZedƒZd'dd„ZGdd„dƒZGdd„dƒZeGdd„dƒƒZGd d!„d!ƒZd(d%d&„ZdS))é)ÚannotationsN)Ú dataclass)ÚAny)ÚCallable)Ú Coroutine)ÚDeque)ÚList)ÚOptional)ÚTypeVaré)ÚClient)ÚInArgs)ÚInStatement)ÚLibsqlError)ÚTransaction©Ú create_client)Ú ResultSetÚTÚargsrÚkwargsÚreturnÚ ClientSynccs@tƒ}z| ‡‡fdd„¡}t||ƒWSty| ¡‚w)Ncstˆiˆ¤ŽS©Nr©©rrrúF/home/ubuntu/.local/lib/python3.10/site-packages/libsql_client/sync.pyÚsz$create_client_sync..)Ú_AsyncExecutorÚsubmit_funcrÚ ExceptionÚclose)rrÚexecutorÚclientrrrÚcreate_client_syncsþr$c@steZdZUded<ded<d(dd„Zd)d*dd„Zd+dd„Zd,dd„Zd-dd„Ze d.dd„ƒZ d/d d!„Zd0d&d'„Zd S)1rrÚ _executorrÚ_clientr"r#cCó||_||_dSr)r%r&)Úselfr"r#rrrÚ__init__)ó zClientSync.__init__NÚstmtrrr rrcCó|j |j ||¡¡Sr)r%Úsubmit_coror&Úexecute©r(r+rrrrr.-ózClientSync.executeÚstmtsúList[InStatement]úList[ResultSet]cCs|j |j |¡¡Sr)r%r-r&Úbatch)r(r1rrrr40ózClientSync.batchÚTransactionSynccCs|j |jj¡}t|j|ƒSr)r%rr&Útransactionr6)r(r7rrrr73szClientSync.transactionÚNonecCs|j |jj¡dSr)r%Úclose_with_coror&r!©r(rrrr!7r5zClientSync.closeÚboolcCs |j ¡Sr)r%Ú is_closedr:rrrÚclosed:s zClientSync.closedcCó|Srrr:rrrÚ __enter__>ózClientSync.__enter__Úexc_typerÚexc_valÚexc_tbcCó| ¡dSr©r!©r(rArBrCrrrÚ__exit__AózClientSync.__exit__)r"rr#rr©r+rrr rr)r1r2rr3©rr6©rr8©rr;)rr©rArrBrrCrrr8) Ú__name__Ú __module__Ú__qualname__Ú__annotations__r)r.r4r7r!Úpropertyr=r?rGrrrrr%s c@steZdZUded<ded<d$dd„Zd%d&dd„Zd'dd„Zd'dd„Zd'dd„Ze d(dd„ƒZ d)dd„Zd*d"d#„Zd S)+r6rr%rÚ_transactionr"r7cCr'r)r%rS)r(r"r7rrrr)Ir*zTransactionSync.__init__Nr+rrr rrcCr,r)r%r-rSr.r/rrrr.Mr0zTransactionSync.executer8cCó|j |j ¡¡Sr)r%r-rSÚrollbackr:rrrrUPózTransactionSync.rollbackcCrTr)r%r-rSÚcommitr:rrrrWSrVzTransactionSync.commitcCs|j |jjdd„¡dS)NcSsdSrrrrrrrWóz'TransactionSync.close..)r%Úsubmit_func_unless_closedrSr!r:rrrr!VszTransactionSync.closer;csˆj ‡fdd„dd„¡S)NcsˆjjSr)rSr=rr:rrr\sz(TransactionSync.closed..cSsdS)NTrrrrrr\rX)r%rYr:rr:rr=YsÿzTransactionSync.closedcCr>rrr:rrrr?_r@zTransactionSync.__enter__rArrBrCcCrDrrErFrrrrGbrHzTransactionSync.__exit__)r"rr7rrrIrKrLrJrM) rNrOrPrQr)r.rUrWr!rRr=r?rGrrrrr6Es r6c@seZdZUded<ded<dS)Ú _QueueItemzCoroutine[Any, Any, Any]Ú coroutinezconcurrent.futures.FutureÚfutureN)rNrOrPrQrrrrrZfs rZc@s¬eZdZUded<ded<ded<ded<d ed <ded<d1dd„Zd1dd„Zd1dd„Zd2dd„Zd3dd„Zd4dd„Z d5d"d#„Z d6d&d'„Zd7d*d+„Zd1d,d-„Z d8d.d/„Zd0S)9rzthreading.ThreadÚ_threadúasyncio.AbstractEventLoopÚ_loopzthreading.LockÚ_lockr;Ú_closedzDeque[Optional[_QueueItem]]Ú_queuezOptional[asyncio.Future[None]]Ú_wakerrr8cCsJtj|jdd|_t ¡|_t ¡|_d|_ t ¡|_d|_ |j ¡dS)NÚ libsql_client)ÚtargetÚnameF)Ú threadingÚThreadÚ_runr]ÚasyncioÚnew_event_loopr_ÚLockr`raÚcollectionsÚdequerbrcÚstartr:rrrr)us z_AsyncExecutor.__init__cCsFt |j¡|j | ¡¡t|jƒ|j |j ¡¡|j ¡dSr)rjÚset_event_loopr_Úrun_until_completeÚ_run_on_loopÚ_cancel_all_tasksÚshutdown_asyncgensr!r:rrrri€s z_AsyncExecutor._runc ÃsÊ | ¡IdH}|durn&z|j |jIdH¡Wnty2}z|j |¡WYd}~nd}~wwq|j$d|_|jD]}|durM|j t ddƒ¡q>|j ¡WdƒdS1s^wYdS)NTúClient is closedÚ CLIENT_CLOSED)Ú _dequeue_itemr\Ú set_resultr[r Ú set_exceptionr`rarbrÚclear)r(ÚitemÚerrrrr‡s,€€ÿú ÿ€"ùz_AsyncExecutor._run_on_loopúOptional[_QueueItem]cÃsz |j*t|jƒdkr|j ¡WdƒS|jdus J‚|j ¡}||_Wdƒn1s2wY|IdHq)NTr)r`ÚlenrbÚpopleftrcr_Ú create_future)r(Úwakerrrrrwšs€þ û ùz_AsyncExecutor._dequeue_itemr{csH|j |¡|jd}|_|dur"|‰d‡fdd„}|j |¡dSdS)Nrr8csˆ d¡dSr)rxr©Úwaker_rrÚ resolve_wakerªsz=_AsyncExecutor._enqueue_item_with_lock..resolve_wakerrK)rbÚappendrcr_Úcall_soon_threadsafe)r(r{rr„rr‚rÚ_enqueue_item_with_lock¤súz&_AsyncExecutor._enqueue_item_with_lockÚcoroúCoroutine[Any, Any, T]rcCs^|j |jrtddƒ‚tj ¡}| t||ƒ¡Wdƒ| ¡S1s&wY| ¡S)Nrurv) r`rarÚ concurrentÚfuturesÚFuturer‡rZÚresult)r(rˆÚfutrrrr-¯s üûz_AsyncExecutor.submit_coroÚfuncúCallable[[], T]csd‡fdd„}| |ƒ¡S)Nrrc“óˆƒSrrr©rrrrˆ¸ó€z(_AsyncExecutor.submit_func..coro©rr)r-)r(rrˆrr’rr·sz_AsyncExecutor.submit_funcÚon_openÚ on_closedcsxd‡fdd„}|j&|jr|ƒWdƒStj ¡}| t|ƒ|ƒ¡Wdƒ| ¡S1s3wY| ¡S)Nrrc“r‘rrr©r•rrÚon_open_coroÀr“z>_AsyncExecutor.submit_func_unless_closed..on_open_coror”)r`rarŠr‹rŒr‡rZr)r(r•r–r˜rŽrr—rrY½sþ üûz(_AsyncExecutor.submit_func_unless_closedÚ coro_funcú'Callable[[], Coroutine[Any, Any, None]]cCsz|j'|jr WdƒdStj ¡}| t|ƒ|ƒ¡| d¡Wdƒn1s-wY|j ¡| ¡dSr) r`rarŠr‹rŒr‡rZr]Újoinr)r(r™rŽrrrr9Êsþ û z_AsyncExecutor.close_with_corocCsddd„}| |¡dS)Nrr8cÓsdSrrrrrrÚnoop×s€z"_AsyncExecutor.close..nooprK)r9)r(rœrrrr!Ös z_AsyncExecutor.closecCs0|j|jWdƒS1swYdSr)r`rar:rrrr<Üs$ÿz_AsyncExecutor.is_closedNrK)rr})r{r}rr8)rˆr‰rr)rrrr)r•rr–rrr)r™ršrr8rL)rNrOrPrQr)rirrrwr‡r-rrYr9r!r<rrrrrls$ rÚloopr^r8cCstt |¡}|s dS|D]}| ¡q| tj|ddiŽ¡|D]}| ¡r&q| ¡dur7| d| ¡|dœ¡qdS)NÚreturn_exceptionsTz2unhandled exception during _AsyncExecutor shutdown)ÚmessageÚ exceptionÚtask)rjÚ all_tasksÚcancelrqÚgatherÚ cancelledr Úcall_exception_handler)rÚ to_cancelr¡rrrrsâs$ ýÿ€ürs)rrrrrr)rr^rr8) Ú __future__rrjrmrŠÚdataclassesrrgÚtypingrrrrrr r r#rr rrrrrrrr$rr6rZrrsrrrrÚs8 !v