^i0dZddlmZddlZddlmZddlmZmZm Z erddlm Z GddeZ ejd Z ejd Zd ZGd d e ZeZddZddZddZ ddZddZddZy)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)SequenceceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZdZ dZ!y )!_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_WHITEW/home/ubuntu/crypto_trading_bot/.venv/lib/python3.12/site-packages/wcwidth/sgr_state.pyr r s E D C FI EKG FMLJMIKJHHKJHHKJOOOOrKr z\x1b\[([\d;:]*)mz\x1b\[[\d;:]*mzceZdZUdZdZded<dZded<dZded<dZded<dZ ded<dZ ded <dZ ded <dZ ded <dZ ded <dZded <dZded<dZded<y) _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+rP__annotations__rQrRrSrTrUrVrWrXrYr[r\rJrKrLrNrNJs"D$CFDItE4KGTFDM4"d")-J&-)-J&-rKrNcV|jxs|jxs|jxs|jxsr|jxsd|j xsV|j xsH|jxs:|jxs,|jxs|jduxs|jduS)z Return True if any attributes are set. :param state: The SGR state to check. :returns: True if any attribute differs from default. N) rPrQrRrSrTrUrVrWrXrYr[r\)states rL_sgr_state_is_activer`ns JJ L%)) Lu|| Lu L{{ L#// L38== LDILL L"" L&+&<&< Lt+ L05/?/?t/KMrKc zt|sy|jdf|jdf|jdf|jdf|j df|j df|jdf|jd f|jd f|jd fg }|Dcgc] \}}|s | }}}|j1|jd jd |jD|j1|jd jd|jDdd j|dScc}}w)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. 12345678921;c32K|]}t|ywNstr.0ps rL z)_sgr_state_to_sequence..@!s1v@c32K|]}t|ywrorprrs rLruz)_sgr_state_to_sequence..rvrwz[m)r`rPrQrRrSrTrUrVrWrXrYr[appendjoinr\)r_ bool_attrsactivecodeparamss rL_sgr_state_to_sequencer{s%  & SEIIs+ellC-@ #c 2U5F5F4L u||S1E4G4G3M   & J (2 <|vtVd [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:) _SGR_PATTERNmatchgroupsplitintrztuple)sequencer params_strresultparamrtpartss rL_parse_sgr_paramsrs   x (E  QJ s *,F!!#&6 %<27S1ABAqSVa'BEB MM%, ' MM#e*1 56 M CsB2c t|}t|try|dk(r*t|}t|tryt|d|fS|dk(rbt|t|t|}}}t|ts t|tst|tryt|d|||fS y#t$rYywxYw)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. Nrr )next isinstancerr StopIteration)rbasemodenrgbs rL_parse_extended_colorrs F| dE " 19V A!U#Iq!$ $ 196lDL$v,!qA!U#z!U';z!U?SIq!Q* *      s(B6 B6 B6 AB6$B66 CCc t|}t|}|D]}t|trnt |dk\r)|dt j k(r|j|}n6t |dk\r(|dt jk(r|j|}|t jk(rt}|t jk(r|jd}|t jk(r|jd}|t jk(r|jd}|t jk(r|jd }6|t jk(r|jd }]|t j k(r|jd }|t j"k(r|jd }|t j$k(r|jd }|t j&k(r|jd}|t j(k(r|jd} |t j*k(r|jdd}H|t j,k(r|jd}o|t j.k(r|jdd}|t j0k(r|jdd}|t j2k(r|jd }|t j4k(r|jd } |t j6k(r|jd}4t j8|cxkrt j:ks,nt j<|cxkrt j>krnn|j|f}t j@|cxkrt jBks,nt jD|cxkrt jFkrnn|j|f}|t jHk(r|jd}+|t jJk(r|jd}R|t j k(r2tM|t j x}s|j|}|t jk(stM|t jx}s|j|}|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)r[)r\T)rP)rQ)rR)rS)rT)rU)rV)rW)rX)rYF)rPrQ)rSrY)rTrUN)'riterrrlenr r@_replacerDr,_SGR_STATE_DEFAULTr-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?rFrGrBrCrHrIrArEr)r_r params_listrrtcolors rL_sgr_state_updaters$H-K + F A9 a 1v{qtt'7'77!4Q11)9)9!9!4   ?&E $))^NNN-E $((]NNtN,E $++ NN$N/E $.. NNTN2E $**_NNN.E $"" "NNtN4E $,, NN4N0E $++ NN$N/E $$$ $NNN6E $'' 'NNDN9E $## #NN5N9E $// !NN%N0E $$$ $NNUUNKE $.. NNENBE $"" "NN5N1E $// !NN%N0E $(( (NNN7Emmq1DMM1%%Bd.B.BBNNqdN3Emmq1DMM1%%Bd.B.BBNNqdN3E $// !NNdN3E $// !NNdN3E $"" "-fd6F6FGGuG%8 $"" "-fd6F6FGGuG%8CA9D LrKc<td|Dr|s t|Sg}t}|D]o}t|}tj |D]}t ||j}|r||zn|}t|r |tz}|j|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``). c3FK|]}tj|ywro)_SGR_QUICK_CHECKsearch)rslines rLruz propagate_sgr..=s?&&t,?s!) anylistrrrfinditerrrr` _SGR_RESETrz)linesrr_rprefixr output_lines rL propagate_sgrr#s4 ?? ?uE{F E #'."**40 rs# 66&7&Zrzz-. 2::/0 . .B[ M'8< +36:Nb/rK