o j3@sdZddlmZddlZddlmZddlmZmZm Z er$ddlm Z GdddeZ e d Z e d Zd ZGd d d e ZeZd,ddZd-ddZd.ddZd/ddZd0d#d$Zd1d%d&Zd2d*d+ZdS)3as SGR (Select Graphic Rendition) state tracking for terminal escape sequences. This module provides functions for tracking and propagating terminal styling (bold, italic, colors, etc.) via public API propagate_sgr(), and its dependent functions, cut() and wrap(). It only has attributes necessary to perform its functions, eg 'RED' and 'BLUE' attributes are not defined. ) annotationsN)IntEnum) TYPE_CHECKINGIterator NamedTuple)Sequencec@seZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!d S)!_SGRz SGR (Select Graphic Rendition) parameter codes. References: - https://invisible-island.net/xterm/ctlseqs/ctlseqs.html - https://github.com/tehmaze/ansi/tree/master/ansi/colour r %&'(/01ZadkN)"__name__ __module__ __qualname____doc__RESETBOLDDIMITALIC UNDERLINEBLINK RAPID_BLINKINVERSEHIDDEN STRIKETHROUGHDOUBLE_UNDERLINE BOLD_DIM_OFF ITALIC_OFF UNDERLINE_OFF BLINK_OFF INVERSE_OFF HIDDEN_OFFSTRIKETHROUGH_OFFFG_BLACKFG_WHITE FG_EXTENDED FG_DEFAULTBG_BLACKBG_WHITE BG_EXTENDED BG_DEFAULTFG_BRIGHT_BLACKFG_BRIGHT_WHITEBG_BRIGHT_BLACKBG_BRIGHT_WHITErHrH/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/wcwidth/sgr_state.pyrs@rz\x1b\[([\d;:]*)mz\x1b\[[\d;:]*mzc@seZdZUdZdZded<dZded<dZded<dZded<dZ ded<dZ ded <dZ ded <dZ ded <dZ ded <dZded <dZded<dZded<dS) _SGRStatea Track active SGR terminal attributes by category (immutable). :param bold: Bold attribute (SGR 1). :param dim: Dim/faint attribute (SGR 2). :param italic: Italic attribute (SGR 3). :param underline: Underline attribute (SGR 4). :param blink: Slow blink attribute (SGR 5). :param rapid_blink: Rapid blink attribute (SGR 6). :param inverse: Inverse/reverse attribute (SGR 7). :param hidden: Hidden/invisible attribute (SGR 8). :param strikethrough: Strikethrough attribute (SGR 9). :param double_underline: Double underline attribute (SGR 21). :param foreground: Foreground color as tuple of SGR params, or None for default. :param background: Background color as tuple of SGR params, or None for default. Fboolbolddimitalic underlineblink rapid_blinkinversehidden strikethroughdouble_underlineNtuple[int, ...] | None foreground background)r&r'r(r)rL__annotations__rMrNrOrPrQrRrSrTrUrWrXrHrHrHrIrJKs            rJstatereturnrKcCsP|jp'|jp'|jp'|jp'|jp'|jp'|jp'|jp'|jp'|j p'|j dup'|j duS)z Return True if any attributes are set. :param state: The SGR state to check. :returns: True if any attribute differs from default. N) rLrMrNrOrPrQrRrSrTrUrWrX)rZrHrHrI_sgr_state_is_activeos"r\colortuple[int, ...]strcCs@t|dkr|ddkrddd|DSddd|DS) a Join color parameters, preserving the ITU T.416 colon form where it was used. ``38:2::R:G:B`` carries a colour space element that the legacy ``38;2;R;G;B`` form has no slot for, so joining it with ``;`` would shift R, G and B one position and leave a trailing parameter. :param color: Color parameters as parsed by :func:`_parse_sgr_params`. :returns: Parameter string for embedding in an SGR sequence. r r r :cs|]}t|VqdSNr_.0prHrHrI z'_color_params_to_str..;csrarbrcrdrHrHrIrgrh)lenjoin)r]rHrHrI_color_params_to_str|s rlc Cst|sdS|jdf|jdf|jdf|jdf|jdf|jdf|jdf|jd f|j d f|j d fg }d d |D}|j durD| t |j |jdurQ| t |jdd|dS)z Generate minimal SGR sequence to restore this state from reset. :param state: The SGR state to convert. :returns: SGR escape sequence string, or empty string if no attributes set. 12345678921cSsg|]\}}|r|qSrHrH)reactivecoderHrHrI sz*_sgr_state_to_sequence..Nzim)r\rLrMrNrOrPrQrRrSrTrUrWappendrlrXrk)rZ bool_attrsparamsrHrHrI_sgr_state_to_sequences  rsequencelist[int | tuple[int, ...]]cCst|}|s gS|d}|sdgSg}|dD]#}d|vr2dd|dD}|t|q||r:t|ndq|S)a Parse SGR sequence and return list of parameter values. Handles compound sequences like ``\x1b[1;31;4m`` -> [1, 31, 4]. Empty params (e.g., ``\x1b[m``) are treated as [0] (reset). Colon-separated extended colors like ``\x1b[38:2::255:0:0m`` are returned as tuples: [(38, 2, 255, 0, 0)]. :param sequence: SGR escape sequence string. :returns: List of integer parameters or tuples for colon-separated colors. r rrir`cSsg|] }|r t|ndqS)r)intrdrHrHrIrzsz%_parse_sgr_params..) _SGR_PATTERNmatchgroupsplitr|tupler)rr params_strresultparampartsrHrHrI_parse_sgr_paramss  rr~Iterator[int | tuple[int, ...]]baserrVcCszUt|}t|tr WdS|dkr%t|}t|trWdSt|d|fWS|dkrSt|t|t|}}}t|tsFt|tsFt|trIWdSt|d|||fWSWdSty_YdSw)a9 Parse extended color (256-color or RGB) from parameter iterator. :param params: Iterator of remaining SGR parameters (semicolon-separated format). :param base: Base code (38 for foreground, 48 for background). :returns: Color tuple like (38, 5, N) or (38, 2, R, G, B), or None if malformed. Nr r )next isinstancerr StopIteration)r~rmodenrgbrHrHrI_parse_extended_colors(    rcCsRt|}t|}|D]}t|tr:t|dkr&|dtjkr&|j|d}nt|dkr9|dtjkr9|j|d}q |tj krBt }q |tj krN|jdd}q |tj krZ|jdd}q |tj krf|jdd}q |tjkrr|jdd }q |tjkr~|jdd }q |tjkr|jdd }q |tjkr|jdd }q |tjkr|jdd }q |tjkr|jdd}q |tjkr|jdd}q |tjkr|jddd}q |tjkr|jdd}q |tjkr|jddd}q |tjkr|jddd}q |tjkr|jdd }q |tjkr|jdd }q |tjkr|jdd}q tj|kr!tjks2ntj|kr0tjkr:nn|j|fd}q tj |krHtj!ksYntj"|krWtj#krann|j|fd}q |tj$krn|jdd}q |tj%kr{|jdd}q |tjkrt&|tj}r|j|d}q |tjkrt&|tj}r|j|d}q |S)z Parse SGR sequence and return new state with updates applied. :param state: Current SGR state. :param sequence: SGR escape sequence string. :returns: New SGRState with updates applied. r r)rW)rXT)rL)rM)rN)rO)rP)rQ)rR)rS)rT)rUF)rLrM)rOrU)rPrQN)'riterrrrjrr>_replacerBr*_SGR_STATE_DEFAULTr+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=rDrEr@rArFrGr?rCr)rZr params_listr~rfr]rHrHrI_sgr_state_updates                              rlines Sequence[str] list[str]cCstdd|Dr |st|Sg}t}|D]*}t|}t|D] }t||}q |r0||n|}t|r:|t }| |q|S)a' Propagate SGR codes across wrapped lines. When text with SGR styling is wrapped across multiple lines, each line needs to be self-contained for proper display. This function: - Ends each line with ``\x1b[0m`` if styles are active (prevents bleeding) - Starts each subsequent line with the active style restored :param lines: List of text lines, possibly containing SGR sequences. :returns: List of lines with SGR codes propagated. Example:: >>> propagate_sgr(['\x1b[31mhello', 'world\x1b[0m']) ['\x1b[31mhello\x1b[0m', '\x1b[31mworld\x1b[0m'] This is useful in cases of making special editors and viewers, and is used for the default modes (propagate_sgr=True) of :func:`wcwidth.wrap` and :func:`wcwidth.clip`. When wrapping and clipping text containing SGR sequences, maybe a previous line enabled the BLUE color--if we are viewing *only* the line following, we would want the carry over the BLUE color, and all lines with sequences should end with terminating reset (``\x1b[0m``). css|]}t|VqdSrb)_SGR_QUICK_CHECKsearch)relinerHrHrIrgNsz propagate_sgr..) anylistrrrfinditerrrr\ _SGR_RESETr|)rrrZrprefixr output_linerHrHrI propagate_sgr4s r)rZrJr[rK)r]r^r[r_)rZrJr[r_)rr_r[r)r~rrrr[rV)rZrJrr_r[rJ)rrr[r)r) __future__rreenumrtypingrrrrrcompilerrrrJrr\rlrrrrrrHrHrHrIs(    - !     Q