gpdZddlZdZdZdZdZdZdZejd Z d Z d Z ddZ ddZ ddZdZdZddZddZdS)a String split and join operations for dealing with literals and identifiers. Notably, the functions in this module are intended to be used for simple use-cases. It attempts to stay away from "real" parsing and simply provides functions for common needs, like the ability to identify unquoted portions of a query string so that logic or transformations can be applied to only unquoted portions. Scanning for statement terminators, or safely interpolating identifiers. All functions deal with strict quoting rules. Nc.|ddS)z#Replace every instance of ' with '''''replacetexts G/granary/edev/quantum/lib/python3.11/site-packages/postgresql/string.pyescape_literalr  S$c:d|ddzdzS)z5Escape the literal and wrap it in [single] quotationsrrrrs r quote_literalr  dll3%%%++r c.|ddS)z#Replace every instance of " with """""rrs r escape_identrr r c|oB|d o'|dd S)Nr_a) isdecimalrisalnumrs r needs_quotingrsB  R$q'++--- R$,,sC2H2H2P2P2R2RSSr c:d|ddzdzS)z;Replace every instance of ' with *and* place ' on each endrrrrs r quote_identr!rr cBt|rt|n|S)zu If needed, replace every instance of '"' with '""' *and* place '"' on each end. Otherwise, just return the text. )rrrs r quote_ident_if_neededr%s$ +400: Dd:r a(?xu) E'(?:''|\\.|[^'])*(?:'|$) (?# Backslash escapes E'str') | '(?:''|[^'])*(?:'|$) (?# Regular literals 'str') | "(?:""|[^"])*(?:"|$) (?# Identifiers "str") | (\$(?:[^0-9$]\w*)?\$).*?(?:\1|$) (?# Dollar quotes $$str$$) c#|Kd}t}|j|}|}|}|||V|d}|t |}|}|}n+d}||} | dkrd}d}n| x}}|||z ||ks-||z |kr2|||t |zdfVdS|||t |z||z fV|}|}|}||dVdS)a split the string up by into non-quoted and quoted portions. Zero and even numbered indexes are unquoted portions, while odd indexes are quoted portions. Unquoted portions are regular strings, whereas quoted portions are pair-tuples specifying the quotation mechanism and the content thereof. >>> list(split("select $$foobar$$")) ['select ', ('$$', 'foobar'), ''] If the split ends on a quoted section, it means the string's quote was not terminated. Subsequently, there will be an even number of objects in the list. Quotation errors are detected, but never raised. Rather it's up to the user to identify the best course of action for the given split. rNEE'r)quote_rescannersearchstartgroupslenend) r lastendrescanmatchdqendoffquoter)qs r splitr23s&  4   WU[[]] "#### ||~~a"^ GG6 5 33 6 EKKMM13hh E CCOC%  %))++f UYY[[ ()S00 YY[[6U[[]]** ekkmmCJJ.//0 11115 ekkmmCJJ.uyy{{6/AAB CCCC IIKK' ++--%; @ WXXr cd}d}t|}d}|D]A}|||zz } t|\}}|||zz }|dkrd}n|}1#t$rYnwxYw|S)z catenate a split string. This is needed to handle the special cases created by pg.string.split(). (Run-away quotations, primarily) Fr"r)iternext StopIteration) splitted_itersquotediendqxr1qtexts r unsplitr?ls   -    Qtax-! 1gg81eE >14ii DD D    55  s%A  AA.c|dz}t|}d}t|}|}||vr|||Sg} t| |kr|||} | dkrn|||| } | dkr||| dz} | dkr>||| | |z| krn"||| dz} | dk>d} n@||| dz} | dkrn ||| dz| } | dk| dkrn6| ||| | dz}t| |k| ||d| S)z split the string on the seperator ignoring the separator in quoted areas. This is only useful for simple quoted strings. Dollar quotes, and backslash escapes are not supported. rrAr N)r(r2findappend) r r0sepmaxsplitescapeesclenoffsettlr)lnextsep nextquotes r split_usingrOs !) f++  $ii   C " "" 1vv IIc6 " "' ]]yy009 b %Q00I r// &)Yv%566)CC 5)a-00Y r//W iiY]++G"}}  %Q88I+ b,mm 88D !!! aK6G 1vvH$vww- r ,rcg}t||||D]}|}|drW|dst d|||ddddt|rt d||||S)z@ Split a series of identifiers using the specified separator. )rFrGrz!unterminated identifier quotationr rArz+non-ident characters in unquoted identifier) rOstrip startswithendswith ValueErrorrErrlower)r rFr0rGnrr=s r split_identrXs  dE B B BQggii!\\# **S//* (!  IIa"goodC(())))Q 11   99QWWYY r c&t||dS)z4 Call to .split_ident() with a '.' sep parameter. r@)rGrF)rX)r rGs r split_qnamerZs DXS9999r c@dd|DS)z-Quote the identifiers and join them using '.'r@c,g|]}t|S)r.0r=s r zqname..s///Q+a..///r joinargss r qnameres# //$///000r c@dd|DS)Nr@c,g|]}t|Sr])rr^s r r`z#qname_if_needed..s!999q'**999r rarcs r qname_if_neededrhs# 99D999:::r ;c#Ktt|}g}|D]}||}t|dkr||n:||d|V|ddD]}|fV|dg} |t |#t $rYnwxYw|r|VdSdS)aF Given SQL, safely split using the given separator. Notably, this yields fully split text. This should be used instead of split_sql_str() when quoted sections need be still be isolated. >>> list(split_sql('select $$1$$ AS "foo;"; select 2;')) [['select ', ('$$', '1'), ' AS ', ('"', 'foo;'), ''], (' select 2',), ['']] rCrr rAN)r5r2r(rEr6r7)sqlrFr;curpartsectionsr=s r split_sqlros %**    T ZZ__(]]Q::d::hqk 999 QrTNq $JJJJ 2,3 ::d1gg    55   )))))  s"C CCc#TKt||D]}t|VdS)z Identical to split_sql but yields unsplit text. >>> list(split_sql_str('select $$1$$ AS "foo;"; select 2;')) ['select $$1$$ AS "foo;"', ' select 2', ''] )rFN)ror?)rkrFr=s r split_sql_strrqsC Cs # # #Q r )r@rA)rPrrA)rA)ri)__doc__r+r rrrrrcompiler#r2r?rOrXrZrerhrorqr]r r rts=      ,,,   TTT,,,;;; 2:  777r   ,6 6 6 6 p    0:::: 111;;;    :r