o i@sddlmZddlZddlmZddlmZddlmZddlmZddd,ddZddd-ddZ hdZ d.ddZ hdZ e e Zd d!hZd/d$d%Zd0d'd(Zd1d*d+ZdS)2) annotationsN)Iterable)List)Optional)Sequence)exc_infologgerlogging.LoggerprefixstrlevelintmsgargsobjectrOptional[BaseException]returnNonecGs"|j|||g|Rd|idS)Nr)log)rr r rrrrO/home/ubuntu/.local/lib/python3.10/site-packages/libsql_client/dbapi2/_utils.py log_prefix s"robjcGs>t|dd}|dur|d}t||||g|Rd|idS)N _log_prefixz: r)getattrr)rrr rrrr rrrlog_objs   r>     sql Optional[str]cCsLd}t|}||kr||}|tvr|d7}q|dkrV|d|krP||ddkrP|d7}||krE||dkrE|d7}||krE||dks7||krKdS|d7}q||dS|dkr|d|kr||ddkr|d7}||kr||dkr|d|kr||ddkrn|d7}||ksp||krdS|d7}q||dS||dSdS)Nr-r/*)len_lstrip_sql_whitespace_charsr!posend_poschrrr lstrip_sql&sD    r.>(),;[]"'r+r,cCsr|d|kr|S||}|d|kr7|d7}|||kr1|d7}||ks*|||kr- |S|d7}|d|ks|S)aFind the end of the quoted string starting at ``pos``. The returned position includes the position of the quote character, that matches the one at ``pos``. >>> def test_quote(ts): ... end = _iter_sql_get_quoted_end(ts, 0, len(ts)) ... return (end, ts[:end]) >>> test_quote("'abc'") (5, "'abc'") >>> test_quote('"abc "') (6, '"abc "') >>> test_quote("'abc' def") (5, "'abc'") >>> test_quote('"abc "def') (6, '"abc "') It also handles escaping by double quotes, see https://www.sqlite.org/faq.html#q14 >>> test_quote('"abc""def" ghi') (10, '"abc""def"') >>> test_quote('"abc "def" ghi') (6, '"abc "') r#rr*rrr_iter_sql_get_quoted_endis     r7 Iterable[str]ccs|rYt|pd}|s dSd}t|}||kr8||}|tvr+|dkr$nt|||}n |tvr0n|d7}||ks|dkrJ|d|V||d}n |dV|dd}|sdSdS)aYCalls lstrip_sql() to get the start of the next token and yield it. It will handle stop chars (;,[]()) as their own tokens, as well as handle double and single quotes, as well as quote escaping with \ >>> list(iter_sql_tokens("--COMMENT\nBEGIN X")) ['BEGIN', 'X'] >>> list(iter_sql_tokens( ... "CREATE TABLE x (id INTEGER /* COMMENT */, name TEXT)")) ['CREATE', 'TABLE', 'x', '(', 'id', 'INTEGER', ',', 'name', 'TEXT', ')'] >>> list(iter_sql_tokens("SELECT; INSERT; DELETE")) ['SELECT', ';', 'INSERT', ';', 'DELETE'] >>> print("\n".join(iter_sql_tokens( ... 'INSERT INTO t (a,b) VALUES("s -- c /* c */", \'x\')' ... ))) INSERT INTO t ( a , b ) VALUES ( "s -- c /* c */" , 'x' ) >>> list(iter_sql_tokens('SELECT"abc"')) ['SELECT', '"abc"'] rr#N)r.r(_iter_sql_quote_charsr7_iter_sql_stop_charsr*rrriter_sql_tokenss0!      r<Iterable[Sequence[str]]ccsBg}t|D]}|||dkr|Vg}q|r|VdSdS)zYields sql statements based on ";" tokens. >>> list(iter_sql_statements("BEGIN")) [['BEGIN']] >>> list(iter_sql_statements("BEGIN;INSERT INTO x;COMMIT")) [['BEGIN', ';'], ['INSERT', 'INTO', 'x', ';'], ['COMMIT']] r2N)r<append)r!stmttokenrrriter_sql_statementss   rA)rr r r r r rr rrrrrr)rr rrr r rr rrrrrr)r!r rr")r!r r+r r,r rr )r!r rr8)r!r rr=) __future__rloggingtypingrrrrrrr)r._iter_sql_delim_charsunionr;r:r7r<rArrrrs(       <  -<