Yai7ddZddlmZddlZddlmZddlmZedgdZedgd Z ed gd Z ed d gZ ed d gZ edd gZ edd gZdZdezZdZdededZej&dezZej&dZGddeZGddeZy)zJ Utilities for processing ANSI escape codes and special ASCII characters. ) namedtupleN)QtGui) dark_style EraseAction)actionareaerase_to MoveAction)rdirunitcount ScrollActionCarriageReturnActionr NewLineAction BeepActionBackSpaceActionABCDEFGHJKSTfmnsuz \[(.*?)([%s])z \](.*?)[]z?(|z)?z(|| (?! )| ? )|(?:%s)z([ ])cLeZdZdZdZiZdZdZdZgfdZ dZ dZ d Z d Z y ) AnsiCodeProcessorz Translates special ASCII characters and ANSI escape codes into readable attributes. It also supports a few non-standard, xterm-specific codes. Fcpg|_|jj|_|j y)N)actionsdefault_color_mapcopy color_map reset_sgrselfs ?/usr/lib/python3/dist-packages/qtconsole/ansi_code_processor.py__init__zAnsiCodeProcessor.__init__Js) //446 cXd|_d|_d|_d|_d|_d|_y)z; Reset graphics attributs to their default values. rFN) intensityitalicbold underlineforeground_colorbackground_colorrs rrzAnsiCodeProcessor.reset_sgrOs0   $ $r!c#`Kg|_d}t|dkDr |ddk(rdnd}||ddn|}tj|D]}|||j }t j |j|}|s |jr |g|_|j}|jDcgc]}|| }}|d} | dk(r0|jjtddg|_| dk(r0|jjtddg|_| d k(r1|jjtd dg|_%| dk(s| d k(r1|jjtd | g|_`|d jdD cgc]} | s|  } } | j!dr1 t#t%t&| } |j)|d| | j!ds|j-| ||d}t j |j|}|s |jr||)|jjtd |yycc}wcc} w#t*$rYfwxYww)zC Yields substrings for which the same escape code applies. r Nbeep zcarriage-return backspacez newline;[])rlenANSI_OR_SPECIAL_PATTERNfinditerstartSPECIAL_PATTERNsub_replace_specialendgroupsappendrrrrsplit startswithlistmapint set_csi_code ValueError set_osc_code) rstringr: last_charmatchraw substringgr?g0paramparamss r split_stringzAnsiCodeProcessor.split_stringYsn   K!Or d0BD ) 56,55f=( .Eu{{}-C'++D,A,A3GIDLL! IIKE!&CAAMaCFCBTz ##Jv$67 ! t ##$89J$KL ! t ##OK$@A ! trV| ##M)$<=! .4Qiooc.BMUe5MM==%=!%c#v&6!7 ))&)V<]]3'%%f-Q( .TUVn#''(=(=sC  O  LL   i 8 9O !OD&N &sVB4J.6J>JC8J.:JJJ.J3(J.BJ. J+'J.*J++J.c|dk(r'|r|j|y|jdgy|dk(s|dk(r_|r|dnd}d|cxkrdkrJny|dk(rdnd}|dk(rd}n|d k(rd }n|dk(rd }|jjtd |yy|d k(s|dk(r:|d k(rdnd}|r|dnd }|jjt d|d|yy)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. mrJKr5screenliner>r2r:alleraseSTupdownscrollN) set_sgr_coderr@rr)rcommandrQcoderr r r s rrFzAnsiCodeProcessor.set_csi_codes c>!!&)!!1#&nn &6!9ADD~A~#*c>xv19$HQY&HQY$H ##Kx$HInn!S.$fC!'F1IQE LL   XsFE J Kr!c. t|jd}|dk(rK t|jd}|jd}|j ||j |<yy#ttf$rYywxYw#ttf$rYywxYw)z Set attributes based on OSC (Operating System Command) parameters. Parameters ---------- params : sequence of str The parameters for the command. rN)rEpop IndexErrorrG_parse_xterm_color_specr)rrQracolorspecs rrHzAnsiCodeProcessor.set_osc_codes &**Q-(G a< FJJqM*zz!}(,(D(DT(Ju% J'    +  s#A-A B-A?>A?BBc$|sy|jd}|dk(r|jnT|dk(r|jr d|_n:d|_n1|dk(r d|_n#|dk(r d|_n|dk(r d|_n|dk(rd|_d |_n|d k(rd |_n|d k(rd |_n|d k\r|d kr |d z |_n|dk(rF|rD|jd}|dk(r|r|jd|_n|dk(r|dd|_g|ddny|dk(rd|_nl|dk\r|dkr |dz |_nW|dk(rF|rD|jd}|dk(r|r|jd|_n!|dk(r|dd|_g|ddn |dk(rd|_|j|y)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. Nrr2Tr5rdF%&'(/01) rerbold_text_enabledr%r#r$r&r'r(r`)rrQrb _color_types rr`zAnsiCodeProcessor.set_sgr_codes zz!} 19 NN  QY%%  !" QYDN QYDK QY!DN RZDNDI RZDK RZ"DN RZDBJ$(2ID ! RZF **Q-KaF(. 1 %!(.r %r RZ$(D ! RZDBJ$(2ID ! RZF **Q-KaF(. 1 %!(.r %r RZ$(D ! &!r!c |jdr(ttd|ddjdS|jdr(ttd|ddjdS|dk(r t d |S) Nzrgb:ct|dS)N)rExs rz;AnsiCodeProcessor._parse_xterm_color_spec..s s1bzr!rd/zrgbi:c0tt|dzS)N)rEfloatr}s rrz;AnsiCodeProcessor._parse_xterm_color_spec..ss58c>':r!rr?zUnsupported xterm color spec)rBtuplerDrArG)rris rrgz)AnsiCodeProcessor._parse_xterm_color_specs} ??6 "148>>#3FGH H __W %:!!"X^^C023 3 S[;< < r!c~|jd}|dk(r'|jjtddddy)Nr2 r_r^page)grouprr@r)rrKspecials rr=z"AnsiCodeProcessor._replace_special!s6++a. d? LL   Xvvq I Jr!N)__name__ __module__ __qualname____doc__rxrr rrRrFrHr`rgr=r!rrr9sH   %=~,."LH,?"Jr!rceZdZdZidddddddd d d d d dddddddddddddddddddd Zej Zd%d!Zd"Zd#Z y$)&QtAnsiCodeProcessorz9 Translates ANSI escape codes into QTextCharFormats. rblackr2darkredr5 darkgreenrkbrownrddarkbluerr darkviolet steelbluegrey red lime yellow deepskyblue magentacyanwhitecJt|tr,|dkr |dkDr|dz }|jj|d}nt|tt fr|}nyt|t rtj|St|tt frtj|Sy)zp Returns a QColor for a given color code or rgb list, or None if one cannot be constructed. rrN) isinstancerErgetrrCstrrQColor)rrhr# constructors r get_colorzQtAnsiCodeProcessor.get_colorBs eS !qyY] ..,,UD9K t} -K k3 ' << , ,  eT] 3<<- -r!c<tj}|j|j|j}||j ||j|j |j}||j||jr*|jtjjn)|jtjj|j|j|j|j |S)zN Returns a QTextCharFormat that encodes the current style attributes. )rQTextCharFormatrr'r# setForegroundr( setBackgroundr% setFontWeightQFontBoldNormal setFontItalicr$setFontUnderliner&)rformatqcolors r get_formatzQtAnsiCodeProcessor.get_format]s&&( 5 5t~~F     ( 5 5t~~F     ( 99  !1!1 2  !3!3 4T[[)/ r!c<|jj|_t|sMt dD]!}|j||j|dz<#dx|jd<|jd<|j j |jy)zo Given a syntax style, attempt to set a color map that will be aesthetically pleasing. rrrrN)darkbg_color_maprrrrangerupdate)rstyleis rset_background_colorz(QtAnsiCodeProcessor.set_background_colorvs "&!6!6!;!;!=% 1X J040F0Fq0I&&q1u- JFM LD " "1 %(>(>r(B d445r!N)r) rrrrrrrrrrrr!rrr(s W Y [ W  Z  \  [ V V U V X ] Y V W!&)--/626r!r)r collectionsrreqtpyrqtconsole.stylesrrr rrrrr CSI_COMMANDSCSI_SUBPATTERNOSC_SUBPATTERN ANSI_PATTERNcompiler8r;objectrrrr!rrs# ((FG  &H I .*LM ""88*E?XJ7  xj 1 . ;# !L0%1 $"**%D|%ST"**X& ll^`6+`6r!