+hOdZddlZddlmZddlmZmZGddZGddZ Gd d e Z Gd d e Z Gd dee Z Gdde Z Gdde ZGdde ZGdde ZGdde ZGdde ZGdde ZGdde ZGdd e ZGd!d"e ZGd#d$e ZGd%d&e ZGd'd(ee ZGd)d*e ZGd+d,e ZGd-d.e ZGd/d0e Zy)1zFThis module contains classes representing syntactical elements of SQL.Ntokens)imt remove_quotesceZdZdZdZdZy)NameAliasMixinz'Implements get_real_name and get_alias.cr|jtjdf\}}|j|dS)7Returns the real name (object name) of this identifier..mT) real_name) token_next_byT Punctuation_get_first_name)selfdot_idx_s ?/var/www/html/venv/lib/python3.12/site-packages/sqlparse/sql.py get_real_namezNameAliasMixin.get_real_names:''1==#*>'? ##Gt#<<c|jtjdf\}}||j|dzdS|jtj\}}t |j dkDr||jd Syy) 2Returns the alias for this identifier or ``None``.ASr NT)keywordstreverse)rrKeywordr Whitespacelenr)rkw_idxkwrwss r get_aliaszNameAliasMixin.get_aliass''199d*;'<  >'' T'B B""Q\\"22 t{{ a BN'''5 5%3 rN)__name__ __module__ __qualname____doc__rr)rrrrs1= 6rrcReZdZdZdZdZdZdZdZdZ dZ dd Z d Z d Z d Zy )TokenzBase class for all other classes in this module. It represents a single token and has two instance attributes: ``value`` is the unchanged value of the token and ``ttype`` is the type of the token. )valuettypeparent normalized is_keywordis_group is_whitespacect|}||_||_d|_d|_|t j v|_|jt jv|_ |jr|j|_ y||_ y)NF) strr1r2r3r6rr#r5r$r7upperr4)rr2r1s r__init__zToken.__init__2seE     199,!ZZ1<<7+/??%++-rc|jSNr1rs r__str__z Token.__str__<s zzrc|j}|j}|jdr|jdrdnd}djddt |it S)N'"z"<{cls} {q}{value}{q} at 0x{id:2X}>idr.)_get_repr_name_get_repr_value startswithendswithformatrDlocals)rclsr1qs r__repr__zToken.__repr__Csi!!#$$&##C(U^^C-@Cc:3::%$x%!8% %rcPt|jjddS)Nr )r9r2splitr?s rrEzToken._get_repr_nameKs 4::$$S)"--rcrt|}t|dkDr|dddz}tjdd|S)Nz...z\s+ )r9r%resub)rraws rrFzToken._get_repr_valueNs8$i s8a<bq'E/Cvvfc3''rc#K|yw)zResolve subgroups.Nr.r?s rflattenz Token.flattenTs  sc8|j|u}|r||St|tr|f}|rN|jrtj ndfd|D}|D]}|j |jsyy|jr d|D}|j|vS)a,Checks whether the token matches the given arguments. *ttype* is a token type. If this token doesn't match the given token type. *values* is a list of possible values for this token. The values are OR'ed together so if only one of the values matches ``True`` is returned. Except for keyword tokens the comparison is case-sensitive. For convenience it's OK to pass in a single string. If *regex* is ``True`` (default is ``False``) the given values are treated as regular expressions. rc3JK|]}tj|ywr=)rUcompile).0vflags r zToken.match..ns:abjjD):s #TFc3<K|]}|jywr=)r:)r]r^s rr`zToken.match..vs0Aaggi0s)r2 isinstancer9r5rU IGNORECASEsearchr4)rr2valuesregex type_matchedpatternr_s @rmatchz Token.matchXszzU* v~  fc "YF $(OO2==D:6:F! >>$//2  ??00F&((rcX|j}|rt||ry|j}|ry)zReturns ``True`` if this token is within *group_cls*. Use this method for example to check if an identifier is within a function: ``t.within(sql.Function)``. TF)r3rb)r group_clsr3s rwithinz Token.withinzs0 &),]]Frc |j|k(S)z)r]rzs rr`z$TokenList.__str__..s?uu{{?s)joinrYr?s rr@zTokenList.__str__sww? ???rc,t|jSr=)iterrr?s r__iter__zTokenList.__iter__sDKK  rc |j|Sr=r)ritems r __getitem__zTokenList.__getitem__s{{4  rc,t|jSr=)typer*r?s rrEzTokenList._get_repr_namesDz"""rc t|j}t|jD]\}}|j}|j } ||dz k(} | rdnd} | j dr| j drdnd} tdjd it||js|||ks| rd nd } |j||dz||| zy) zPretty-print the object tree.rz`- z|- rBrCz${_pre}{pre}{idx} {cls} {q}{value}{q})fileNz z| r.) r%r enumeraterErFrGrHprintrIrJr6 _pprint_tree)r max_depthdepthf_pre token_countidxrzrKr1lastprerL parent_pres rrzTokenList._pprint_trees$++& #DKK0 OJC&&(C))+E;?+D%UC'',1D#A 86&"H&,- /~~9#4 8I&*U ""9eaiD: Ac#PK|jD]}|js|ywr=)rr6rs r get_sublistszTokenList.get_sublistss%[[ E~~  s&&c|jSr=rr?s r_groupable_tokenszTokenList._groupable_tokenss {{rc |yt|ttfs|f}|r|Jt|dz dd}n#|t |j }t||}|D](}|j |}|D]}||s ||fccS*y)znext token that match functionsNr rONN)rblisttupleranger%r) rfuncsstartrr"indexesrrzfuncs r_token_matchingzTokenList._token_matchings =%$/HE ; ;EAIr2.G{$++&E3'G &CKK$E &;:% & & rc:fd}|j|dS)zReturns the first child token. If *skip_ws* is ``True`` (the default), whitespace tokens are ignored. if *skip_cm* is ``True`` (default: ``False``), comments are ignored too. crxr |jxs$xr t|tjt SN)rir7rrCommenttkskip_cmskip_wss rmatcherz&TokenList.token_first..matcher8 5R%5%5I#GB!))w(GJ Jrrr)rrrrs `` r token_firstzTokenList.token_firsts! J##G,Q//rcB|dz }|jfd||S)Nrc t|Sr=)r)rrr rs rz)TokenList.token_next_by.. ss2q!Q/?rr)rrr rrrs ``` rrzTokenList.token_next_bys" q##$?cJJrct|ttfs|fn|}|Dcgc]fd }}|j||Scc}w)Nc| Sr=r.)rrs rrz.TokenList.token_not_matching..s RLr)rbrrr)rrrrs `rtoken_not_matchingzTokenList.token_not_matching sE *54- @e5:;T(;;##E3//.matcher)rrr!r)rrrrrrs `` rrzTokenList.token_nexts6 ; q J##GS(#CCrct|tr|n|j|}||j|dj |zS)zReturn list index of token.N)rbint token_indexrindex)rrzrs rrzTokenList.token_index.s>#E3/T5E5Ee5Lt{{56*00777rcj|}|j|}||z}|r^t||rR|j|dz|}|} | jj||j|dz|=t|| _n.|j||}||} | g|j|||| _|D] } | | _ | S)z+Replace tokens by an instance of *grp_cls*.r)rrbextendr9r1r3) rgrp_clsrr include_endr start_idxend_idx subtokensgrprzs r group_tokenszTokenList.group_tokens3s  I& # j0 IM':IC JJ  i ( IM'12E CI Ig6I)$C.1UDKK ' *CJ EEL  rct|ts|j|}||_|jj ||y)zInserts *token* before *where*.N)rbrrr3rinsert)rwhererzs r insert_beforezTokenList.insert_beforeQs7%%$$U+E  5%(rct|ts|j|}|j||\}}||_||j j |y|j j||y)zInserts *token* after *where*.rN)rbrrrr3rappendr)rrrzrnidxnext_s r insert_afterzTokenList.insert_afterXsd%%$$U+EooeWo= e = KK  u % KK  tU +rc&|jduS)z(Returns ``True`` if an alias is present.N)r)r?s r has_aliaszTokenList.has_aliascs~~t++rcy)rNr.r?s rr)zTokenList.get_aliasgrcF|jxs|jS)aReturns the name of this identifier. This is either it's alias or it's real name. The returned valued can be considered as the name under which the object corresponding to this identifier is known within the current statement. )r)rr?s rget_namezTokenList.get_nameks~~74#5#5#77rcy)r Nr.r?s rrzTokenList.get_real_nametrrc|jtjdf\}}|j|\}}|t |j SdS)ztReturn name of the parent object if any. A parent object is identified by the first occurring dot. r r N)rrrrrr1)rrrprev_s rget_parent_namezTokenList.get_parent_namexsO ''1==#*>'? ??7+5-2->}U[[)HDHrc|r|j|dn |j}|r t|n|}tjtjtj j g}|r|jtj|D]c}|j|vrt|jcSt|ttfs?|r|jcS|j!cSy)z/Returns the name of the first token with a nameN)rreversedrNameWildcardStringSymbolrr#r2rr1rb IdentifierFunctionrr)rrr"rrrtypesrzs rrzTokenList._get_first_names'*ST"t{{%,&!&QXX__5  LL # PE{{e#$U[[11EJ#9:09u**,Ou~~?OO  Prr=)NrNr})rNF)TF)NNNrON)TFF)r)T)NFFF)"r*r+r,r-rtr;r@rrrErrrYrpropertyrrrrrrrrrrrrrr)rrrr __classcell__)r{s@rrvrvs I @!!#O$  , 0K0 3ED 8 =A!<) ,,8IAF"'PrrvceZdZdZdZy) StatementzRepresents a SQL statement.c.|jd}|y|jtjjtjj fvr |j S|jtjjk(r|j|}|z|j|d\}}t|ttfrK|j|d\}}|3|jtjjk(r |j S|zy)aMReturns the type of a statement. The returned value is a string holding an upper-cased reprint of the first DML or DDL keyword. If the first token in this group isn't a DML or DDL keyword "UNKNOWN" is returned. Whitespaces and comments at the beginning of the statement are ignored. T)rUNKNOWNr) rr2rr#DMLDDLr4CTErrrbrIdentifierList)rrztidxs rget_typezStatement.get_types   . = [[QYY]]AIIMM: :## # [[AIIMM )##E*D""oodDoA eej.%AB"&//$/"EKD%( % qyy}} <$///"rN)r*r+r,r-rr.rrrrs %"rrc(eZdZdZdZdZdZdZy)rzNRepresents an identifier. Identifiers may have aliases or typecasts. cP|jtj\}}|duS)z7Return ``True`` if this identifier contains a wildcard.rN)rrr)rrrzs r is_wildcardzIdentifier.is_wildcards(%% %35D  rc|jtjdf\}}|j|d\}}|r |jSdS)z ?Fhhoot#GyyNNHrrcXeZdZdZej dfZej dfZedZ y) ParenthesiszTokens between parenthesis.()c |jddSNrrOrr?s rrzParenthesis._groupable_tokens{{1R  rN r*r+r,r-rrrrrrr.rrrrs3% ]]C FmmS G !!rrcXeZdZdZej dfZej dfZedZ y)rzTokens between square brackets[]c |jddSr!rr?s rrz SquareBrackets._groupable_tokensr"rNr#r.rrrrs3( ]]C FmmS G !!rrceZdZdZy) Assignmentz An assignment like 'var := val;'Nr*r+r,r-r.rrr)r)s*rr)cHeZdZdZej dfZej dfZy)Ifz7An 'if' clause with possible 'else if' or 'else' parts.IFzEND IFNr*r+r,r-rr#rrr.rrr,r,s A YY_Fii!Grr,cHeZdZdZej dfZej dfZy)Forz A 'FOR' loop.)FORFOREACHzEND LOOPNr.r.rrr0r0 s" YY* *Fii#Grr0c0eZdZdZedZedZy) Comparisonz/A comparison used for example in WHERE clauses.c |jdSNrrr?s rleftzComparison.lefts{{1~rc |jdS)NrOrr?s rrightzComparison.rights{{2rN)r*r+r,r-rr7r9r.rrr4r4s+9 rr4ceZdZdZdZy)rz A comment.c|jxr4|jdjtjjk(Sr6)rr2rr Multiliner?s r is_multilinezComment.is_multiline!s.{{Jt{{1~33qyy7J7JJJrN)r*r+r,r-r=r.rrrrs KrrcHeZdZdZej dfZej dfZy)WherezA WHERE clause.WHERE) ORDER BYzGROUP BYLIMITUNIONz UNION ALLEXCEPTHAVING RETURNINGINTONr.r.rrr?r?%s& YY Fii''Grr?cHeZdZdZej dfZej dfZy)HavingzA HAVING clause.rE)rArBNr.r.rrrIrI-s" YY Fii..GrrIcPeZdZdZej dfZej dfZddZy)CasezAA CASE statement with one or more WHEN and possibly an ELSE part.CASEENDcd}d}g}|}|jD]G}|jtjdr%|r|jtj vrD|jtjdr|j ggf|}n{|jtjdr|}nX|jtjdr|j dgf|}n"|jtjdrd}|r|s|j ggf||k(r|d d j |*||k(s1|d dj |J|S) zeReturns a list of 2-tuples (condition, value). If an ELSE exists condition is None. rr rLWHENTHENELSENrMrOr)rrirr#r2r$r)rr CONDITIONVALUEretmoderzs r get_caseszCase.get_cases8s-  [[ )E{{199f-U[[ALL8QYY/ B8$ QYY/QYY/ D":&QYY.C B8$y B !!%(B !!%(? )D rNrs) r*r+r,r-rr#rrrVr.rrrKrK3s&K YY FiiG-rrKceZdZdZdZy)rzA function or procedure call.c|jd}|jD]O}t|tr|jcSt |t t ftjsL|gcSgS)zReturn a list of parameters.rO)rr) rrbrr rrrrLiteral)r parenthesisrzs rget_parameterszFunction.get_parametersksakk"o  '' !E%0,,..Ux4 By  !  rN)r*r+r,r-r[r.rrrrhs 'rrcHeZdZdZej dfZej dfZy)BeginzA BEGIN/END block.BEGINrMNr.r.rrr]r]vs! YY FiiGrr]ceZdZdZy) OperationzGrouping of operationsNr*r.rrr`r`|s rr`ceZdZdZy)ValueszGrouping of valuesNr*r.rrrbrbsrrbceZdZdZy)CommandzGrouping of CLI commands.Nr*r.rrrdrds#rrd)r-rUsqlparserrsqlparse.utilsrrrr0rvrrrrrrr)r,r0r4rr?rIrKrr]r`rbrdr.rrrgs*M -66.kk\{P{P|% %P))> Y O9O!)!!Y!++"" $)$  KiK'I'/Y/ 292j ~y I ! !Y$i$r