U ou fY@sdZddlmZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z ddlmZddlmZddlmZddlmZmZdd lmZd d lmZd d lmZd gZddddddddddddddddddZd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0dZd1d1d2d3d4d5d6d7d8d9d:d;dd?d@dAZe ee eks8t!e ee eksNt!e ee eksdt!dWdCdCdCdDdEdFdGdHZ"e e#e$fZ%GdIdJdJZ&GdKdLdLee e#e#e#fe#fZ'e&dMdNZ(e&dOdNZ)e'Z*GdPdQdQeee$fZ+dCdRdSdTdUZ,GdVd d eZ-dS)Xz 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/ ) annotationsN)CallableDictHashableIterableSequenceTextIOTuple) CursorShape)Size)Output)ANSI_COLOR_NAMESAttrs)is_dumb_terminal) ColorDepth) flush_stdout Vt100_Output' !"#$%Z[\]^_`a) ansidefault ansiblackansired ansigreen ansiyellowansiblue ansimagentaansicyanansigrayansibrightblack ansibrightredansibrightgreenansibrightyellowansibrightblueansibrightmagentaansibrightcyan ansiwhite1()*+,-./defghijkrrrrIrIrKrKrMrMrrrrOrrOrOr)rrrOrOrrOrrOrOrMrrrrMrrMrMr)rrrMrMrrMrrMrM)r%r&r-r.r5r'r(r)r*r+r,r/r0r1r2r3r4int Sequence[str]str)rgbexcludereturnc Cst|}t||t||t||}|dkrF|ddddgd}d}tD]R\}\}} } |dkrV||krV||d|| d|| d} | |krV|}| }qV|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', )``.) rZ ansilightgrayZ ansidarkgrayr5r&r%)listabsextendANSI_COLORS_TO_RGBitems) r]r^r_r`Z saturationdistancematchnamer2g2b2drYrY?/tmp/pip-unpacked-wheel-8milen2s/prompt_toolkit/output/vt100.py_get_closest_ansi_colorks $$rqc@sJeZdZdZddddddZdd d d d d dZdd d d d ddZdS) _16ColorCachez Cache which maps (r, g, b) tuples to 16 ansi colors. :param bg: Cache for background colors, instead of foreground. FboolNone)bgracCs||_i|_dSN)ru_cache)selfrurYrYrp__init__sz_16ColorCache.__init__rYtuple[int, int, int]r[_ColorCodeAndName)valuer`racCs2|t|f}|j}||kr*|||||<||S)zx Return a (ansi_code, ansi_name) tuple. (E.g. ``(44, 'ansiblue')``.) for a given (r,g,b) value. )tuplerw_get)rxr|r`keycacherYrYrpget_codes  z_16ColorCache.get_codecCs:|\}}}t||||d}|jr*t|}nt|}||fS)Nr`)rqruBG_ANSI_COLORSFG_ANSI_COLORS)rxr|r`r]r^r_rjcoderYrYrpr~s   z_16ColorCache._getN)F)rY)rY)__name__ __module__ __qualname____doc__ryrr~rYrYrYrprrs rrc@s.eZdZdZddddZdddd d Zd S) _256ColorCachez: Cache which maps (r, g, b) tuples to 256 colors. rtracCs(g}|d|d|d|d|d|d|d|d|d |d |d |d |d |d|d|dd}tdD]@}||dd}||dd}||d}||||fqtddD] }d|d}||||fq||_dS)NrGrNrPrQ)rrrRrSrHrJrTrUrV)rrrMrWrXrL)rr"rMrr )appendrangecolors)rxrZ valuerangeir]r^r_vrYrYrprys6                   z_256ColorCache.__init__rzrZ)r|rac Cst|\}}}d}d}t|jD]J\}\}} } |dkr||d|| d|| d} | |kr|}| }q|||<|S)Nrbrrc) enumerater) rxr|r]r^r_rirjrrlrmrnrorYrYrp __missing__s $z_256ColorCache.__missing__N)rrrrryrrYrYrYrprs&rF)ruTc@sReZdZdZdddddZddd d d Zdd d ddZddddddZdS)_EscapeCodeCachez Cache for VT100 escape codes. It maps (fgcolor, bgcolor, bold, underline, strike, reverse) tuples to VT100 escape sequences. :param true_color: When True, use 24bit colors instead of 256 colors. rrt) color_depthracCs ||_dSrv)r)rxrrYrYrpry sz_EscapeCodeCache.__init__rr\)attrsrac Cs|\ }}}}}}}} } g} | ||p(d|p.d|rB| d|rP| d|r^| d|rl| d| rz| d| r| d|r| d| rd d | d } nd } | ||<| S) N1354789z[0;;m)rf_colors_to_coderjoin) rxrZfgcolorZbgcolorZboldZ underlinestrikeZitalicZblinkreversehiddenpartsresultrYrYrpr s@        z_EscapeCodeCache.__missing__rz)colorracCsTzt|d}Wntk r$Yn,X|d?d@}|d?d@}|d@}|||fSdS)z'Turn 'ffffff', into (0xff, 0xff, 0xff).rrMrN)rZ ValueError)rxrrgbr]r^r_rYrYrp_color_name_to_rgb3s  z#_EscapeCodeCache._color_name_to_rgbz Iterable[str])fg_colorbg_colorracsNdddddfdd }g}||d||d tt|S) zR Return a tuple with the vt100 values that represent this color. rr\rsz list[int])rrurac s|rtnt}|rjtjkr gS||kr2||gSz|}Wntk rXgYSXjtjkr|rkrzg}ng}tj ||d\}}|gSt |\}}||gSnBjtj kr|\}}} |rdndd||| gS|rdnddt |gSdS)Nr0&rc) rrrr DEPTH_1_BITrr DEPTH_4_BIT _16_bg_colorsr _16_fg_colors DEPTH_24_BIT _256_colors) rrutablerr`rrkr]r^r_rZfg_ansirrxrYrpgetHs.      z-_EscapeCodeCache._colors_to_code..getFT)rfmapr\)rxrrrrrYrrpr?s *z _EscapeCodeCache._colors_to_codeN)rrrrryrrrrYrYrYrprs ' rztuple[int, int])filenoracCst|}|j|jfS)zw Get the size of this pseudo terminal. :param fileno: stdout.fileno() :returns: A (rows, cols) tuple. )osget_terminal_sizelinescolumns)rsizerYrYrp _get_sizeys rc @s`eZdZUdZeZded<dfdddd d d d d d dZedgddd d ddddZ ddddZ ddddZ ddddZ dd dddZ dd ddd Zdd d!d"d#Zd dd$d%Zd dd&d'Zd dd(d)Zd dd*d+Zd dd,d-Zd dd.d/Zd dd0d1Zd dd2d3Zd dd4d5Zd6d7d d8d9d:Zd dd;d<Zd dd=d>Zd dd?d@Zd ddAdBZd ddCdDZdhddd dFdGdHZdd dIdJdKZ dd dIdLdMZ!dd dIdNdOZ"dd dIdPdQZ#d ddRdSZ$d ddTdUZ%dVd dWdXdYZ&d ddZd[Z'd dd\d]Z(d dd^d_Z)e*d dd`daZ+d ddbdcZ,d7ddddeZ-dS)ira :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 term: The terminal environment variable. (xterm, xterm-256color, linux, ...) :param enable_cpr: When `True` (the default), send "cursor position request" escape sequences to the output in order to detect the cursor position. That way, we can properly determine how much space there is available for the UI (especially for drop down menus) to render. The `Renderer` will still try to figure out whether the current terminal does respond to CPR escapes. When `False`, never attempt to send CPR requests. zset[int]_fds_not_a_terminalNTrzCallable[[], Size]z str | NonezColorDepth | Nonersrt)stdoutget_sizetermdefault_color_depth enable_bell enable_cprrac stfdddDstg|_|_||_||_||_||_||_t j t t j t j t t j t j t t j t jt t ji|_d|_dS)Nc3s|]}t|VqdSrv)hasattr).0arrYrp sz(Vt100_Output.__init__..)writeflushF)allAssertionError_bufferrrrrrrrrrrZ DEPTH_8_BITr_escape_code_caches_cursor_shape_changed)rxrrrrrrrYrrprys$  zVt100_Output.__init__)rrrrracsz }Wntjk r&d}YnXst|dksB||jkrtd}tj||tj|dk rt|j |ddfdd }|||||dS)z Create an Output class from a pseudo terminal. (This will take the dimensions by reading the pseudo terminal attributes.) Nz+Warning: Output is not a terminal (fd=%r). r rcsFd\}}zt\}}Wntk r0YnXt|p:d|p@ddS)N)NNPZrowsr)rrOSErrorr rrrYrprs z'Vt100_Output.from_pty..get_size)rrr) rioUnsupportedOperationisattyrsysstderrrradd)clsrrrrfdmsgrrYrrpfrom_ptys$    zVt100_Output.from_ptyr rcCs|Srv)rrxrYrYrprszVt100_Output.get_sizerZcCs |jS)zReturn file descriptor.)rrrrYrYrprszVt100_Output.filenor\cCs|jjS)z Return encoding used for stdout.)rencodingrrYrYrprszVt100_Output.encoding)dataracCs|j|dS)z+ Write raw data to output. N)rrrxrrYrYrp write_rawszVt100_Output.write_rawcCs|j|dddS)zn Write text to output. (Removes vt100 escape codes. -- used for safely writing text.) ?N)rrreplacerrYrYrprszVt100_Output.write)titleracCs,|jdkr(|d|dddddS)z% Set terminal title. linuxz eterm-colorz]2;%srrN)rrr)rxrrYrYrp set_titles zVt100_Output.set_titlecCs|ddS)Nr)rrrYrYrp clear_titleszVt100_Output.clear_titlecCs|ddS)zc Erases the screen with the background color and moves the cursor to home. zNrrrYrYrp erase_screenszVt100_Output.erase_screencCs|ddS)Nz [?1049hrrrYrYrpenter_alternate_screensz#Vt100_Output.enter_alternate_screencCs|ddS)Nz[?1049lrrrYrYrpquit_alternate_screensz"Vt100_Output.quit_alternate_screencCs,|d|d|d|ddS)Nz[?1000hz[?1003hz[?1015hz[?1006hrrrYrYrpenable_mouse_supports   z!Vt100_Output.enable_mouse_supportcCs,|d|d|d|ddS)Nz[?1000lz[?1015lz[?1006lz[?1003lrrrYrYrpdisable_mouse_support.s   z"Vt100_Output.disable_mouse_supportcCs|ddS)zY Erases from the current cursor position to the end of the current line. zNrrrYrYrperase_end_of_line4szVt100_Output.erase_end_of_linecCs|ddS)zc Erases the screen from the current line down to the bottom of the screen. zNrrrYrYrp erase_down:szVt100_Output.erase_downcCs|ddS)NrrrrYrYrpreset_attributesAszVt100_Output.reset_attributesrr)rrracCs|j|}|||dS)zW Create new style and output. :param attrs: `Attrs` instance. N)rr)rxrrZescape_code_cacherYrYrpset_attributesDs zVt100_Output.set_attributescCs|ddS)Nz[?7lrrrYrYrpdisable_autowrapPszVt100_Output.disable_autowrapcCs|ddS)Nz[?7hrrrYrYrpenable_autowrapSszVt100_Output.enable_autowrapcCs|ddS)Nz[?2004hrrrYrYrpenable_bracketed_pasteVsz#Vt100_Output.enable_bracketed_pastecCs|ddS)Nz[?2004lrrrYrYrpdisable_bracketed_pasteYsz$Vt100_Output.disable_bracketed_pastecCs|ddS)zh For vt100 only. Put the terminal in cursor mode (instead of application mode). z[?1lNrrrYrYrpreset_cursor_key_mode\sz"Vt100_Output.reset_cursor_key_moder)rowcolumnracCs|d||fdS)z' Move cursor position. z[%i;%iHNr)rxr r rYrYrp cursor_gotodszVt100_Output.cursor_goto)amountracCs0|dkr n"|dkr|dn|d|dS)Nrrzz[%iArrxr rYrYrp cursor_upjs  zVt100_Output.cursor_upcCs0|dkr n"|dkr|dn|d|dS)Nrrzz[%iBrrrYrYrp cursor_downrs  zVt100_Output.cursor_downcCs0|dkr n"|dkr|dn|d|dS)Nrrzz[%iCrrrYrYrpcursor_forward|s  zVt100_Output.cursor_forwardcCs0|dkr n"|dkr|dn|d|dS)Nrrz[%iDrrrYrYrpcursor_backwards  zVt100_Output.cursor_backwardcCs|ddS)Nz[?25lrrrYrYrp hide_cursorszVt100_Output.hide_cursorcCs|ddS)Nz [?12l[?25hrrrYrYrp show_cursorszVt100_Output.show_cursorr ) cursor_shaperacCsN|tjkrdSd|_|tjdtjdtjdtjdtjdtj di |ddS) NTz[2 qz[6 qz[4 qz[1 qz[5 qz[3 qr) r Z _NEVER_CHANGErrZBLOCKZBEAMZ UNDERLINEZBLINKING_BLOCKZ BLINKING_BEAMZBLINKING_UNDERLINEr)rxrrYrYrpset_cursor_shapes* zVt100_Output.set_cursor_shapecCs|jrd|_|ddS)zReset cursor shape.Fz[0 qN)rrrrYrYrpreset_cursor_shapeszVt100_Output.reset_cursor_shapecCs,|js dSd|j}g|_t|j|dS)z3 Write to output stream and flush. Nr)rrrrrrYrYrprs  zVt100_Output.flushcCs|d|dS)z: Asks for a cursor position report (CPR). zN)rrrrYrYrp ask_for_cprs zVt100_Output.ask_for_cprcCsV|js dStjdddkr dSt|jr.dSz |jWStk rPYdSXdS)NFZPROMPT_TOOLKIT_NO_CPRrr) rrenvironrrrrrrrrYrYrpresponds_to_cprs  zVt100_Output.responds_to_cprcCs|jr|d|dS)z Sound bell.rN)rrrrrYrYrpbells zVt100_Output.bellcCsF|jdk r|jS|j}|dkr$tjSt|r2tjS|dkr@tjStjS)z Return the default color depth for a vt100 terminal, according to the our term value. We prefer 256 colors almost always, because this is what most terminals support these days, and is a good default. Nr)rrrDEFAULTrrr)rxrrYrYrpget_default_color_depths z$Vt100_Output.get_default_color_depth)NNTT)NNT)rr).rrrrsetr__annotations__ry classmethodrrrrrrrrrrrrrrrrrrrrrr r rrrrrrrrrrpropertyrrrrYrYrYrprs`  4     )rY).r __future__rrrrtypingrrrrrrr Zprompt_toolkit.cursor_shapesr Zprompt_toolkit.data_structuresr Zprompt_toolkit.outputr Zprompt_toolkit.stylesr rZprompt_toolkit.utilsrrrr__all__rrrgrrrqrZr\r{rrrrrrrrrrYrYrYrps $      " )"B  y