|*^c@s*ddlZddlmZmZddlmZmZddlm Z ddl m Z ddl m Z ddlmZmZmZmZmZmZmZddlmZdd lmZd d/d YZd efd YZdefdYZdd0dYZdd1dYZdd2dYZdZdd3dYZ dd4dYZ!dd5dYZ"dd6dYZ#dd7d YZ$d!d8d"YZ%d#d9d$YZ&d%Z'd&d:d'YZ(d(e(fd)YZ)d*e(fd+YZ*d,d;d-YZ+d.Z,dS(<iN(tabspathtjoin(tsqlttokens(t FilterStack(ttokenize(tPipeline(tCommentt ComparisontKeywordtNamet PunctuationtStringt Whitespace(tmemoize_generator(tsplit_unquoted_newlinest _CaseFiltercBs#eZdZddZdZRS(cCs=|dkrd}n|dks'ttt||_dS(Ntuppertlowert capitalize(RRR(tNonetAssertionErrortgetattrtunicodetconvert(tselftcase((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyt__init__s  ccsGx@|D]8\}}||jkr4|j|}n||fVqWdS(N(tttypeR(RtstacktstreamRtvalue((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pytprocesssN(t__name__t __module__RRRR (((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRs tKeywordCaseFiltercBseZejZRS((R!R"tTR R(((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR#%stIdentifierCaseFiltercBs&eZejejjfZdZRS(ccs^xW|D]O\}}||jkrK|jddk rK|j|}n||fVqWdS(Nit"(RtstripR(RRRRR((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR ,s&(R!R"R$R R tSymbolRR (((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR%)stTruncateStringFiltercBseZdZdZRS(cCs%t|d|_t||_dS(Ni(tmaxtwidthRtchar(RR+R,((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR5sccsx|D]\}}|tjjjkr|d dkrN|dd!}d}n|dd!}d}t||jkrdj|||j |j|f}qn||fVqWdS( Nis''iu''iiu'u(R$tLiteralR tSingletlenR+RR,(RRRRRtinnertquote((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR 9s   (R!R"RR (((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR)3s t GetCommentscBseZdZdZRS(sGet the comments from a stackccs5x.|D]&\}}|tkr||fVqqWdS(N(R(RRRt token_typeR((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR Js (R!R"t__doc__R (((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR2Hst StripCommentscBseZdZdZRS(sStrip the comments from a stackccs5x.|D]&\}}|tkr||fVqqWdS(N(R(RRRR3R((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR Rs (R!R"R4R (((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR5Psccsd}t}tttf}x|D]\}}|rR|tkrpt}q%qpn|ttj|fkrpq%n|r|j||fstdfVnt}n||fV|}q%WdS(sIStrip the useless whitespaces from a stream leaving only the minimal onest N( RtFalset frozensetRR R tTruetNewlinet intersection(Rt last_typet has_spacet ignore_groupR3R((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pytStripWhitespaceXs     tIncludeStatementcBs/eZdZddedZedZRS(s&Filter that enable a INCLUDE statementt.i cCsI|dkrtdnt||_||_||_t|_dS(NisMax recursion limit reached(t ValueErrorRtdirpatht maxRecursivetraiseexceptionsR7tdetected(RRCt maxrecursiveRE((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRws    c csx|D]\}}|tkr@|jdkr@t|_qn>|jr~|tkr[qn|tjkrt|j|dd!}y&t |}|j }|j Wn1t k r}|j rntd|fVnXy#t|j|jd|j } Wn1tk r3}|j r!ntd|fVnXt}|jj| x|j|D] } | Vq]Wt|_qqn||fVqWdS(NtINCLUDEiiu-- IOError: %s u-- ValueError: %s (R RR9RFR R R(RRCtopentreadtclosetIOErrorRERR@RDRBRt preprocesstappendtrunR7( RRRR3Rtpathtftraw_sqlterrtfiltrttv((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR s@            (R!R"R4R7RRR (((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR@ts tStripCommentsFiltercBs#eZdZdZdZRS(cCs=|jdtj}|dkr9|jdtj}n|S(Ni(ttoken_next_by_instanceRRRttoken_next_by_typeR$(Rttlistttoken((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyt_get_next_comments cCs|j|}x|r|j|}|j|t}|j|t}|dk r|dk r|j r|j r|jtj dp|jtj d rt j tj d|j ||jj}x(|D] }|V||krtqqWdS(s0Yields all tokens up to token plus the next one.N(R~tflattent StopIteration(RRZtiteratortt((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyt_flatten_up_to_token%s   cCscdjtt|j|}|jd}t|t|j|j|j}||j S(NRri( RtmapRRt splitlinesR/R,R+R{R|(RRZtrawtlinet full_offset((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyt _get_offset.s!'cCs~|j|j|j|j}t|dkrad|_|_|j|j|j|j}nd|}tjtj|S(Niis ( R,R{R+R|R/RRcR$R (Rtspacetws((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pytnl5s ! c s:dfdd}|}t}x|r5jj|t}d}|r|jr||krjjj||d7}nt|}|r|jds|jdrj |}n3|j }|j |j |||d7}j||}q9WdS(NtFROMsSTRAIGHT_JOIN$sJOIN$tANDtORtGROUPtORDERtUNIONtVALUEStSETtBETWEENtEXCEPTtHAVINGcsj|tjdt}|r|jjdkrj|d}|r|jjdkrj|d}qn|S(NtregexRiR(ttoken_next_matchR$R R9RRR^(tiR(t _next_tokent split_wordsRY(sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyREs iis s ( RsSTRAIGHT_JOIN$sJOIN$RRRRRRRRRR( tsetR_R^R7RaRRdRtendswithR`Rtaddt insert_before( RRYtidxRZtaddedRfR|tuprevR((RRRYsd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyt _split_kwds@s*      $   cCsd}|j|tjjtjjf}x|r|j|j|t}|r||jr||j j |j|n|r|j }|j ||n|j|j|dtjjtjjf}q-WdS(Nii( RXR$R tDDLtDMLR_R^R7RaRRdRR(RRYRRZRfR((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyt_split_statementsas$  cCs9d|jjj}t|||j}||dS(Ns _process_%s(RmR!RRt_process_default(RRYRoRp((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRhoscCsu|jdtjd}y|j||jWntk rEnX|jd7_|j||jd8_dS(NitWHEREi(RR$R RRRBR{R(RRYRZ((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyt_process_wherets  cCsu|jdtjd}y|j||jWntk rEnX|jd7_|j||jd8_dS(NiRi(RR$R RRRBR{R(RRYRZ((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyt_process_having~s  cCs|jd}t}|rm|jtjjtjjfkrm|jd7_|jj d|j t }n|j |j dtjd}|j|7_|j|d| |r|jd8_n|j|8_dS(NiiR\tstmts(R`R7RR$R RRR{RtinsertRR9RRR R|R(RRYtfirsttindentedt num_offset((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyt_process_parenthesiss' cCst|j}t|dkr|jtj rt|djd}|jdkrid}n|j|t|j }|j |7_ x(|dD]}|j ||j qW|j |8_ n|j |dS(Niis (tlisttget_identifiersR/twithinRtFunctionRR,RRR|RRR(RRYt identifiersRRRZ((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyt_process_identifierlists% c Cs{t}d}|jd}|j|t|j}|j|7_x|jD]\}}|rt|dj d}t }|j|t|j}|j|7_qQn|dkr|d} n |d} |j | |j qQW|jd7_|j ||jd8_|dk r:|j|8_n|jdtjd} |j | |j |j|8_dS(NiitEND(R9RRRR/RR|t get_casesRRR7RRRRR$R ( RRYtis_firstRRt outer_offsettcondRttcondRZtend((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyt _process_cases0      cCsV|r|j|n|r,|j|ng|jD]}|j|^q9dS(N(RRRiRh(RRYRtkwdsRk((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRs cCst|tjr||_n|j|t|tjr|jdk rt|jjdrmd}nd}|j j dtj t j |n|j|kr||_qndS(Ns s i(t isinstanceRt StatementR~RhRRRRRRRcR$R (RRRjR((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR s    N(R!R"RRRRRRRRhRRRRRR9RR (((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRzs  !     tRightMarginFiltercBs,eZdZddZdZdZRS(iOcCs||_d|_dS(NRr(R+R(RR+((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRs ccs?x8|D]0}|jr_d|jkr_|jjdrFd|_q2|jjd|_n|jr|j|jkr|j|||j |_ nt |}t |jt ||j kr#t jd|j}|dk r|j}nd}tjtjd|V||_n|j|7_|VqWdS(Ns Rris^ +s %s(RaRRRRtis_groupRmt keep_togetherRhRRR/R+tretsearchRtgroupRRcR$R (RRRRRZtvalRbR{((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRhs$    "  cCsdS|j|||j|_(N(RhR(RRR((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR s((R!R"RRRhR (((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRs  t ColumnsSelectcBseZdZdZRS(s'Get the columns names of a SELECT queryccsrd}d}d}xY|D]Q\}}|tkr7qn|dkrg|tkrj|dkrjd}qjq|dkr5|dkr|r|Vnd}qj|dkrd}d}qj|tkr|d kr| r|r|Vnd}qj|tkrj|d kr |d7}n|d kr%|d8}n||7}qjq|dkr|tksY|tkrj|Vd}qjqqWdS( NiRrtSELECTiRitASiRtR\R](RR R R R (RRRtmodetoldValuet parenthesisR3R((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR s@                (R!R"R4R (((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRstSerializerUnicodecBseZdZRS(cCs5t|}t|}djd|D}|S(Ns css|]}|jVqdS(N(trstrip(t.0R((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pys =s(RRR(RRRjRtlinestres((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR :s  (R!R"R (((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR8scCs1d}x$|D]\}}|t|7}q W|S(NRr(R(Rtresultt_R((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pytTokens2UnicodeAst OutputFiltercBs,eZdZddZdZdZRS(RrRcCs|j||_d|_dS(Ni(tvarname_prefixtvarnametcount(RR((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRMscCs tdS(N(tNotImplementedError(RRRthas_nl((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRhQscCs|jd7_|jdkr7d|j|jf}n |j}tt|jjdk}|j|j|||_|S(Nis%s%d(RRR/RR'RRhR(RRRjRR((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR Ts $(R!R"RRRhR (((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRJs  tOutputPythonFiltercBseZdZRS(ccs|jdkr&tjtjdVntjtj|VtjtjdVtjtjdVtjtjdV|rtjtjdVntjtjdVx|D]}|jrid|j kritjtjdVtjtjdVtjtjdt |dVtjtjdV|j j ddd}|rtjtj|Vqqn*d|j kr|j j dd |_ ntjtj|j VqWtjtjdV|rtjtjd VndS( Nis R6t=R\t's 'is\'R]( RRRcR$R R tOperatortTextRaRR/tsplittreplace(RRRRRZtafter_lb((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRhas2 "(R!R"Rh(((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR`stOutputPHPFiltercBseZdZdZRS(t$ccs|jdkr&tjtjdVntjtj|VtjtjdV|rktjtjdVntjtjdVtjtjdVtjtjdVx2|D]*}|jrd|j krtjtjdVtjtjdVtjtj|VtjtjdVtjtjdVtjtjdVtjtjdV|j j ddd}|rtjtj|Vqqn*d|j kr|j j dd|_ ntjtj|j VqWtjtjdVtjtj d VdS( Nis R6RR&s ";s.=s\"t;( RRRcR$R R RRRaRRRR (RRRRRZR((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRhs6 (R!R"RRh(((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRstLimitcBseZdZdZRS(sdGet the LIMIT of a query. If not defined, return -1 (SQL specification for no LIMIT query) cCsod}t|}|jxL|D]D\}}|d8}|r#|tkr#|dkr#|d|dSq#WdS(NiitLIMITii(RtreverseR (RRRtindexR3R((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyR s   (R!R"R4R (((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyRscCs0t}|jt|jt||S(s6Function that return a compacted version of the stream(RRNR5R?(Rtpipe((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pytcompacts  ((((((((((((((-Rtos.pathRRtsqlparseRRR$tsqlparse.engineRtsqlparse.lexerRtsqlparse.pipelineRtsqlparse.tokensRRR R R R R tsqlparse.utilsRRRR#R%R)R2R5R?R@RVRlRzRRRRRRRRR(((sd/home/gitlab-runner/builds/8480fa44/0/bergerc/fluidmanager-web/art-framework/bin/sqlparse/filters.pyts8 4  U2&6 ,0