+j<PddlZddlZddlZddlmZddlmZmZGddejZ GddejZ Gdd ejZ Gd d ejZ Gd d ejZ GddejZGddejZGddejZGddejZGddejZGddejZGddejZGddejZGddeZd efd!ZdS)"N)override)ComfyExtensioniocveZdZedejfdZedejjde dej fdZ dS) StringFormatreturnc tjtjdt t jd}tjdddddgd tjd | tj d d dgtj gS)Nvaluer)inputnamesminrz Format Texttextstringformatz\Same as Python's string format method. Supports all of Python's format options and features.values)templatef_stringz{a}T)default multiline)node_id display_namecategorysearch_aliases descriptioninputsoutputs) rAutogrow TemplateNamesAnyTypeInputlistrascii_lowercaseSchemaStringOutput)clsautogrows 9/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_string.py define_schemazStringFormat.define_schema s;,,*""7++v-..-   y"&$h/v !!(X!>>  ETJJ    ""    rrc@tj|jdi|S)N)r NodeOutputr)r&rrs r(executezStringFormat.execute s&}_X_66v66777r*N) __name__ __module__ __qualname__ classmethodrr#r)rTypestrr-r.r,r*r(rr sz bi   [ *8[%8148 888[888r*rc:eZdZedZedZdS)StringConcatenatec 6tjdgdddtjddtjddtjd d d gtjg S)Nr6) concatenatez text concatz join textz merge textzcombine stringsz string concatz append textz combine textzConcatenate Textrstring_aTrstring_b delimiterF)rrrrrrrrrr#r$r r%r&s r(r)zStringConcatenate.define_schema(sy'XXX+  d;;  d;;  ubII    ""    r*cTtj|||fSN)rr-join)r&r9r;r<s r(r.zStringConcatenate.execute9s#}Y^^Xx,@AABBBr*Nr/r0r1r2r)r.r,r*r(r6r6'sM  [ CC[CCCr*r6c:eZdZedZedZdS)StringSubstringc ,tjdgdddtjddtjdtjd gtjg S) NrF) substringz extract textz text portion SubstringrrTr:startendr>rr#r$r Intr%r@s r(r)zStringSubstring.define_schema?sy%HHH$ D99 W%% U##    ""    r*c:tj|||SrB)rr-)r&rrJrKs r(r.zStringSubstring.executePs}VE#I.///r*NrDr,r*r(rFrF>sH  [ 00[000r*rFc:eZdZedZedZdS) StringLengthc tjdgdddtjddgtjd g S) NrP)zcharacter countz text sizez string lengthz Text LengthrrTr:lengthrr>rLr@s r(r)zStringLength.define_schemaVsay"LLL& D99 8 44    r*cDtjt|SrB)rr-len)r&rs r(r.zStringLength.executees}S[[)))r*NrDr,r*r(rPrPUsH   [  **[***r*rPc:eZdZedZedZdS) CaseConverterc tjdgdddtjddtjdgd  gtjg S) NrW)zcase converterz text case uppercase lowercase capitalizezConvert Text CaserrTr:mode) UPPERCASErZ Capitalize Title Caseoptionsr>rr#r$r Combor%r@s r(r)zCaseConverter.define_schemaks{y#bbb, D99v/e/e/eff    ""    r*c|dkr|j}nD|dkr|j}n.|dkr|j}n|dkr|j}n|}t j|S)Nr]rZr^r_)upperlowerr[titlerr-r&rr\results r(r.zCaseConverter.execute{s ;  !V\^^FF [ !V\^^FF \ ! !&V&((FF \ ! !!V\^^FFF}V$$$r*NrDr,r*r(rWrWjsH   [   % %[ % % %r*rWc:eZdZedZedZdS) StringTrimc tjdgdddtjddtjdgd  gtjg S) Nrk)trimzclean whitespacezremove whitespacez remove spacesstripz Trim TextrrTr:r\)BothLeftRightr`r>rbr@s r(r)zStringTrim.define_schemas{y eee$ D99v/H/H/HII    ""    r*c|dkr|j}n.|dkr|j}n|dkr|j}n|}tj|S)Nrorprq)rnlstriprstriprr-rhs r(r.zStringTrim.executes_ 6>>!V\^^FF V^^"V]__FF W__"V]__FFF}V$$$r*NrDr,r*r(rkrksH   [   % %[ % % %r*rkc:eZdZedZedZdS) StringReplacec 4tjdgdddtjddtjddtjd dgtjg S) Nrv)replacezfind and replace substitutez swap textz Replace TextrrTr:findrxr>r?r@s r(r)zStringReplace.define_schemasy#UUU' D99 $77  T::    ""    r*cHtj|j||SrB)rr-rx)r&rrzrxs r(r.zStringReplace.executes }^V^D'::;;;r*NrDr,r*r(rvrvsH  [ <<[<<rr#r$r Booleanr%r@s r(r)zStringContains.define_schemasy$KKK( D99  t<<   !14$ OO  !!z!::    r*c~|r||v}n#||jv}tj|SrB)rfrr-)r&rrHrrs r(r.zStringContains.executesB  ; F*HH ((LFLNN:H}X&&&r*NrDr,r*r(r}r}sH  [ ''['''r*r}c:eZdZedZedZdS) StringComparecztjdgdddtjddtjddtjd gd  tjd dd gtjgS)Nr)comparez text matchz string equalsz starts withz ends withz Compare Textrr9Tr:r;r\) Starts With Ends WithEqualr`rrr>)rr#r$r rcrr%r@s r(r)zStringCompare.define_schemasy#aaa'  d;;  d;;v/T/T/TUU   !14$ OO  !!##    r*cT|r|}|}n(|}|}|dkrtj||kS|dkr'tj||S|dkr'tj||SdS)Nrrr)rfrr- startswithendswith)r&r9r;r\rabs r(r.zStringCompare.executes  !AAA  A  A 7??=a(( ( ] " "=a11 1 [ =A// /! r*NrDr,r*r(rrsH  [ " 0 0[ 0 0 0r*rc:eZdZedZedZdS) RegexMatchctjdgdddtjddtjddtjd dd tjd d d tjd d d gtjdgS)Nr)z regex matchregexz pattern matchz text containsz string matchz Match TextrrTr: regex_patterncase_insensitiverrFdotallmatchesrSr>rr@s r(r)zRegexMatch.define_schemasy eee% D99 4@@   !3TD QQ   ed KK   54 HH  !!y!99    r*cd}|r|tjz}|r|tjz}|r|tjz} tj|||}|du}n#tj$rd}YnwxYwt j|S)NrF)re IGNORECASE MULTILINEDOTALLsearcherrorrr-) r&rrrrrflagsmatchris r(r.zRegexMatch.executes  # R] "E  " R\ !E   RY E ImVU;;E$&FFx   FFF }V$$$sAA&%A&NrDr,r*r(rrsH  [ $%%[%%%r*rc:eZdZedZedZdS) RegexExtractcDtjdgdddtjddtjddtjd gd  tjd dd tjddd tjddd tjdddddgtjgS)Nr)z regex extractrzpattern extractz text parserz parse textz Extract TextrrTr:rr\) First Match All Matches First Group All Groupsr`rrrFr group_indexrd)rr maxrr>)rr#r$r rcrrMr%r@s r(r)zRegexExtract.define_schema+sy"eee' D99 4@@v/j/j/jkk   !3TD QQ   ed KK   54 HH ]A1#PT UU   ""    r*cd}d} |r| tjz} |r| tjz} |r| tjz} |dkr3tj||| } | r| d} nod} nk|dkrmtj||| } | rRt| dtr!| d| D} n| | } nd} n|dkrVtj||| } | r;t| |kr| |} nd} n|dkrtj ||| } g} | D]c} | rMt| |kr(| | |d| | } nd} n#tj$rd} YnwxYwtj| S) N rrr=rcg|] }|d S)rr,).0ms r( z(RegexExtract.execute..Xs5L5L5Lqad5L5L5Lr*rr)rrrrrgroupfindall isinstancetuplerCrUgroupsfinditerappendrrr-)r&rrr\rrrrjoin_delimiterrrrirresultss r(r.zRegexExtract.execute@s2  # R] "E  " R\ !E   RY E$ }$$ -?? "[[^^FFFF&&*]FEBB !'!*e44>!/!4!45L5LG5L5L5L!M!M!/!4!4W!=!=FF&& -?? S00K??"[[55FFFF%%+mVUCC$AAE||~~A#ellnn*=*=*L*Lu{{;'?'?@@@',,W55x   FFF }V$$$sF$GG21G2NrDr,r*r(rr*sH  [ (1%1%[1%1%1%r*rc<eZdZedZeddZdS) RegexReplacecNtjdgddddtjddtjd dtjd dtjd ddd tjd ddd tjdddddtjdddddddgtjgS)Nr)z regex replacerzpattern replace substitutionzReplace Text (Regex)rz+Find and replace text using regex patterns.rTr:rrxr)roptionalrrFrzWhen enabled, the dot (.) character will match any character including newline characters. When disabled, dots won't match newlines.)rrrtooltipcountrrzMaximum number of replacements to make. Set to 0 to replace all occurrences (default). Set to 1 to replace only the first match, 2 for the first two matches, etc.)rr rrrr)rrrrrrr)rr#r$r rrMr%r@s r(r)zRegexReplace.define_schemavsy"XXX/E D99 4@@  T::   !3TD[_ ``   edUY ZZ   54RVag hh WaQC$Y]hL MM   ""    r*TFrcd}|r|tjz}|r|tjz}|r|tjz}tj|||||} t j| S)Nr)rr)rrrrsubrr-) r&rrrxrrrrrris r(r.zRegexReplace.executesm  # R] "E  " R\ !E   RY E we5QQQ}V$$$r*N)TFFrrDr,r*r(rrusM  [ * % % %[ % % %r*rc:eZdZedZedZdS)JsonExtractStringc tjdddgdtjddtjdd gtjg S) NrzExtract Text from JSONr)jsonz extract jsonz parse jsonz json valuez read json json_stringTr:keyF)rrrrrrr?r@s r(r)zJsonExtractString.define_schemasty'1\\\  >> 77    ""    r*cd tj|}t|trC||vr?||}|t jdSt jt |St jdS#tjtf$rt jdcYSwxYw)Nr=) rloadsrdictrr-r4JSONDecodeError TypeError)r&rrdatar s r(r.zJsonExtractString.executes %:k**D$%% 1#++S ==,,,}SZZ000=$$ $$i0 % % %=$$ $ $ $ %sA B B.B*B/.B/NrDr,r*r(rrsH   [   % %[ % % %r*rcLeZdZedeeejfdZdS)StringExtensionrc Ktttttt t ttttttg SrB) rr6rFrPrWrkrvr}rrrrr)selfs r( get_node_listzStringExtension.get_node_lists9               r*N) r/r0r1rr!typer ComfyNoderr,r*r(rrsE  T$r|*<%=   X   r*rrc"KtSrB)rr,r*r(comfy_entrypointrs   r*)rrrtyping_extensionsrcomfy_api.latestrrrrr6rFrPrWrkrvr}rrrrrrrr,r*r(rs &&&&&&////////888882<888<CCCCC CCC.00000bl000.*****2<****%%%%%BL%%%B%%%%%%%%><<<<