Pi&dZddlmZddlZejdejejzZejdZejdZ e hdZ e hdZ dd Z ddZddZddZddZe e dddZdS)zsSQL statement classification for Dippy. Provides dialect-agnostic detection of read-only vs write SQL statements. ) annotationsNa '(?:[^']*'')*[^']*' # Single-quoted string ('' for escape) | "(?:[^"]*"")*[^"]*" # Double-quoted string ("" for escape) | `[^`]*` # Backtick identifier (MySQL) | \[[^\]]*\] # Bracket identifier (SQL Server) | --[^\n]* # Single-line comment | /\*.*?\*/ # Block comment z\s+z [A-Za-z_]\w*>SHOWSELECTEXPLAINDESCRIBE> DROPALTERGRANTMERGECREATEDELETEINSERTREVOKEUPDATEREPLACETRUNCATEsqlstrreturnc8td|S)zBRemove string literals, quoted identifiers, and comments from SQL. )_QUOTED_PATTERNsub)rs //root/projects/gits/Dippy/src/dippy/core/sql.py _strip_quotedr-s   sC ( ((boolcht|}|d}|dkrdS||dzd}|}|sdStd|DrGt |D]5\}}|rd||dzdvrdS,|dkrdS6dSdS)z@Check if SQL contains multiple statements (semicolon-separated).;FNc3"K|] }|dkV dS)rN).0cs r z+_has_multiple_statements..As& , ,18 , , , , , ,rT)rfindstripall enumerateisspace)rstripped first_semiafterafter_strippedir%s r_has_multiple_statementsr12sS!!Hs##JRu Z!^%% &E[[]]N u , ,^ , , ,,, e$$  DAqyy{{ %A.((44)cttu 4rposintcht||}|r|n|S)zSkip whitespace at position.)_WHITESPACE_PATTERNmatchend)rr2ms r_skip_whitespacer9Rs.!!#s++A 155777S rc$t|}d}||krt||}||krn||dkrJd}|dz }||kr:|dkr4||dkr|dz }n||dkr|dz}|dz }||kr|dk4d}s||dkr|dz }d}t||}|rO|}|r%|}|dkrd}n|d kr |S|dz }||k|S) zFSkip over CTE definitions (name AS (...), ...) to find main statement.T(r!r)F,AS RECURSIVElenr9_KEYWORD_PATTERNr6groupupperr7)rr2length expect_asdepthr8kws r _skip_cterIXs` XXFI ,,sC(( &==  s8s??E 1HC,,5199s8s??QJEEX__QJEq ,,5199 I  s8s?? 1HCI   " "3 , , ""B eegg:: %II;&&J qE ,,F JrcDt|}||krt||}||krnst||}|rK|}|dkrdS|dkrdS|}|dz }||kdS)zFCheck if SELECT statement contains INTO (making it a write operation).INTOTFROMFr!r@)rr2rEr8rHs r_check_select_intorMsXXF ,,sC(( &==   " "3 , ,  ""BV||tV||u%%''C  q ,, 5r)extra_readonly extra_writerNfrozenset[str]rO bool | Nonec0t|rdSt|}t|z}t|z}d}|t |krt ||}|t |krnt ||}|sdS| }|dkr#t|| }|dkr&t|| rdSdS||vrdS||vrdSdSdS)a Determine if a SQL statement is read-only. Args: sql: The SQL statement to analyze. extra_readonly: Additional keywords to treat as read-only (dialect-specific). extra_write: Additional keywords to treat as write operations (dialect-specific). Returns: True: Statement is definitely read-only (safe to auto-approve). False: Statement is definitely a write operation. None: Unknown or ambiguous (caller should prompt user). Notes: - Multiple statements (semicolon-separated) return None. - CTEs (WITH ... AS) are handled by analyzing the main statement. - Side-effect functions (e.g., SQLite's writefile) are NOT detected. NrWITHrFT) r1r_READONLY_KEYWORDS_WRITE_KEYWORDSrAr9rBr6rCrDrIr7rM) rrNrOr,readonly_keywordswrite_keywordsr2r8rHs ris_readonly_sqlrXs*0 $$tS!!H*^;${2N C H  x-- #h--     " "8S 1 1 4 WWYY__   <<Haeegg..C  >>!(AEEGG44 u4 " " "4   5t 4r)rrrr)rrrr)rrr2r3rr3)rrr2r3rr)rrrNrPrOrPrrQ)__doc__ __future__rrecompileVERBOSEDOTALLrr5rB frozensetrTrUrr1r9rIrMrXr#rrr`sS #""""" "*J  !bj((2:o..YHHHII)   ")))) @!!!! ''''T2&/Y[["+)++ 77777777r