B 't\lM@sdZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l Z dd lZdd lZdd lZd gZd d d dddddddddddddddZdddd d!d"d#d$d%d&d'd(d)d*d+d,d-dZd.d.d/d0d1d2d3d4d5d6d7d8d9d:d;dZeeeeksteeeeksteeeeks*tdNd@dAZGdBdCdCeZGdDdEdEeZedFdGZedHdGZeZGdIdJdJeZdKdLZGdMd d eZ d S)Oz 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)Size)Output)ANSI_COLOR_NAMES) ColorDepth)rangeN Vt100_Output' !"#$%Z[\]^_`a) ansidefault ansiblackansired ansigreen ansiyellowansiblue ansimagentaansicyanansigrayansibrightblack ansibrightredansibrightgreenansibrightyellowansibrightblueansibrightmagentaansibrightcyan ansiwhite1()*+,-./defghijk)rrr)r=r=)r>r>)r?r?)rr)rr@r)r@r@r)rrr@)r@rr@)rr@r@)r?rr)rr?r)r?r?r)rrr?)r?rr?)rr?r?)rrr#r$r+rrrr r!r"r%r&r'r(r)r*c Cst|tstt||t||t||}|dkrB|d7}d}d}x^tD]R\}\}} } |dkrT||krT||d|| d|| d} | |krT|}| }qTW|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 )Z ansilightgrayZ ansidarkgrayr+rir) isinstancetupleAssertionErrorabsANSI_COLORS_TO_RGBitems) rgbexcludeZ saturationdistancematchnamer2g2b2drArA:lib/python3.7/site-packages/prompt_toolkit/output/vt100.py_get_closest_ansi_colorks $$rUc@s.eZdZdZd ddZd ddZd dd Zd S) _16ColorCachez Cache which maps (r, g, b) tuples to 16 ansi colors. :param bg: Cache for background colors, instead of foreground. FcCst|tst||_dS)N)rCboolrEbg)selfrXrArArT__init__sz_16ColorCache.__init__rAcCs(||f}||kr |||||<||S)zx Return a (ansi_code, ansi_name) tuple. (E.g. ``(44, 'ansiblue')``.) for a given (r,g,b) value. )_get)rYvaluerLkeyrArArTget_codesz_16ColorCache.get_codecCsB|\}}}t||||d}|jr*t|}nt|}|||<||fS)N)rL)rUrXBG_ANSI_COLORSFG_ANSI_COLORS)rYr\rLrIrJrKrNcoderArArTr[s  z_16ColorCache._getN)F)rA)rA)__name__ __module__ __qualname____doc__rZr^r[rArArArTrVs  rVc@s eZdZdZddZddZdS)_256ColorCachez: Cache which maps (r, g, b) tuples to 256 colors. cCs2g}|d|d|d|d|d|d|d|d|d |d |d |d |d |d|d|dd}xLtdD]@}||dd}||dd}||d}||||fqWx0tddD]"}d|d}||||fqW||_dS)N)rrr)r@rr)rr@r)r@r@r)rr)r@rr@)rr@r@)r=r=r=)r>r>r>)r?rr)rr?r)r?r?r)rrr?)r?rr?)rr?r?)r?r?r?)rrr?rr )appendrcolors)rYrqZ valuerangeirIrJrKvrArArTrZs6                  z_256ColorCache.__init__c Csx|\}}}d}d}xXt|jD]J\}\}} } |dkr||d|| d|| d} | |kr|}| }qW|||<|S)NirrB) enumeraterq) rYr\rIrJrKrMrNrrrPrQrRrSrArArT __missing__s $z_256ColorCache.__missing__N)rbrcrdrerZrvrArArArTrfs&rfF)rXTc@s0eZdZdZddZddZddZdd Zd S) _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. cCs|tjkst||_dS)N)rZ_ALLrE color_depth)rYrxrArArTrZsz_EscapeCodeCache.__init__c Cs|\}}}}}}}} g} | ||||r8| d|rF| d|rT| d|rb| d|rp| d| r~| d| rdd| d } nd } | ||<| S) N135478z[0;;mz)extend_colors_to_coderpjoin) rYattrsZfgcolorZbgcolorZboldZ underlineZitalicZblinkreverseZhiddenpartsresultrArArTrvs(      z_EscapeCodeCache.__missing__cCsTyt|d}Wntk r$Yn,X|d?d@}|d?d@}|d@}|||fSdS)z) Turn 'ffffff', into (0xff, 0xff, 0xff). rtr?rnN)int ValueError)rYcolorrgbrIrJrKrArArT_color_name_to_rgbs  z#_EscapeCodeCache._color_name_to_rgbcsHdgfdd}g}||d||dttj|S)zB Return a tuple with the vt100 values that represent this color. rAc s|rtnt}|rjtjkr dS||kr2||fSy|}Wntk rTdSXjtjkr|rkrzdf}nd}tj ||d\}}|fSt |\}}|d<|fSnBjtj kr|\}}} |rdndd||| fS|rdnddt |fSdS)NrAr)rL0&rB) r_r`rxr DEPTH_1_BITrr DEPTH_4_BIT _16_bg_colorsr^ _16_fg_colors DEPTH_24_BIT _256_colors) rrXtablerrLrarOrIrJrK)bg_colorfg_ansifg_colorrYrArTget.s.      z-_EscapeCodeCache._colors_to_code..getFT)rmapsix text_type)rYrrrrrA)rrrrYrTr's (z _EscapeCodeCache._colors_to_codeN)rbrcrdrerZrvrrrArArArTrws  rwcCsNddl}ddl}ttjrdndddddg}|||j||d|dfS)zw Get the size of this pseudo terminal. :param fileno: stdout.fileno() :returns: A (rows, cols) tuple. rNhhr)fcntltermiosarrayrZPY2ZioctlZ TIOCGWINSZ)filenorrZbufrArArT _get_size]s rc@seZdZdZeZdCddZedDddZdd Z d d Z d d Z ddZ ddZ ddZddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-ZdEd/d0Zd1d2Zd3d4Zd5d6Zd7d8Z d9d:Z!d;d<Z"d=d>Z#d?d@Z$dAdBZ%dS)Fr 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 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. NTc st|s t|dks$t|tjs$ttfdddDs>t|rPtdsPtg|_|_||_ ||_ |pnd|_ t j tt j t jtt jt jtt jt jtt ji|_dS)Nc3s|]}t|VqdS)N)hasattr).0a)stdoutrArT sz(Vt100_Output.__init__..)writeflushencodingZxterm)callablerErCrrallr_bufferr write_binaryget_sizetermrrrwrZ DEPTH_8_BITr_escape_code_caches)rYrrrrrA)rrTrZs     zVt100_Output.__init__csZ}s>||jkr>d}tj|||j|fdd}|||dS)z Create an Output class from a pseudo terminal. (This will take the dimensions by reading the pseudo terminal attributes.) z.Warning: Output is not to a terminal (fd=%r). cs0r t\}}t||dStdddSdS)N)rowscolumnsP)rrr)rr)isattyrrArTrs z'Vt100_Output.from_pty..get_size)r)rr_fds_not_a_terminalsysstderrradd)clsrrfdmsgrrA)rrrTfrom_ptys  zVt100_Output.from_ptycCs |jS)z Return file descriptor. )rr)rYrArArTrszVt100_Output.filenocCs|jjS)z" Return encoding used for stdout. )rr)rYrArArTrszVt100_Output.encodingcCs|j|dS)z+ Write raw data to output. N)rrp)rYdatarArArT write_rawszVt100_Output.write_rawcCs|j|dddS)zn Write text to output. (Removes vt100 escape codes. -- used for safely writing text.) ?N)rrpreplace)rYrrArArTrszVt100_Output.writecCs,|jdkr(|d|dddddS)z% Set terminal title. )Zlinuxz eterm-colorz]2;%srN)rrr)rYtitlerArArT set_titles zVt100_Output.set_titlecCs|ddS)Nr)r)rYrArArT clear_titleszVt100_Output.clear_titlecCs|ddS)zd Erases the screen with the background colour and moves the cursor to home. zN)r)rYrArArT erase_screenszVt100_Output.erase_screencCs|ddS)Nz [?1049h)r)rYrArArTenter_alternate_screensz#Vt100_Output.enter_alternate_screencCs|ddS)Nz[?1049l)r)rYrArArTquit_alternate_screensz"Vt100_Output.quit_alternate_screencCs"|d|d|ddS)Nz[?1000hz[?1015hz[?1006h)r)rYrArArTenable_mouse_supports  z!Vt100_Output.enable_mouse_supportcCs"|d|d|ddS)Nz[?1000lz[?1015lz[?1006l)r)rYrArArTdisable_mouse_supports  z"Vt100_Output.disable_mouse_supportcCs|ddS)zY Erases from the current cursor position to the end of the current line. zN)r)rYrArArTerase_end_of_lineszVt100_Output.erase_end_of_linecCs|ddS)zc Erases the screen from the current line down to the bottom of the screen. zN)r)rYrArArT erase_downszVt100_Output.erase_downcCs|ddS)Nz)r)rYrArArTreset_attributesszVt100_Output.reset_attributescCs|j|}|||dS)zW Create new style and output. :param attrs: `Attrs` instance. N)rr)rYrrxZescape_code_cacherArArTset_attributess zVt100_Output.set_attributescCs|ddS)Nz[?7l)r)rYrArArTdisable_autowrap szVt100_Output.disable_autowrapcCs|ddS)Nz[?7h)r)rYrArArTenable_autowrapszVt100_Output.enable_autowrapcCs|ddS)Nz[?2004h)r)rYrArArTenable_bracketed_pastesz#Vt100_Output.enable_bracketed_pastecCs|ddS)Nz[?2004l)r)rYrArArTdisable_bracketed_pastesz$Vt100_Output.disable_bracketed_pastercCs|d||fdS)z Move cursor position. z[%i;%iHN)r)rYrowcolumnrArArT cursor_gotoszVt100_Output.cursor_gotocCs0|dkr n"|dkr|dn|d|dS)Nrrzz[%iA)r)rYamountrArArT cursor_ups  zVt100_Output.cursor_upcCs0|dkr n"|dkr|dn|d|dS)Nrrzz[%iB)r)rYrrArArT cursor_down%s  zVt100_Output.cursor_downcCs0|dkr n"|dkr|dn|d|dS)Nrrzz[%iC)r)rYrrArArTcursor_forward/s  zVt100_Output.cursor_forwardcCs0|dkr n"|dkr|dn|d|dS)Nrrz[%iD)r)rYrrArArTcursor_backward7s  zVt100_Output.cursor_backwardcCs|ddS)Nz[?25l)r)rYrArArT hide_cursor?szVt100_Output.hide_cursorcCs|ddS)Nz [?12l[?25h)r)rYrArArT show_cursorBszVt100_Output.show_cursorc Cs|js dSd|j}yX|jrVt|jdr4|jj}n|j}|||jjpLddn |j||j WnRt k r}z4|j r|j dt j krn|j r|j ddkrnWdd}~XYnXg|_dS)z3 Write to output stream and flush. Nrbufferzutf-8rr)rrrrrrrencoderrIOErrorargserrnoZEINTR)rYrouterArArTrEs$    zVt100_Output.flushcCs|d|dS)z: Asks for a cursor position report (CPR). zN)rr)rYrArArT ask_for_cprqs zVt100_Output.ask_for_cprcCs|d|dS)z Sound bell. rN)rr)rYrArArTbellxs zVt100_Output.bell)NT)N)rr)&rbrcrdresetrrZ classmethodrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrArArArTr xsD      ,)rA)!reZ __future__rZprompt_toolkit.layout.screenrZprompt_toolkit.outputrZprompt_toolkit.styles.baserrxrZ six.movesrrrrr__all__r`r_rGrrErUdictrVrfrrrrwrr rArArArTs       ""A  h