o NEgM1@s.dZddlZddlZddlmZddlmZmZmZm Z ddl Z ddl m Z m Z m Z ddlmZe jddd Z d Z d Z eZ ed ZGd ddeZGdddeZGdddeZejjZ ejjZ e jZ e jZ e j!Z" e j#Z$ e j%&dZ' e j%&dZ( de)de)fddZ*de)de+fddZ,dede)ddfddZ-de e)efde)fd d!Z.d"e e)efde)fd#d$Z/d%d%dddd&dede e)efde e)efd'e0d(e0d)e0de)fd*d+Z1ej2e1ej3d,Z4 ej2e1ej5d,Z6 ej2e1ej7d,Z8 d-e+d.e)d/e)d0e+d1e)de)f d2d3Z9d4e)de)fd5d6Z:dS)7z Support for ANSI escape sequences which are used for things like applying style to text, setting the window title, and asynchronous alerts. N)Enum)IOAnyListUnion)BackForeStyle)wcswidthF)stripNeverTerminalAlwaysz \x1b\[[^m]*mc@sXeZdZdZdefddZdedefddZdedefdd Ze de efd d Z d S) ColorBasea  Base class used for defining color enums. See fg and bg classes for examples. Child classes should define enums in the follow structure: key: color name (e.g. black) value: anything that when cast to a string returns an ANSI sequence returncCs t|jS)z Return ANSI color sequence instead of enum name This is helpful when using a ColorBase in an f-string or format() call e.g. my_str = "{}hello{}".format(fg.blue, fg.reset) )strvalue)selfrc/var/www/eduai.edurigo.com/doc_train/edurigo_ai/Puru/venv/lib/python3.10/site-packages/cmd2/ansi.py__str__Ks zColorBase.__str__othercCs t||S)zn Support building a color string when self is the left operand e.g. fg.blue + "hello" rrrrrr__add__S zColorBase.__add__cCs |t|S)zp Support building a color string when self is the right operand e.g. "hello" + fg.reset rrrrr__radd__ZrzColorBase.__radd__cCsdd|jDS)zReturn a list of color names.cSsg|]}|qSrr).0colorrrr esz$ColorBase.colors..) __members__)clsrrrcolorsaszColorBase.colorsN) __name__ __module__ __qualname____doc__rrrrr classmethodrr"rrrrrAs rc@veZdZdZejZejZej Z ej Z ej ZejZejZejZejZejZejZejZejZejZ ej!Z"ej#Z$ej%Z&dS)fgz Enum class for foreground colorsN)'r#r$r%r&rBLACKblackREDredGREENgreenYELLOWyellowBLUEblueMAGENTAmagentaCYANcyanWHITEwhite LIGHTBLACK_EX bright_black LIGHTRED_EX bright_red LIGHTGREEN_EX bright_greenLIGHTYELLOW_EX bright_yellow LIGHTBLUE_EX bright_blueLIGHTMAGENTA_EXbright_magenta LIGHTCYAN_EX bright_cyan LIGHTWHITE_EX bright_whiteRESETresetrrrrr)j& r)c@r()bgz Enum class for background colorsN)'r#r$r%r&rr*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArBrCrDrErFrGrHrIrJrKrrrrrMrLrMtextrcCs td|S)z Strip ANSI style sequences from a string. :param text: string which may contain ANSI style sequences :return: the same string with any ANSI style sequences removed ) ANSI_STYLE_REsubrPrrr strip_styles rUcCs tt|S)z Wrap wcswidth to make it compatible with strings that contains ANSI style sequences :param text: the string being measured :return: the width of the string when printed to the terminal )r rUrTrrrstyle_aware_wcswidths rVfileobjmsgcCs>ttksttkr|st|}||dS)z Write a string to a fileobject and strip its ANSI style sequences if required by allow_style setting :param fileobj: the file object being written to :param msg: the string being written N) allow_stylelower STYLE_NEVERSTYLE_TERMINALisattyrUwrite)rWrXrrrstyle_aware_writesr_fg_namec CHt|tr|jSz t|j}W|Sty#td|tw)a  Look up ANSI escape codes based on foreground color name. :param fg_name: foreground color name or enum to look up ANSI escape code(s) for :return: ANSI escape code(s) associated with this color :raises: ValueError: if the color cannot be found z8Foreground color {!r} does not exist; must be one of: {}) isinstancer)rrZKeyError ValueErrorformatr")r` ansi_escaperrr fg_lookup  rgbg_namec Cra)a  Look up ANSI escape codes based on background color name. :param bg_name: background color name or enum to look up ANSI escape code(s) for :return: ANSI escape code(s) associated with this color :raises: ValueError: if the color cannot be found z8Background color {!r} does not exist; must be one of: {})rbrMrrZrcrdrer")rirfrrr bg_lookuprhrjrQ)r)rMbolddim underlinerkrlrmcCsg}g}d|}|r|t||t|r%|t||t|r1|t|t|r=|t|t|rI|t |t d ||d |S)aY Apply ANSI colors and/or styles to a string and return it. The styling is self contained which means that at the end of the string reset code(s) are issued to undo whatever styling was done at the beginning. :param text: Any object compatible with str.format() :param fg: foreground color. Relies on `fg_lookup()` to retrieve ANSI escape based on name or enum. Defaults to no color. :param bg: background color. Relies on `bg_lookup()` to retrieve ANSI escape based on name or enum. Defaults to no color. :param bold: apply the bold style if True. Can be combined with dim. Defaults to False. :param dim: apply the dim style if True. Can be combined with bold. Defaults to False. :param underline: apply the underline style if True. Defaults to False. :return: the stylized string z{}rQ) reappendrgFG_RESETrjBG_RESETINTENSITY_BRIGHTINTENSITY_NORMAL INTENSITY_DIMUNDERLINE_ENABLEUNDERLINE_DISABLEjoin)rPr)rMrkrlrm additionsremovalsrrrstyles&         ry)r)terminal_columnspromptline cursor_offset alert_msgcCsddlm}|}d}|ddD]}t|}|t||d7}q|d} t| } | t|} t| |d} | |} t| |d}d}|| krT||| |7}|| }|tj|d|d7}|tj7}|d|7}|S)a)Calculate the desired string, including ANSI escape codes, for displaying an asynchronous alert message. :param terminal_columns: terminal width (number of columns) :param prompt: prompt that is displayed on the current line :param line: current contents of the Readline line buffer :param cursor_offset: the offset of the current cursor position within line :param alert_msg: the message to display to the user :return: the correct string so that the alert message appears to the user to be printed above the current line. r)CursorNrQ ) coloramar splitlinesrVintDOWNansi clear_lineUP)rzr{r|r}r~r prompt_linesnum_prompt_terminal_lines line_widthlast_prompt_linelast_prompt_line_width input_widthnum_input_terminal_linescursor_input_offsetcursor_input_line terminal_str total_linesrrrasync_alert_str3s(    rtitlecCs tj|S)zGet the required string, including ANSI escape codes, for setting window title for the terminal. :param title: new title for the window :return: string to write to sys.stderr in order to set the window title to the desired test )rr set_title)rrrr set_title_stris r);r& functoolsreenumrtypingrrrrrrrr wcwidthr initr[r\ STYLE_ALWAYSrYcompilerRrr)rMrKrrorp RESET_ALLBRIGHTrqDIMrsNORMALrrr code_to_charsrtrurrUrrVr_rgrjboolrypartialr/ style_successrA style_warningr= style_errorrrrrrrsv    )     & 6"6