o aĎi0@sdZddlmZddlZddlmZddlmZmZm Z er$ddlm Z GdddeZ e d Z e d Zd ZGd d d e ZeZd(ddZd)ddZd*ddZd+dd Zd,d!d"Zd-d&d'ZdS).as 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_WHITErHrHM/home/jeff/fluffinator/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__rMrNrOrPrQrRrSrTrUrWrXrHrHrHrIrJJs            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_activens"r\strc Cst|sdS|jdf|jdf|jdf|jdf|jdf|jdf|jdf|jd f|j d f|j d fg }d d |D}|j durJ| d dd|j D|jdur]| d dd|jDdd |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).0activecoderHrHrI sz*_sgr_state_to_sequence..N;cs|]}t|VqdSNr]riprHrHrI z)_sgr_state_to_sequence..csrnrorprqrHrHrIrsrtz[m)r\rLrMrNrOrPrQrRrSrTrUrWappendjoinrX)rZ bool_attrsparamsrHrHrI_sgr_state_to_sequence{s  rzsequencelist[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 rrm:cSsg|] }|r t|ndqS)r)intrqrHrHrIrlsz%_parse_sgr_params..) _SGR_PATTERNmatchgroupsplitrvtupler~)r{r params_strresultparampartsrHrHrI_parse_sgr_paramss  rryIterator[int | tuple[int, ...]]baser~rVcCszUt|}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)ryrmodenrgbrHrHrI_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)'riterrrlenrr>_replacerBr*_SGR_STATE_DEFAULTr+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=rDrEr@rArFrGr?rCr)rZr{ params_listryrrcolorrHrHrI_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.width` 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|VqdSro)_SGR_QUICK_CHECKsearch)rilinerHrHrIrs=sz propagate_sgr..) anylistrrzrfinditerrrr\ _SGR_RESETrv)rrrZrprefixr output_linerHrHrI propagate_sgr#s r)rZrJr[rK)rZrJr[r])r{r]r[r|)ryrrr~r[rV)rZrJr{r]r[rJ)rrr[r)r) __future__rreenumrtypingrrrrrcompilerrrrJrr\rzrrrrrHrHrHrIs&    - !    Q