o i\@s(ddlZddlZddlZddlZddlmZmZmZmZe ej dr&ej j ndZ e p+dZ e dvZdZGdddeZejd ejd ejd ejd d d dd dd dd ddddddddddi Zejesbdndejesidndejespdnd ejeswd!nd"iZd#Zd$Z   % %dMd&ed'eeeefd(eeeefd)ed*ed+ef d,d-ZdNd&ed/ed0ed+efd1d2ZdOd5ed6ed7ed+efd8d9Z  : %dPd;edd?Z! dQd@edAeeeefd0ed+efdBdCZ"dRdEedFed+efdGdHZ#dEed+efdIdJZ$d+efdKdLZ%dS)SN)AnyOptionalTupleUnionencodingascii)utf8zutf-8ANSI_COLORS_DISABLEDc@seZdZdZdZdZdZdS)MESSAGESgoodfailwarninfoN)__name__ __module__ __qualname__GOODFAILWARNINFOrr?/home/ubuntu/.local/lib/python3.10/site-packages/wasabi/util.pyr s r redgreenyellowbluepinkcyanwhitegreyblacku✔z[+]u✘z[x]u⚠z[!]uℹz[i]+-Ftextfgbgbold underlinereturncCst||}t||}t|||gs|Sg}|r|d|r%|d|r/|d||r9|d|dd||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,r-r.r/r0stylesrrrcolor3s    r<Pwrap_maxindentcCs2|d}|t|}t|}tj||||dddS)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,r>r? indent_str wrap_widthrrrwrapSs rL2...objmax_lenellipsiscCsDt|}t||kr t|d}d|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)reprrFintr9)rOrPrQstringhalfrrr format_reprhs    rVrrab add_symbolsc Cs|d}|d}g}td||}|D]g\} } } } } | dkr0|| | D]}||q(| dks8| dkrW|| | D]}|rHdt|n|}|t|||ddq>| d ks_| dkr~|| | D]}|rodt|n|}|t|||d dqeqd |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)r-r.deleter) splitdifflibSequenceMatcher get_opcodesr8r9 INSERT_SYMBOLr< DELETE_SYMBOLr:)rXrYr-r.rZa_listb_listoutputmatcheropcodea0a1b0b1itemrrr diff_stringsys$    rp descriptiondefaultcCs2|rd|nd}td|||d}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{}{}: )r?)r9rLinput)rqrrr? additionalprompt user_inputrrr get_raw_inputs rxr^rTerrorscCst|}|t|d}|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)rGencodeENCODINGdecode)rTryrrr locale_escapesr}cCs&z|tWdStyYdSw)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)rzr{UnicodeEncodeError)rTrrr can_renders   rcCs^ttrdSzddlm}Wnty)tjdkr#dtjvr&YdSYdSYdSw|dS)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 ImportErrorsysplatformenvironrrrr supports_ansis  r)NNFF)r=r)rMrN)r(rWF)Fr)r^)&rarrrHtypingrrrrhasattrstdoutrSTDOUT_ENCODINGr{lowerNO_UTF8robjectr rrrrr5ICONSrdrerGrSboolr<rLrVrprxr}rrrrrrs     '