o i5@s ddlZddlmZddlmZmZGdddejZGdddejZGdd d ejZ Gd d d ejZ Gd d d ejZ GdddejZ GdddejZ GdddejZGdddejZGdddejZGdddejZGdddeZdefddZdS)N)override)ComfyExtensionioc@$eZdZeddZeddZdS)StringConcatenatec CsNtjdddgdtjjdddtjjdddtjjd d d d gtjgd S)NrzText Concatenate utils/string) Concatenatez text concatz join textz merge textzcombine stringsconcat concatenatez append textz combine textstringstring_aT multilinestring_b delimiterF)rdefault)node_id display_namecategorysearch_aliasesinputsoutputsrSchemaStringInputOutputclsr 7/mnt/c/Users/fbmor/ComfyUI/comfy_extras/nodes_string.py define_schemaszStringConcatenate.define_schemacCst|||fSN)r NodeOutputjoin)rr rrr r r!executeszStringConcatenate.executeN__name__ __module__ __qualname__ classmethodr"r&r r r r!r  rc@r)StringSubstringc CsDtjdgdddtjjdddtjdtjd gtjgd S) Nr-) Substringz extract textz text portionzText Substringrr Tr startendrrrrrrrrrrIntrrr r r!r"s  zStringSubstring.define_schemacCst|||Sr#)rr$)rr r/r0r r r!r&0zStringSubstring.executeNr'r r r r!r-r,r-c@r) StringLengthc Cs4tjdgdddtjjdddgtjjdd gd S) Nr5)zcharacter countz text sizez string lengthz Text Lengthrr Tr lengthrr1r2rr r r!r"6s zStringLength.define_schemacCstt|Sr#)rr$len)rr r r r!r&EszStringLength.executeNr'r r r r!r55s  r5c@r) CaseConverterc CBtjdgdddtjjdddtjjdgd d gtjgd S) Nr9)zCase Converterz text case uppercase lowercase capitalizezText Case Converterrr Tr mode) UPPERCASEr< Capitalize Title Caseoptionsr1rrrrComborrr r r!r"KzCaseConverter.define_schemacCsV|dkr |}n|dkr|}n|dkr|}n |dkr$|}n|}t|S)Nr?r<r@rA)upperlowerr=titlerr$rr r>resultr r r!r&[s     zCaseConverter.executeNr'r r r r!r9J  r9c@r) StringTrimc Cr:) NrM)Trimzclean whitespacezremove whitespacestripz Text Trimrr Tr r>)BothLeftRightrBr1rDrr r r!r"lrFzStringTrim.define_schemacCsD|dkr |}n|dkr|}n |dkr|}n|}t|S)NrPrQrR)rOlstriprstriprr$rJr r r!r&|s    zStringTrim.executeNr'r r r r!rMkrLrMc@r) StringReplacec CsLtjdgdddtjjdddtjjdddtjjd ddgtjgd S) NrU)Replacezfind and replace substitutez swap textz Text Replacerr Tr findreplacer1rrr r r!r"szStringReplace.define_schemacCst|||Sr#)rr$rY)rr rXrYr r r!r&r4zStringReplace.executeNr'r r r r!rUr,rUc@r)StringContainsc CsRtjdgdddtjjdddtjjdddtjjd ddd gtjjd d gd S)NrZ)Containsz text includeszstring includesz Text Containsrr Tr substringcase_sensitiveradvancedcontainsr7r1rrrrBooleanrrr r r!r"s zStringContains.define_schemacCs(|r||v}n||v}t|Sr#)rHrr$)rr r\r]r`r r r!r&s  zStringContains.executeNr'r r r r!rZr,rZc@r) StringComparec Cs`tjdgdddtjjdddtjjdddtjjd gd d tjjd ddd gtjgdS)Nrc)Comparez text matchz string equalsz starts withz ends withz Text Comparerr Tr rr>) Starts With Ends WithEqualrBr]r^r1)rrrrrErbrrr r r!r"szStringCompare.define_schemacCsh|r|}|}n|}|}|dkrt||kS|dkr&t||S|dkr2t||SdS)Nrgrerf)rHrr$ startswithendswith)rr rr>r]abr r r!r&szStringCompare.executeNr'r r r r!rcs  rcc@r) RegexMatchcCsrtjdgdddtjjdddtjjdddtjjd ddd tjjd d dd tjjd d dd gtjjddgdS)Nrl)z Regex Matchregexz pattern matchz text containsz string matchz Text Matchrr Tr regex_patterncase_insensitiver^rFdotallmatchesr7r1rarr r r!r"s zRegexMatch.define_schemac Csld}|r |tjO}|r|tjO}|r|tjO}z t|||}|du}Wn tjy0d}Ynwt|S)NrF)re IGNORECASE MULTILINEDOTALLsearcherrorrr$) rr rnrorrpflagsmatchrKr r r!r&s     zRegexMatch.executeNr'r r r r!rls  rlc@r) RegexExtractcCstjdgdddtjjdddtjjdddtjjd gd d tjjd ddd tjjdddd tjjdddd tjjddddddgtjgdS)Nrz)z Regex Extractrmzpattern extractz text parserz parse textzText Extract Substringrr Tr rnr>) First Match All Matches First Group All GroupsrBror^rFrp group_indexrd)rminmaxr_r1)rrrrrErbr3rrr r r!r" s  zRegexExtract.define_schemacCsnd}d} |r | tjO} |r| tjO} |r| tjO} z|dkr0t||| } | r-| d} nwd} nt|dkrXt||| } | rUt| dtrO| dd| D} nU| | } nOd} nL|dkrvt||| } | rst | |krs| |} n1d} n.|d krt ||| } g} | D]} | rt | |kr| | |q| | } nd} Wn tjyd} Ynwt| S) N rr{rr|cSsg|]}|dqS)rr ).0mr r r! 8sz(RegexExtract.execute..r}r~)rrrsrtrurvgroupfindall isinstancetupler%r8groupsfinditerappendrwrr$)rr rnr>rorrprjoin_delimiterrxryrKrqresultsr r r!r& sP        zRegexExtract.executeNr'r r r r!rz s  rzc@s&eZdZeddZed ddZdS) RegexReplacecCstjdgddddtjjdddtjjd ddtjjd ddtjjd dddd tjjd dddd tjjddddddtjjddddddddgtjgdS)Nr)z Regex Replacermzpattern replacez regex replace substitutionzText Replace (Regex)rz+Find and replace text using regex patterns.r Tr rnrYro)roptionalr_rFrpzWhen enabled, the dot (.) character will match any character including newline characters. When disabled, dots won't match newlines.)rrr_tooltipcountrrzMaximum 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.)rrrrr_r)rrrr descriptionrr)rrrrrbr3rrr r r!r"Vs" zRegexReplace.define_schemaTFrc CsLd}|r |tjO}|r|tjO}|r|tjO}tj|||||d} t| S)Nr)rrx)rrrsrtrusubrr$) rr rnrYrorrprrxrKr r r!r&ls    zRegexReplace.executeN)TFFrr'r r r r!rUs  rc@s(eZdZedeeejfddZdS)StringExtensionreturnc stttttttttt t g Sr#) rr-r5r9rMrUrZrcrlrzr)selfr r r! get_node_list{szStringExtension.get_node_listN) r(r)r*rlisttyper ComfyNoderr r r r!rzsrrcstSr#)rr r r r!comfy_entrypointsr)rrtyping_extensionsrcomfy_api.latestrrrrr-r5r9rMrUrZrcrlrzrrrr r r r!s  !$)K%