j\ddlZddlZddlZddlZddlmZmZmZmZe ejdrejjndZ e xsdZ e jdvZdZGddeZej&d ej(d ej*d ej,d d d dd dd dd ddddddddddi Zej&esdndej(esdndej*esdnd ej,esd!nd"iZd#Zd$Z d?d%ed&eeeefd'eeeefd(ed)ed*ef d+Zd@d%ed,ed-ed*efd.ZdAd/ed0ed1ed*efd2Z dBd3ed4ed&eeefd'eeeefeeeffd5ed*ef d6Z! dCd7ed8eeeefd-ed*efd9Z"dDd:ed;ed*efd<Z#d:ed*efd=Z$d*efd>Z%y)EN)AnyOptionalTupleUnionencodingascii)utf8zutf-8ANSI_COLORS_DISABLEDceZdZdZdZdZdZy)MESSAGESgoodfailwarninfoN)__name__ __module__ __qualname__GOODFAILWARNINFOU/home/spazio/kingdom_calls/work5/transenv/lib/python3.12/site-packages/wasabi/util.pyr r s D D D Drr redgreenyellowbluepinkcyanwhitegreyblacku✔z[+]u✘z[x]u⚠z[!]uℹz[i]+-textfgbgbold underlinereturnctj||}tj||}t|||gs|Sg}|r|jd|r|jd|r |jdj ||r |jdj |dj dj ||S)aColor text by applying ANSI escape sequence. text (str): The text to be formatted. fg (Optional[Union[str, int]]): Optional foreground color. String name or 0 - 256 (see COLORS). bg (Optional[Union[str, int]]): Optional background color. String name or 0 - 256 (see COLORS). bold (bool): Format text in bold. underline (bool): Underline text. RETURNS (str): The formatted text. 14z38;5;{}z48;5;{}z [{}m{};)COLORSgetanyappendformatjoin)r/r0r1r2r3styless rcolorr@3s B B B B B~  F  c c  i&&r*+  i&&r*+  % %chhv&6 ==rwrap_maxindentct|dz}|t|z }t|}tj||||ddS)a$Wrap text at given width using textwrap module. text (Any): The text to wrap. wrap_max (int): Maximum line width, including indentation. Defaults to 80. indent (int): Number of spaces used for indentation. Defaults to 4. RETURNS (str): The wrapped text with line breaks.  F)widthinitial_indentsubsequent_indentbreak_long_wordsbreak_on_hyphens)lenstrtextwrapfill)r/rArB indent_str wrap_widths rwraprPSsG#JC O+J t9D == !$  robjmax_lenellipsisct|}t||k\r(t|dz }dj|d|||| dS|S)avWrapper around `repr()` to print shortened and formatted string version. obj: The object to represent. max_len (int): Maximum string length. Longer strings will be cut in the middle so only the beginning and end is displayed, separated by ellipsis. ellipsis (str): Ellipsis character(s), e.g. "...". RETURNS (str): The formatted representation. rz{} {} {}N)reprrJintr=)rQrRrSstringhalfs r format_reprrYhsO#YF 6{g7Q;  &$.II rab add_symbolsc J|jd}|jd}g}tjd||}|jD]\} } } } } | dk(r|| | D]}|j || dk(s| dk(rD|| | D]<}|rdj t |n|}|j t|||d>| d k(s| dk(s|| | D]<}|rdj t|n|}|j t|||d >dj|S) aOCompare two strings and return a colored diff with red/green background for deletion and insertions. a (str): The first string to diff. b (str): The second string to diff. fg (Union[str, int]): Foreground color. String name or 0 - 256 (see COLORS). bg (Union[Tuple[str, str], Tuple[int, int]]): Background colors as (insert, delete) tuple of string name or 0 - 256 (see COLORS). add_symbols (bool): Whether to add symbols before the diff lines. Uses '+' for inserts and '-' for deletions. Default is False. RETURNS (str): The formatted diff.  Nequalinsertreplacez{} {}r)r0r1deleter) splitdifflibSequenceMatcher get_opcodesr<r= INSERT_SYMBOLr@ DELETE_SYMBOLr>)rZr[r0r1r\a_listb_listoutputmatcheropcodea0a1b0b1items r diff_stringsrsys8&WWT]F WWT]F F%%dFF;G")"5"5"7 <BB W r"  $ d# $ X 9!4r"  <>Iw~~mT:t eDRBqE:; < X 9!4r"  <>Iw~~mT:t eDRBqE:; < < 99V r descriptiondefaultc|rdj|nd}tdj|||}t|}|S)aGet user input from the command line via raw_input / input. description (str): Text to display before prompt. default (Optional[Union[str, bool]]): Optional default value to display with prompt. indent (int): Indentation in spaces. RETURNS (str): User input. z (default: {})z{}{}: )rB)r=rPinput)rtrurB additionalprompt user_inputs r get_raw_inputr|s?6=!((1"J (//+z:6 JFvJ rrWerrorscft|}|jt|jd}|S)zMangle non-supported characters, for savages with ASCII terminals. string (Any): The string to escape. errors (str): The str.encode errors setting. Defaults to `"replace"`. RETURNS (str): The escaped string. r )rKencodeENCODINGdecode)rWr}s r locale_escapers-[F ]]8V , 3 3F ;F MrcN |jty#t$rYywxYw)zCheck if terminal can render unicode characters, e.g. special loading icons. Can be used to display fallbacks for ASCII terminals. string (str): The string to render. RETURNS (bool): Whether the terminal can render the text. TF)rrUnicodeEncodeError)rWs r can_renderrs( h s  $$ctjtry ddlm}|y#t $r*t jdk(rdtjvrYyYywxYw)zReturns True if the running system's terminal supports ANSI escape sequences for color, formatting etc. and False otherwise. RETURNS (bool): Whether the terminal supports ANSI colors. Frjust_fix_windows_consolewin32ANSICONT) osgetenvENV_ANSI_DISABLEDcoloramar ImportErrorsysplatformenvironrs r supports_ansirsX  yy"# #5 !"   <<7 "y 'B  s*.AA)NNFF)Pr)2z...)r+)r rF)Fr)ra)&rdrrrLtypingrrrrhasattrstdoutrSTDOUT_ENCODINGrlowerNO_UTF8robjectr rrrrr9ICONSrgrhrKrVboolr@rPrYrsr|rrrrrrrs ..)0Z)H#**%%d  %g .. "3 3+v MM1 MM1 MM1 MM1 1 Q a A A A Q A R $ MM78 MM78 MM78 MM78    %)$( > >sCx!> sCx!>  >  >  >@scC*S3ss("2B # # # c3h# eCHouS#X./ #  #  #NRS  'c4i(89 KN   # s 3  s t tr