o NK&h/ @s@dZddlZddlZddlZddlmZmZmZddlm Z ddl m Z ddl m Z mZmZddlmZddlmZmZmZmZmZdd lmZd d lmZd d lmZgd ZedZ eedZ!dej"deddfddZ#dej"de$ddfddZ%e&Z'GdddeZ(ddddee)e fdeej*dede(fdd Z+dS)!z* Core implementation of aiosqlite proxies N) AsyncIterator GeneratorIterable)partial)Path)EmptyQueue SimpleQueue)Thread)AnyCallableLiteralOptionalUnion)warn)contextmanager)Cursor)connect Connectionr aiosqlite)DEFERRED IMMEDIATE EXCLUSIVEfutresultreturncC|s ||dSdS)z9Set the result of a future if it hasn't been set already.N)done set_result)rrr A/var/www/html/venv/lib/python3.10/site-packages/aiosqlite/core.pyrrecCr)ze,ddfd?d=Z*e dee-fd@dAZ.e.j+dBee-ddfdCdAZ.e dee/geffdDdEZ0e0j+dBee/gefddfdFdEZ0e defdGdHZ1d>e'ddfdIdJZ2dKefdLdMZ3dNegeefdOeddfdPdQZ4dNeddfdRdSZ5de6efdTdUZ7dVddWdXdYdZe8dejfd[ed\eeeeegdfd4ed]e9ddf d^d_Z:Z;S)drN connectoriter_chunk_sizelooprcsDtd|_d|_||_t|_||_|dur tdt dSdS)NTz8aiosqlite.Connection no longer uses the `loop` parameter) super__init___running _connection _connectorr _tx_iter_chunk_sizerDeprecationWarning)selfr%r&r' __class__r r!r).s zConnection.__init__cCsd|_|jtdS)NF)r*r- put_nowait_STOP_RUNNING_SENTINELr0r r r! _stop_runningAszConnection._stop_runningcCs|jdur td|jS)Nzno active connection)r+ ValueErrorr5r r r!_connFs zConnection._connsql parameterscCs |j||}|d|S)NzSELECT last_insert_rowid())r8executefetchoner0r9r:cursorr r r!_execute_insertMs zConnection._execute_insertcCs|j||}|SN)r8r;fetchallr=r r r!_execute_fetchallRszConnection._execute_fetchallc Cs |j}|tur dS|\}}ztd||}td||t||Wn!tyK}ztd||t ||WYd}~nd}~wwq)zV Execute function calls on a separate thread. :meta private: Tz executing %szoperation %s completedzreturning exception %sN) r-getr4LOGdebugget_loopcall_soon_threadsafer BaseExceptionr$)r0tx_itemfuturefunctionrr#r r r!runVs     zConnection.runcsR|jr|js tdt|g|Ri|}t}|j||f|IdHS)z8Queue a function with the given arguments for execution.zConnection closedN) r*r+r7rasyncioget_event_loop create_futurer-r3)r0fnargskwargsrKrJr r r!_executeps   zConnection._executecs`|jdur.zt}|j||jf|IdH|_W|Sty-|d|_w|S)z&Connect to the actual sqlite database.N) r+rMrNrOr-r3r,rHr6)r0rJr r r!_connect|s   zConnection._connectcCs||Sr@)startrT __await__r5r r r!rVs zConnection.__await__cs |IdHSr@r r5r r r! __aenter__s zConnection.__aenter__cs|IdHdSr@)close)r0exc_typeexc_valexc_tbr r r! __aexit__szConnection.__aexit__cst|||jjIdHS)zr5r r r!r>szConnection.cursorc||jjIdHdS)zCommit the current transaction.N)rSr8commitr5r r r!r^zConnection.commitcr])z"Roll back the current transaction.N)rSr8rollbackr5r r r!r`r_zConnection.rollbackcsj|jdurdSz$z ||jjIdHWn ty"tdwW|d|_dS|d|_w)z9Complete queued queries/cursors and close the connection.Nz+exception occurred while closing connection)r+rSr8rX ExceptionrDinfor6r5r r r!rXs    zConnection.closecs0|durg}||jj||IdH}t||S)z6Helper to create a cursor and execute the given query.N)rSr8r;rr=r r r!r;s  zConnection.executec$|durg}||j||IdHS)z/Helper to insert and get the last_insert_rowid.N)rSr?r0r9r:r r r!execute_insertzConnection.execute_insertcrc)z2Helper to execute a query and return all the data.N)rSrBrdr r r!execute_fetchallrfzConnection.execute_fetchallcs$||jj||IdH}t||S)z;Helper to create a cursor and execute the given multiquery.N)rSr8 executemanyrr=r r r!rhs zConnection.executemany sql_scriptcs"||jj|IdH}t||S)z4Helper to create a cursor and execute a user script.N)rSr8 executescriptr)r0rir>r r r!rjs zConnection.executescriptcs |jS)zInterrupt pending queries.)r8 interruptr5r r r!rks zConnection.interruptFname num_paramsfunc deterministiccs$|j|jj||||dIdHdS)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. )roN)rSr8create_function)r0rlrmrnror r r!rpszConnection.create_functioncC|jjSr@)r8in_transactionr5r r r!rrzConnection.in_transactioncCrqr@r8isolation_levelr5r r r!rurszConnection.isolation_levelvaluecC ||j_dSr@rtr0rvr r r!ru cCrqr@r8 row_factoryr5r r r!r{rszConnection.row_factoryfactorycCrwr@rzr0r|r r r!r{rycCrqr@r8 text_factoryr5r r r!rrszConnection.text_factorycCrwr@r~r}r r r!r rycCrqr@)r8 total_changesr5r r r!rrszConnection.total_changesc||jj|IdHdSr@)rSr8enable_load_extensionrxr r r!rz Connection.enable_load_extensionpathcrr@)rSr8load_extension)r0rr r r!rrzConnection.load_extensionhandlerncs||jj||IdHdSr@)rSr8set_progress_handler)r0rrr r r!rszConnection.set_progress_handlercrr@)rSr8set_trace_callback)r0rr r r!rrzConnection.set_trace_callbackcstfdd}|}t|} z}|dur!Wn"|VWntyA|r7tdYn t dIdHYnwq|IdHdS)z Return an async iterator to dump the database in SQL text format. Example:: async for line in db.iterdump(): ... csPzjD]}|qdWdSty'tddw)Nzexception while dumping db)r8iterdumpr3rarD exception)line dump_queuer0r r!dumper.s    z#Connection.iterdump..dumperTNziterdump completed unexpectedlyg{Gz?) rrSrM ensure_future get_nowaitrrrDwarningsleep)r0rrtaskrr rr!r"s&     zConnection.iterdumprmaing?pagesprogressrlrtargetrrrcs6t|tr |j}|j|jj|||||dIdHdS)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) isinstancerr8rSbackup)r0rrrrlrr r r!rLs zConnection.backupr@)rN)rr)F)<__name__ __module__ __qualname__r sqlite3rintrrMAbstractEventLoopr)r6propertyr8strr Rowr?rrBrLrSrTrrVrWr\rrr>r^r`rXr;rergrhrjrkboolrprrrusetterIsolationLeveltyper{bytesrrrrrrrrrfloatr __classcell__r r r1r!r-s                    . r@)r&r'databaser'rRc s2|dur tdtdtjffdd }t||S)z.connector)rr/rr)rr&r'rRr%r rr!rgs  r),__doc__rMloggingrcollections.abcrrr functoolsrpathlibrqueuerrr threadingr typingr r r rrwarningsrcontextrr>r__all__ getLoggerrDrFuturerrHr$objectr4rrrrr r r r!sB        ?