ó _y]c@s`dZddlmZddlZddlmZddlmZddlm Z eddd d gƒZ ed dd d dgƒZ eddd d dgƒZ eddgƒZ eddgƒZeddgƒZeddgƒZdZdeZdZdeefZejdeƒZejdƒZdefd„ƒYZdefd„ƒYZdS(sJ Utilities for processing ANSI escape codes and special ASCII characters. iÿÿÿÿ(t namedtupleN(tQtGui(t string_types(t dark_stylet EraseActiontactiontareaterase_tot MoveActiontdirtunittcountt ScrollActiontCarriageReturnActiont NewLineActiont BeepActiontBackSpaceActiontABCDEFGHJKSTfmnsus \[(.*?)([%s])s \](.*?)[]s ?(%s|%s)?s(|| (?! )| ? )|(?:%s)s([ ])tAnsiCodeProcessorcBseeZdZeZiZd„Zd„Zd„Zgd„Z d„Z d„Z d„Z d„Z RS( sœ Translates special ASCII characters and ANSI escape codes into readable attributes. It also supports a few non-standard, xterm-specific codes. cCs)g|_|jjƒ|_|jƒdS(N(tactionstdefault_color_maptcopyt color_mapt reset_sgr(tself((s<lib/python2.7/site-packages/qtconsole/ansi_code_processor.pyt__init__Ks cCs:d|_t|_t|_t|_d|_d|_dS(s; Reset graphics attributs to their default values. iN(t intensitytFalsetitalictboldt underlinetNonetforeground_colortbackground_color(R((s<lib/python2.7/site-packages/qtconsole/ansi_code_processor.pyRPs      c csÖg|_d}t|ƒdkr7|ddkr7dnd}|dk rS|d n|}xtj|ƒD]}|||jƒ!}tj|j|ƒ}|s¦|jr·|Vg|_n|j ƒ}g|j ƒD]}|dk rÐ|^qÐ}|d} | dkr+|jj t dƒƒdVg|_qi| dkr^|jj t dƒƒdVg|_qi| dkr‘|jj td ƒƒdVg|_qi| dks©| d krÐ|jj td ƒƒ| Vg|_qig|d jd ƒD]} | rä| ^qä} | jdƒrOyttt| ƒƒ} Wntk r7qnX|j|d| ƒqi| jdƒri|j| ƒqiqiW||}tj|j|ƒ}|s |jr¨|Vn|dk rÒ|jj td ƒƒ|VndS(sC Yields substrings for which the same escape code applies. iiÿÿÿÿs stbeeps scarriage-returnst backspaces tnewlineit;t[it]N(RtlenRtANSI_OR_SPECIAL_PATTERNtfinditertstarttSPECIAL_PATTERNtsubt_replace_specialtendtgroupstappendRR RRtsplitt startswithtlisttmaptintt ValueErrort set_csi_codet set_osc_code( RtstringR+t last_chartmatchtrawt substringtgR0tg0tparamtparams((s<lib/python2.7/site-packages/qtconsole/ansi_code_processor.pyt split_stringZsZ .  +        ,   cCsd|dkr5|r"|j|ƒq`|jdgƒn+|dksM|dkrø|r]|dnd}d|kozdknr`|dkr‘dnd}|dkr¬d}n*|d krÁd }n|dkrÖd }n|jjtd ||ƒƒq`nh|d ks|dkr`|d kr"dnd}|r8|dnd }|jjtd|d|ƒƒndS(s3 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. tmitJtKitscreentlineR/iR+tallterasetStTtuptdowntscrollN(t set_sgr_codeRR1RR (RtcommandRBtcodeRRR R ((s<lib/python2.7/site-packages/qtconsole/ansi_code_processor.pyR8™s*         "  cCsŸyt|jdƒƒ}Wnttfk r3dSX|dkr›y>t|jdƒƒ}|jdƒ}|j|ƒ|j|tit/srgbi:cSstt|ƒdƒS(Niÿ(R6tfloat(R[((s<lib/python2.7/site-packages/qtconsole/ansi_code_processor.pyR\R]it?sUnsupported xterm color spec(R3ttupleR5R2R7(RRW((s<lib/python2.7/site-packages/qtconsole/ansi_code_processor.pyRUs#  cCsA|jdƒ}|dkr=|jjtddddƒƒndS(Nis RORNtpageR](tgroupRR1R (RR<tspecial((s<lib/python2.7/site-packages/qtconsole/ansi_code_processor.pyR."s "(t__name__t __module__t__doc__RRXRRRRCR8R9RPRUR.(((s<lib/python2.7/site-packages/qtconsole/ansi_code_processor.pyR:s  ? $  E tQtAnsiCodeProcessorcBs®eZdZidd6dd6dd6dd6d d 6d d 6d d6dd6dd6dd6dd6dd6dd6dd6dd6dd6ZejƒZdd „Zd!„Zd"„ZRS(#s9 Translates ANSI escape codes into QTextCharFormats. tblackitdarkredit darkgreenitbrownitdarkblueit darkvioletit steelblueitgreyiitredi tlimei tyellowi t deepskybluei tmagentai tcyanitwhiteicCs°t|tƒrL|dkr4|dkr4|d7}n|jj|dƒ}n"t|ttfƒrj|}ndSt|tƒrŠtj |ƒSt|ttfƒr¬tj |ŒSdS(sp Returns a QColor for a given color code or rgb list, or None if one cannot be constructed. iiN( t isinstanceR6RtgetRRaR4RRtQColor(RRVRt constructor((s<lib/python2.7/site-packages/qtconsole/ansi_code_processor.pyt get_colorCs    cCsÊtjƒ}|j|j|jƒ}|dk r@|j|ƒn|j|j|jƒ}|dk rt|j|ƒn|j r“|j tj j ƒn|j tj j ƒ|j|jƒ|j|jƒ|S(sN Returns a QTextCharFormat that encodes the current style attributes. N(RtQTextCharFormatR|R RRt setForegroundR!t setBackgroundRt setFontWeighttQFonttBoldtNormalt setFontItalicRtsetFontUnderlineR(Rtformattqcolor((s<lib/python2.7/site-packages/qtconsole/ansi_code_processor.pyt get_format^s    cCs|jjƒ|_t|ƒshx,tdƒD]}|j||j|ds*   ï