ci: dZddlZddlZddlZddlmZmZmZddlm Z ddl m Z ddl m Z mZmZddlmZddlmZmZmZmZmZdd lmZd d lmZd d lmZgd Zeee e e e gefZ!ejDdZ#eedZ$dejJdeddfdZ&dejJde'ddfdZ(e)Z*ee+eejJegeffZ,de,fdZ-GddZ.ddddee e fdeej^dede.fdZ0y) z* Core implementation of aiosqlite proxies N) AsyncIterator GeneratorIterable)partial)Path)EmptyQueue SimpleQueue)Thread)AnyCallableLiteralOptionalUnion)warn)contextmanager)Cursor)connect Connectionr aiosqlite)DEFERRED IMMEDIATE EXCLUSIVEfutresultreturncH|js|j|yy)z9Set the result of a future if it hasn't been set already.N)done set_result)rrs 5/tmp/pip-target-z3e9_cxr/lib/python/aiosqlite/core.pyr r s 88: v ecH|js|j|yy)zedhh[Q   J"  r"ch|jyt|dtd|jy)Nzi was deleted before being closed. Please use 'async with' or '.close()' to close the connection properly.r) stacklevel)r:rResourceWarningstopr@s r!__del__zConnection.__del__bs=    #  (ZZ    r"cd_fd} tjj}j j ||f|S#t$rd}Y,wxYw)zBStop the background thread. Prefer `async with` or `await close()`Fcjj!jjd_tSN)r:closer-rFsr!close_and_stopz'Connection.stop..close_and_stopxs/+  &&(#' ) )r"N)r9asyncioget_event_loop create_future Exceptionr< put_nowait)r@rLr/s` r!rEzConnection.stopts_  *  ++-;;=F V^45  F s"A AAcH|j td|jS)Nzno active connection)r: ValueErrorrFs r!_connzConnection._conns&    #34 4r"sql parametersc||jj||}|jd|jS)NzSELECT last_insert_rowid())rTexecutefetchoner@rUrVcursors r!_execute_insertzConnection._execute_inserts2##C434  r"cZ|jj||}|jSrJ)rTrXfetchallrZs r!_execute_fetchallzConnection._execute_fetchalls%##C4  r"cK|jr |js tdt|g|i|}t j j }|jj||f|d{S7w)z8Queue a function with the given arguments for execution.zConnection closedN) r9r:rSrrMrNrOr<rQ)r@fnr8kwargsr0r/s r!_executezConnection._executesm}}D$4$401 12///'')779 VX./||sA7B9A>:Bc*K|j[ tjj}|jj ||j f|d{|_|S|S7 #t$r|jd|_wxYww)z&Connect to the actual sqlite database.N) r:rMrNrOr<rQr;r.rEr@r/s r!_connectzConnection._connects    #  //1??A##VT__$=>)/<  t $0   #'  s/ BAA.A, A.(B,A.."BBcr|jj|jjSrJ)r>startrf __await__rFs r!rizConnection.__await__s' }}((**r"c"K|d{S7wrJrFs r! __aenter__zConnection.__aenter__szzs  c@K|jd{y7wrJ)rK)r@exc_typeexc_valexc_tbs r! __aexit__zConnection.__aexit__sjjls c|Kt||j|jjd{S7w)zA sql_scriptcK|j|jj|d{}t||S7w)z4Helper to create a cursor and execute a user script.N)rcrT executescriptr)r@rr[s r!rzConnection.executescripts8}}TZZ%=%=zJJdF##Ks *?=?c>K|jjSw)zInterrupt pending queries.)rT interruptrFs r!rzConnection.interruptszz##%%sname num_paramsfunc deterministicctK|j|jj||||d{y7w)a/ Create user-defined function that can be later used within SQL statements. Must be run within the same thread that query executions take place so instead of executing directly against the connection, we defer this to `run` function. If ``deterministic`` is true, the created function is marked as deterministic, which allows SQLite to perform additional optimizations. This flag is supported by SQLite 3.8.3 or higher, ``NotSupportedError`` will be raised if used with older versions. )rN)rcrTcreate_function)r@rrrrs r!rzConnection.create_functions<mm JJ & &   '    s .868c.|jjSrJ)rTin_transactionrFs r!rzConnection.in_transactionszz(((r"c.|jjSrJrTisolation_levelrFs r!rzConnection.isolation_level szz)))r"valuec&||j_yrJrr@rs r!rzConnection.isolation_level$s%* "r"c.|jjSrJrT row_factoryrFs r!rzConnection.row_factory(szz%%%r"factoryc&||j_yrJrr@rs r!rzConnection.row_factory,s!( r"c.|jjSrJrT text_factoryrFs r!rzConnection.text_factory0szz&&&r"c&||j_yrJrrs r!rzConnection.text_factory4s") r"c.|jjSrJ)rT total_changesrFs r!rzConnection.total_changes8szz'''r"clK|j|jj|d{y7wrJ)rcrTenable_load_extensionrs r!rz Connection.enable_load_extension<s$mmDJJ<.dumpersm  JJ//10D))$/0%%d+  :;%%d+ s AA1A6Nziterdump completed unexpectedlyg{Gz?) r rcrM ensure_future get_nowaitrrr)warningsleep)r@rrtaskrrs` @r!rzConnection.iterdumpvs"G  mmF#$$S) *&0&;&;&=<   *99;KK ABmmD)))  * sX8B7A!B7A!B7B5B7!.B2B7B2)B,*B2/B71B22B7rmaing?pagesprogressrrr7rrrcKt|tr |j}|j|jj|||||d{y7w)z Make a backup of the current database to the target database. Takes either a standard sqlite3 or aiosqlite Connection object as the target. rN) isinstancerrTrcbackup)r@r7rrrrs r!rzConnection.backupsR fj )\\Fmm JJ       sA A AArJ)rr)rN)F)>__name__ __module__ __qualname__r sqlite3rintrrMAbstractEventLooprArGFuturerEpropertyrTstrr Rowr\rr_rcrfrrirlrqrrr[rtrwrKrXr}rrrrboolrrrsetterIsolationLeveltyperbytesrrrrrrAuthorizerCallbackrrrrfloatrrkr"r!rrNs 59 B 2 223w001   &$hw~~.$ w))  !3!C!HW[[S!  +9S$ %<=+DfDD/1">B$$$,Xc]$;$ $$>BJJ$,Xc]$;J '++ JJ>BLL$,Xc]$;L '++ LL$$$,Xc]$;$ $$$c$f$$ & QV  %( 08 IM  ,)))*#**+^+++&Xd^&&)8D>)d))'hw|4''*HeWc\$:*t**(s((EE$E==IHSM 12I7:I I DDTD*L#+,>#?*L *LX( c 2(\>B lG$6$667   8S#sOT$9:;       r"r@)r4r5databaser5rbc p|tdtdtjffd }t||S)z.connectorsG h $C % (//'*Ch-Cs-f--r")rr?rr)rr4r5rbr3s` ` r!rrs:  E  .w)). i 11r")1__doc__rMloggingrcollections.abcrrr functoolsrpathlibrqueuerr r threadingr typingr r rrrwarningsrcontextrr[r__all__rrr getLoggerr)rrr r.r%objectr-tuple_TxQueuer1rrrrkr"r!rsK>>++::# -sCc37<=g $'"FGHGNNCD w~~-D   uXgnn5xC7HHI JQ(Q