a œòJbi7ã@sâdZddlmZddlZddlmZddlmZedgd¢ƒZedgd ¢ƒZ ed gd ¢ƒZ ed d gƒZ ed d gƒZ edd gƒZ edd gƒZdZdeZdZdeefZe de¡Ze d¡ZGdd„deƒZGdd„deƒZdS)zJ Utilities for processing ANSI escape codes and special ASCII characters. é)Ú namedtupleN)ÚQtGui)Ú dark_styleÚ EraseAction)ÚactionÚareaÚerase_toÚ MoveAction)rÚdirZunitÚcountÚ ScrollActionÚCarriageReturnActionrÚ NewLineActionÚ BeepActionÚBackSpaceActionZABCDEFGHJKSTfmnsuz \[(.*?)([%s])z \](.*?)[]z ?(%s|%s)?z(|| (?! )| ? )|(?:%s)z([ ])c@s\eZdZdZdZiZdd„Zdd„Zdd„Zgfd d „Z d d „Z d d„Z dd„Z dd„Z dS)ÚAnsiCodeProcessorzœ Translates special ASCII characters and ANSI escape codes into readable attributes. It also supports a few non-standard, xterm-specific codes. FcCsg|_|j ¡|_| ¡dS©N)ÚactionsÚdefault_color_mapÚcopyÚ color_mapÚ reset_sgr©Úself©rú|dd…n|}t |¡D]r}||| ¡…}t |j|¡}|sz|jr†|Vg|_| ¡}dd„|  ¡Dƒ}|d}|dkrÎ|j  t dƒ¡dVg|_qL|d krô|j  t d ƒ¡dVg|_qL|d kr|j  t d ƒ¡dVg|_qL|dks0|d krN|j  tdƒ¡|Vg|_qLdd„|d d¡Dƒ} | d¡r¬zttt| ƒƒ} Wnty˜Yn0| |d| ¡qL| d¡rL| | ¡qL||d…}t |j|¡}|sê|jrð|V|dur|j  tdƒ¡|VdS)zC Yields substrings for which the same escape code applies. réÿÿÿÿÚ NcSsg|]}|dur|‘qSrr)Ú.0ÚgrrrÚ móz2AnsiCodeProcessor.split_string..úZbeepú zcarriage-returnúZ backspacez ÚnewlinecSsg|] }|r|‘qSrr)r%Zparamrrrr'€r(éú;ú[éú])rÚlenÚANSI_OR_SPECIAL_PATTERNÚfinditerÚstartÚSPECIAL_PATTERNÚsubÚ_replace_specialÚendÚgroupsÚappendrr rrÚsplitÚ startswithÚlistÚmapÚintÚ ValueErrorÚ set_csi_codeÚ set_osc_code) rÚstringr5Z last_charÚmatchÚrawZ substringr:Zg0ÚparamsrrrÚ split_stringYsZ        zAnsiCodeProcessor.split_stringcCsö|dkr&|r| |¡qò| dg¡nÌ|dks6|dkr¬|rB|dnd}d|krZdkrònqò|dkrjdnd}|dkr|d}n|d krŠd }n |dkr–d }|j td ||ƒ¡nF|d ks¼|dkrò|d krÈdnd}|rØ|dnd }|j td|d|ƒ¡dS)a3 Set attributes based on CSI (Control Sequence Introducer) code. Parameters ---------- command : str The code identifier, i.e. the final character in the sequence. params : sequence of integers, optional The parameter codes for the command. ÚmrÚJÚKr0ZscreenÚliner9r-r5ÚallZeraseÚSÚTZupÚdownÚscrollN)Ú set_sgr_coderr;rr )rÚcommandrGÚcoderrr r rrrrB˜s.  ÿÿzAnsiCodeProcessor.set_csi_codec Cs|zt| d¡ƒ}Wnttfy*YdS0|dkrxz,t| d¡ƒ}| d¡}| |¡|j|<WnttfyvYn0dS)zÀ Set attributes based on OSC (Operating System Command) parameters. Parameters ---------- params : sequence of str The parameters for the command. rNé)r@ÚpopÚ IndexErrorrAÚ_parse_xterm_color_specr)rrGrSÚcolorÚspecrrrrC¼s zAnsiCodeProcessor.set_osc_codecCsà|sdS| d¡}|dkr&| ¡n¬|dkrF|jr|d kr¦d |_n,|d kr¸d |_n|d krÔ|d krÔ|d |_nþ|dkr2|r2| d¡}|dkr |r | d¡|_n$|dkrÒ|dd…|_g|dd…<n |dkrDd|_nŽ|dkrd|dkrd|d|_nn|dkrÂ|rÂ| d¡}|dkrœ|rœ| d¡|_n$|dkrÒ|dd…|_g|dd…<n|dkrÒd|_|  |¡dS)ad Set attributes based on SGR (Select Graphic Rendition) codes. Parameters ---------- params : sequence of ints A list of SGR codes for one or more SGR commands. Usually this sequence will have one element per command, although certain xterm-specific commands requires multiple elements. Nrr-Tr0érUéFéééé%é&éé'é(é/é0é1) rVrÚbold_text_enabledrrrr r!r"rR)rrGrTZ _color_typerrrrRÒsZ                  zAnsiCodeProcessor.set_sgr_codecCsh| d¡r*ttdd„|dd… d¡ƒƒS| d¡rTttdd„|dd… d¡ƒƒS|d krdtd ƒ‚|S) Nzrgb:cSs t|dƒS)Né)r@©ÚxrrrÚr(z;AnsiCodeProcessor._parse_xterm_color_spec..rUú/zrgbi:cSstt|ƒdƒS)Néÿ)r@Úfloatrjrrrrlr(rbú?zUnsupported xterm color spec)r=Útupler?r<rA)rrZrrrrXs    ÿz)AnsiCodeProcessor._parse_xterm_color_speccCs,| d¡}|dkr(|j tddddƒ¡dS)Nr-ú rQrPZpageÚ)Úgrouprr;r )rrEZspecialrrrr8!s z"AnsiCodeProcessor._replace_specialN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rhrrrrHrBrCrRrXr8rrrrr9s ? $E rc@sXeZdZdZdddddddd d d d d d ddddœZe ¡Zddd„Zdd„Zdd„Z dS)ÚQtAnsiCodeProcessorz9 Translates ANSI escape codes into QTextCharFormats. ÚblackZdarkredZ darkgreenZbrownZdarkblueZ darkvioletZ steelblueZgreyZredZlimeZyellowZ deepskyblueZmagentaZcyanZwhite)rr-r0r[rUrbéééé é é é é éércCszt|tƒr2|dkr"|dkr"|d7}|j |d¡}nt|ttfƒrF|}ndSt|tƒr^t |¡St|ttfƒrvtj|ŽSdS)zp Returns a QColor for a given color code or rgb list, or None if one cannot be constructed. r}rN) Ú isinstancer@rÚgetrqr>ÚstrrZQColor)rrYrÚ constructorrrrÚ get_colorBs    zQtAnsiCodeProcessor.get_colorcCsŒt ¡}| |j|j¡}|dur*| |¡| |j|j¡}|durL| |¡|jrb|  tj j ¡n|  tj j ¡|  |j¡| |j¡|S)zN Returns a QTextCharFormat that encodes the current style attributes. N)rZQTextCharFormatr‰r!rZ setForegroundr"Z setBackgroundrZ setFontWeightZQFontZBoldZNormalZ setFontItalicrZsetFontUnderliner )rÚformatZqcolorrrrÚ get_format]s    zQtAnsiCodeProcessor.get_formatcCs\|j ¡|_t|ƒsJtdƒD]}|j||j|d<qd|jd<|jd<|j |j¡dS)zo Given a syntax style, attempt to set a color map that will be aesthetically pleasing. r}rzr|r„N)Údarkbg_color_maprrrÚrangerÚupdate)rZstyleÚirrrÚset_background_colorvs   z(QtAnsiCodeProcessor.set_background_colorN)r) rurvrwrxrŒrrr‰r‹rrrrrry(s,ð ry)rxÚ collectionsrÚreZqtpyrZqtconsole.stylesrrr r r rrrZ CSI_COMMANDSZCSI_SUBPATTERNZOSC_SUBPATTERNZ ANSI_PATTERNÚcompiler3r6ÚobjectrryrrrrÚs,       ÿ p