LdZddlmZddlmZmZddlmZddlm Z ddl m Z ddl m Z ddlZddlZddlZddlZd Zid d d d ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+Zid d,d d-dd.dd/dd0dd1dd2dd3dd4dd5dd6d d7d"d8d$d9d&d:d(d;d*d<Zid d=d d=dd>d*d?dd@ddAddBddCddDddEddFddGd dHd"dId$dJd&dKd(dLZeeee k(sJeeee k(sJeeee k(sJdZdMZGdNdOeZGdPdQeZedRSZedTSZeZGdUdVeZdWZGdXdYe Z y)[z Output for vt100 terminals. A lot of thanks, regarding outputting of colors, goes to the Pygments project: (We don't rely on Pygments anymore, because many things are very custom, and everything has been highly optimized.) http://pygments.org/ )unicode_literals)to_simple_filter Condition)Size)Output)ANSI_COLOR_NAMES)rangeN) Vt100_Output ansidefault' ansiblack ansidarkred ansidarkgreen ansibrown! ansidarkblue" ansipurple#ansiteal$ ansilightgray% ansidarkgrayZansired[ ansigreen\ ansiyellow]ansiblue^ ansifuchsia_ ansiturquoise` ansiwhitea1()*+,-./defghijkrrrr@r@rBrBrDrDrrrrFrrFrFr)rrrFrFrrFrrFrFrBrrrrBrrBrBr)rrrBrBrrBrrBrBc<t|tsJt||z t||z zt||z z}|dkDr|dz }d}d}tj D]8\}\}} } |dk7s||vs||z dz|| z dzz|| z dzz} | |ks5|}| }:|S)z Find closest ANSI color. Return it by name. :param r: Red (Between 0 and 255.) :param g: Green (Between 0 and 255.) :param b: Blue (Between 0 and 255.) :param exclude: A tuple of color names to exclude. (E.g. ``('ansired', )``.) r)rrr+r r ) isinstancetupleabsANSI_COLORS_TO_RGBitems) rgbexclude saturationdistancematchnamer2g2b2ds 7lib/third_party/prompt_toolkit/terminal/vt100_output.py_get_closest_ansi_colorrejs gu %% %QUc!a%j(3q1u:5JBNNH E0668lr2r = T%8RA RA -RA =A8| 9 Lc(eZdZdZddZddZddZy) _16ColorCachez Cache which maps (r, g, b) tuples to 16 ansi colors. :param bg: Cache for background colors, instead of foreground. c6t|tsJ||_yN)rSboolbg)selfrls rd__init__z_16ColorCache.__init__s"d###rfcF||f}||vr|j||||<||S)zx Return a (ansi_code, ansi_name) tuple. (E.g. ``(44, 'ansiblue')``.) for a given (r,g,b) value. )_get)rmvaluer[keys rdget_codez_16ColorCache.get_codes3 g d? %1DICyrfc||\}}}t||||}|jr t|}n t|}|||<||fS)Nr[)rerlBG_ANSI_COLORSFG_ANSI_COLORS)rmrqr[rXrYrZr^codes rdrpz_16ColorCache._getsM1a'1aA 77!%(D!%(DU U{rfN)F)__name__ __module__ __qualname____doc__rnrsrprzrfrdrhrhs  rfrhceZdZdZdZdZy)_256ColorCachez9 Cach which maps (r, g, b) tuples to 256 colors. cg}|jd|jd|jd|jd|jd|jd|jd|jd|jd |jd |jd |jd |jd |jd|jd|jdd}tdD]4}||dzdz}||dzdz}||dz}|j|||f6tddD]}d|dzz}|j|||f ||_y)Nr>rErGrH)rrrIrJrCr?rKrLrM)r"r"rBrNrOrA)rr(rBr )appendr colors)rmr valuerangeirXrYrZvs rdrnz_256ColorCache.__init__sl  () () () () () () () () () () () () () () () (): sAAGq=)AAFa<(A1q5!A MM1a) $ q"AAF A MM1a) $ rfc|\}}}d}d}t|jD]-\}\}} } ||z dz|| z dzz|| z dzz} | |ks*|}| }/|||<|S)NrQrrR) enumerater) rmrqrXrYrZr]r^rr`rarbrcs rd __missing__z_256ColorCache.__missing__s~1a(5OA|BRA RA -RA =A8| 6U  rfN)r{r|r}r~rnrrzrfrdrrs$LrfrF)rlTc*eZdZdZddZdZdZdZy)_EscapeCodeCachez Cache for VT100 escape codes. It maps (fgcolor, bgcolor, bold, underline, reverse) tuples to VT100 escape sequences. :param true_color: When True, use 24bit colors instead of 256 colors. cVt|tsJ||_t||_yrj)rSrk true_colorransi_colors_only)rmrrs rdrnz_EscapeCodeCache.__init__s'*d+++$ 01A Brfc`|\}}}}}}}g} | j|j|||r| jd|r| jd|r| jd|r| jd|r| jd| rddj| zdz} nd } | ||<| S) N13547z[0;;m)extend_colors_to_coderjoin) rmattrsfgcolorbgcolorbold underlineitalicblinkreversepartsresults rdrz_EscapeCodeCache.__missing__sDIA$ 65' T))'7;<  LL   LL   LL   LL   LL  %036FFU  rfcl t|d}|dz dz}|dz dz}|dz}|||fS#t$rwxYw)z) Turn 'ffffff', into (0xff, 0xff, 0xff). rBr)int ValueError)rmcolorrgbrXrYrZs rd_color_name_to_rgbz#_EscapeCodeCache._color_name_to_rgbsX eR.Cd"AT!Ad Aa7N    s ( 3cdgfd}g}|j|d|j|dttj|S)zB Return a tuple with the vt100 values that represent this color. rzc|rtnt}|y||vr||fS j|} j rM|r+ k7r df}nd}t j ||\}}|fStj |\}}| d<|fS jr|\}}} |rdndd||| fS|rdnddt|fS#t$rYywxYw)Nrzrru0&rR) rvrwrrr _16_bg_colorsrs _16_fg_colorsr _256_colors)rrltablerr[rxr_rXrYrZbg_colorfg_ansifg_colorrms rdgetz-_EscapeCodeCache._colors_to_code..get)s&(NnE}%e ''11%8C ((*#x/'.qznG&(G%2%;%;C%;%Q d $x%2%;%;C%@ d%) $x__!GAq!"$B"aAq99#%B"aS1ABB1"sB99 CCFT)rmapsix text_type)rmrrrrrs``` @rdrz _EscapeCodeCache._colors_to_code"sQ $& CP c(E*+ c(D)*3==&))rfN)FF)r{r|r}r~rnrrrrzrfrdrrs C 2 3*rfrcddl}ddl}tjtjrdndgd}|j ||j ||d|dfS)zw Get the size of this pseudo terminal. :param fileno: stdout.fileno() :returns: A (rows, cols) tuple. rNhh)rrrrr)fcntltermiosarrayrPY2ioctl TIOCGWINSZ)filenorrbufs rd _get_sizerXsP ++cggd4 >C  KK**C0 q63q6>rfceZdZdZ d"dZed#dZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZd$dZdZdZdZdZdZdZ dZ!d Z"d!Z#y)%r a :param get_size: A callable which returns the `Size` of the output terminal. :param stdout: Any object with has a `write` and `flush` method + an 'encoding' property. :param true_color: Use 24bit color instead of 256 colors. (Can be a :class:`SimpleFilter`.) When `ansi_colors_only` is set, only 16 colors are used. :param ansi_colors_only: Restrict to 16 ANSI colors only. :param term: The terminal environment variable. (xterm, xterm-256color, linux, ...) :param write_binary: Encode the output before writing it. If `True` (the default), the `stdout` object is supposed to expose an `encoding` attribute. Nct|sJttjsJt fddDsJ|rt dsJg|_|_||_||_ t||_ xsd|_ |9ttjj!ddt"fd}n t|}||_t'||_t'd | |_y) Nc36K|]}t|ywrj)hasattr).0astdouts rd z(Vt100_Output.__init__..sB/A!761%/As)writeflushencodingxtermPROMPT_TOOLKIT_ANSI_COLORS_ONLYFcxsdvS)Nlinuxz eterm-colorrz)ANSI_COLORS_ONLYtermsrdrz/Vt100_Output.__init__..ansi_colors_onlys'K43K+KKrf)rT)rr)callablerSrrallr_bufferr write_binaryget_sizerrrrkosenvironrrrr_escape_code_cache_escape_code_cache_true_color)rmrrrrrrrs ` ` @rdrnzVt100_Output.__init__~s!!!|z$ >>>B/ABBBB 6:. ..  (  *:6OG   ##BJJNN15%: ;  L L 00@A  0#3DT"U-=.>.@*rfcLjsJfd}|||||S)z Create an Output class from a pseudo terminal. (This will take the dimensions by reading the pseudo terminal attributes.) cVtj\}}t||S)N)rowscolumns)rrr)rrrs rdrz'Vt100_Output.from_pty..get_sizes$%fmmo6MD'T73 3rf)rrr)isatty)clsrrrrrs ` rdfrom_ptyzVt100_Output.from_ptys4}} 468 $44A Arfc6|jjS)z Return file descriptor. )rrrms rdrzVt100_Output.filenos{{!!##rfc.|jjS)z" Return encoding used for stdout. )rrrs rdrzVt100_Output.encodings{{###rfc:|jj|y)z+ Write raw data to output. N)rrrmdatas rd write_rawzVt100_Output.write_raws D!rfcZ|jj|jddy)zn Write text to output. (Removes vt100 escape codes. -- used for safely writing text.) ?N)rrreplacers rdrzVt100_Output.writes! DLL56rfc|jdvr5|jd|jddjddzyy)z% Set terminal title. rz]2;%srN)rrr)rmtitles rd set_titlezVt100_Output.set_titles@ 994 4 NN?U]]62-F-N-NvWY-ZZ [ 5rfc&|jdy)Nr)rrs rd clear_titlezVt100_Output.clear_titles rrfc&|jdy)zd Erases the screen with the background colour and moves the cursor to home. zNrrs rd erase_screenzVt100_Output.erase_screens y!rfc&|jdy)Nz [?1049hrrs rdenter_alternate_screenz#Vt100_Output.enter_alternate_screens *+rfc&|jdy)Nz[?1049lrrs rdquit_alternate_screenz"Vt100_Output.quit_alternate_screen }%rfcj|jd|jd|jdy)Nz[?1000hz[?1015hz[?1006hrrs rdenable_mouse_supportz!Vt100_Output.enable_mouse_supports* }% }% }%rfcj|jd|jd|jdy)Nz[?1000lz[?1015lz[?1006lrrs rddisable_mouse_supportz"Vt100_Output.disable_mouse_supports& }% }% }%rfc&|jdy)zY Erases from the current cursor position to the end of the current line. zNrrs rderase_end_of_linezVt100_Output.erase_end_of_lines x rfc&|jdy)zc Erases the screen from the current line down to the bottom of the screen. zNrrs rd erase_downzVt100_Output.erase_downs x rfc&|jdy)Nrrrs rdreset_attributeszVt100_Output.reset_attributess y!rfc|jr/|js|j|j|y|j|j|y)zW Create new style and output. :param attrs: `Attrs` instance. N)rrrrr)rmrs rdset_attributeszVt100_Output.set_attributessG ?? T%:%:%< NN4==eD E NN42259 :rfc&|jdy)Nz[?7lrrs rddisable_autowrapzVt100_Output.disable_autowrap z"rfc&|jdy)Nz[?7hrrs rdenable_autowrapzVt100_Output.enable_autowrap rrfc&|jdy)Nz[?2004hrrs rdenable_bracketed_pastez#Vt100_Output.enable_bracketed_paster rfc&|jdy)Nz[?2004lrrs rddisable_bracketed_pastez$Vt100_Output.disable_bracketed_paster rfc0|jd||fzy)z Move cursor position. z[%i;%iHNr)rmrowcolumns rd cursor_gotozVt100_Output.cursor_gotos }V}45rfcf|dk(ry|dk(r|jdy|jd|zy)Nrrzz[%iArrmamounts rd cursor_upzVt100_Output.cursor_up0 Q;  q[ NN8 $ NN:. /rfcf|dk(ry|dk(r|jdy|jd|zy)Nrrzz[%iBrr(s rd cursor_downzVt100_Output.cursor_down s2 Q;  q[ NN8 $ NN:. /rfcf|dk(ry|dk(r|jdy|jd|zy)Nrrzz[%iCrr(s rdcursor_forwardzVt100_Output.cursor_forward*r+rfcf|dk(ry|dk(r|jdy|jd|zy)Nrrz[%iDrr(s rdcursor_backwardzVt100_Output.cursor_backward2s0 Q;  q[ NN4 NN:. /rfc&|jdy)Nz[?25lrrs rd hide_cursorzVt100_Output.hide_cursor:s {#rfc&|jdy)Nz [?12l[?25hrrs rd show_cursorzVt100_Output.show_cursor=s +,rfc|jsydj|j} |jrst|jdr|jj }n |j}|j |j|jjxsddn|jj ||jjg|_y#t$r^}|jr!|jdtjk(rn |jr|jddk(rnYd}~g|_yd}~wwxYw)z3 Write to output stream and flush. Nrbufferzutf-8rr)rrrrrr8rencoderrIOErrorargserrnoEINTR)rmroutes rdrzVt100_Output.flush@s|| wwt||$    4;;1++,,C++C $++dkk&:&:&EgyQR !!$' KK   $ # vv!&&)u{{2AFF1IN # sB4C&& E /A EE cF|jd|jy)z: Asks for a cursor position report (CPR). zNrrrs rd ask_for_cprzVt100_Output.ask_for_cprls y! rfcF|jd|jy)z Sound bell. rNrArs rdbellzVt100_Output.bellss t rf)FNNT)FNN)rr)$r{r|r}r~rn classmethodrrrrrrrrr r rrrrrrrrr r"r&r*r-r/r2r4r6rrBrDrzrfrdr r ss 5:@D!@F A A$$" 7\",&&& ! !" ;##&&60000$-*Xrfr ry)!r~ __future__rprompt_toolkit.filtersrrprompt_toolkit.layout.screenrprompt_toolkit.rendererrprompt_toolkit.stylesr six.movesr rr<rr__all__rwrvrVsetredictrhrrrrrrr rzrfrdrOsb(>-*2   2r  r  r rrrrrrr r!"r#$r%&r'(r)*r+0rr  r  r rrrrrss s!"s#$s%&s'(s)*s+2''''  ' )))))) )!")#$)%&)'())*)+2 >c"23333 >c"23333 #&6"7777DDD;T;|' &  d*td*N6C6Crf