L-Ph%dZddlmZddlZddlZddlZddlmZmZddl m Z m Z ddl m Z mZmZddlmZddlmZGd d ZdS) z"SQLite abstraction for coverage.py) annotationsN)castAny)IterableIterator) auto_repr clipped_repr exc_one_line) DataError) TDebugCtlceZdZdZd%dZeZd&d Zd&d Zd'd Z d&d Z d(dZ e j d)d*dZd+d,dZd)d-dZd)d.dZd/dZd0d Zd1d"Zd2d#Zd$S)3SqliteDbaUA simple abstraction over a SQLite database. Use as a context manager, then you can use it like a :class:`python:sqlite3.Connection` object:: with SqliteDb(filename, debug_control) as db: with db.execute("select a, b from some_table") as cur: for a, b in cur: etc(a, b) filenamestrdebugr returnNonec>||_||_d|_d|_dS)Nr)rrnestcon)selfrrs Q/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/coverage/sqlitedb.py__init__zSqliteDb.__init__ s"    .2c|jdS|jdr"|jd|j t j|jd|_n2#t j$r }td|jd||d}~wwxYw|jdr*|jd|jd |j|j d d d tt d r%|j t j d| d| dddS)z2Connect to the db and do universal initialization.NsqlzConnecting to F)check_same_threadCouldn't use data file : z Connected to z as REGEXPc0tj||duS)N)research)txtpats rz#SqliteDb._connect..<sryc?R?RZ^?^rSQLITE_DBCONFIG_DEFENSIVEzpragma journal_mode=offzpragma synchronous=offT)fail_ok)rrshouldwritersqlite3connectErrorr create_functionhasattr setconfigr( execute_void)rexcs r_connectzSqliteDb._connect(s 8  F :  U # # A J  ?dm?? @ @ @ Yt}NNNDHH} Y Y YNdmNNNNOOUX X Y :  U # # P J  NT]NN$(NN O O O   1.^.^___ 77 8 8  H  15    3444 2DAAAAAs A((B7BBc|jq|jdkrh|jdr*|jd|jd|j|jd|_dSdSdS)z If needed, close the connection.N:memory:rzClosing z on )rrrr*r+closers rr7zSqliteDb.closeOs 8 DMZ$?$?z  '' O   !MDH!M!MDM!M!MNNN HNN   DHHH $?$?rc|jdkr6||jJ|j|xjdz c_|S)Nr)rr4r __enter__r8s rr;zSqliteDb.__enter__WsP 9>> MMOOO8''' H    Q  rc|xjdzc_|jdkr |jJ|j||||dS#t$rd}|jdr*|jdt|td|j d||d}~wwxYwdS)Nr:rrzEXCEPTION from __exit__: zCouldn't end data file r) rr__exit__r7 Exceptionrr*r+r r r)rexc_type exc_value tracebackr3s rr=zSqliteDb.__exit___s Q 9>> ]x+++!!(IyAAA  ] ] ]:$$U++VJ$$%TcARAR%T%TUUU R$- R RS R RSSY\\ ] >s9A C"ACCr parameters Iterable[Any]sqlite3.Cursorc|jdr(|rd|nd}|jd|| |jJ |j||S#t $r|j||cYSwxYw#t j$r}t|}|j dkri t|j d5}d}| t||krd }dddn #1swxYwYn#t $rYnwxYw|jdr*|jd t|td |j d ||d}~wwxYw) z2Same as :meth:`python:sqlite3.Connection.execute`.r with z Executing Nr6rbs&!coverage.py: This is a private formatzILooks like a coverage 4.x data file. Are you mixing versions of coverage?zEXCEPTION from execute: rr)rr*r+rexecuter>r,r.rropenreadlenr r )rrrBtailr3msgbad_filecov4_sigs r_executezSqliteDb._executeks  :  U # # 9.8@*J***bD J  7#777 8 8 8 Y8''' 9x''Z888 9 9 9x''Z88888  9 } Y Y Yc((C} ** dmT22h#L#==X778CC!G !Dz  '' Q   !OLD+D > D D DD DF D# F"D##A"FF Iterator[sqlite3.Cursor]c#K|||} |V|dS#|wxYw)zContext managed :meth:`python:sqlite3.Connection.execute`. Use with a ``with`` statement to auto-close the returned cursor. N)rQr7)rrrBcurs rrIzSqliteDb.executesLmmC,, III IIKKKKKCIIKKKKs 4A Fr)boolc |||dS#t$r|sYdSwxYw)zSame as :meth:`python:sqlite3.Connection.execute` when you don't need the cursor. If `fail_ok` is True, then SQLite errors are ignored. N)rQr7r )rrrBr)s rr2zSqliteDb.execute_voidsd   MM#z * * 0 0 2 2 2 2 2        s (, ==intc|||5}|jJ|j}dddn #1swxYwY|jdr|jd||S)z(Like execute, but returns the lastrowid.NsqldatazRow id result: )rI lastrowidrr*r+)rrrBrUrowids rexecute_for_rowidzSqliteDb.execute_for_rowids \\#z * * 'c=,,,E ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' :  Y ' ' : J  8u88 9 9 9 s 488tuple[Any, ...] | Nonecb|||5}t|}dddn #1swxYwYt|dkrdSt|dkr(ttt df|dSt d|dt|d)a6Execute a statement and return the one row that results. This is like execute(sql, parameters).fetchone(), except it is correct in reading the entire result set. This will raise an exception if more than one row results. Returns a row, or None if there were no rows. Nrr:.zSQL z shouldn't return  rows)rIlistrLrtuplerAssertionError)rrrBrUrowss r execute_onezSqliteDb.execute_ones\\#z * * c99D                t99>>4 YY!^^c3ha11 1 !Q!Q!QT!Q!Q!QRR Rs 377data list[Any]c|jdr|jdrdnd}|jd|dt|d||jdr5t |D]%\}}|j|dd |&|jJ |j||S#t$r|j||cYSwxYw) z6Same as :meth:`python:sqlite3.Connection.executemany`.rrZ:rGzExecuting many rFr`4dr)rr*r+rL enumerater executemanyr>)rrrffinalirows r _executemanyzSqliteDb._executemanys. :  U # # 9:,,Y77?CCRE J  SsSSCIISSESS T T Tz  ++ 9'oo99FAsJ$$%7%7%7%7%78888x### 38''T22 2 3 3 38''T22 2 2 2  3sC%DDc|t|}|r*|||dSdS)zUSame as :meth:`python:sqlite3.Connection.executemany` when you don't need the cursor.N)rarpr7)rrrfs rexecutemany_voidzSqliteDb.executemany_voidsGDzz  1   c4 ( ( . . 0 0 0 0 0 1 1rscriptc 6|jdrI|jdt |t |d|jJ|j|dS)z8Same as :meth:`python:sqlite3.Connection.executescript`.rz"Executing script with {} chars: {}dN) rr*r+formatrLr r executescriptr7)rrss rrwzSqliteDb.executescripts :  U # #  J  AHHF \&#66   x### v&&,,.....rcl|jJd|jS)z9Return a multi-line string, the SQL dump of the database.N )rjoiniterdumpr8s rdumpz SqliteDb.dumps0x###yy**,,---rN)rrrr rr)rr)rr)rrrBrCrrD)rR)rrrBrCrrS)rRF)rrrBrCr)rVrr)rrrBrCrrX)rrrBrCrr^)rrrfrgrrD)rrrfrCrr)rsrrr)rr)__name__ __module__ __qualname____doc__rr__repr__r4r7r;r=rQ contextlibcontextmanagerrIr2r]rerprrrwr|rRrrrrsy  3333 H%B%B%B%BN ] ] ] ]YYYYB%'          SSSSS$3333"1111 ////......rr)r __future__rrr#r,typingrrcollections.abcrrcoverage.debugrr r coverage.exceptionsr coverage.typesr rrRrrrs)("""""" ........@@@@@@@@@@))))))$$$$$$S.S.S.S.S.S.S.S.S.S.r