LvddlmZddlmZmZmZmZmZmZm Z m Z ddl m Z ddl mZddlmZddlmZmZmZmZmZddlZddlZdZd Zd Zd ZGd d eZGddeZGddZ GddZ!dZ"e"e Z#e"e!Z$e%e#e%ek(sJe%e$e%ek(sJGdde&Z'y))unicode_literals)windllbyref ArgumentErrorc_charc_longc_ulongc_uintpointer)DWORD)Output)ANSI_COLOR_NAMES)CONSOLE_SCREEN_BUFFER_INFOSTD_OUTPUT_HANDLESTD_INPUT_HANDLECOORD SMALL_RECTN) Win32OutputcRt|jdz|jdzzS)aH Turns a COORD object into a c_long. This will cause it to be passed by value instead of by reference. (That is what I think at least.) When runing ``ptipython`` is run (only with IPython), we often got the following error:: Error in 'SetConsoleCursorPosition'. ArgumentError("argument 2: : wrong type",) argument 2: : wrong type It was solved by turning ``COORD`` parameters into a ``c_long`` like this. More info: http://msdn.microsoft.com/en-us/library/windows/desktop/ms686025(v=vs.85).aspx ii)rYX)coords 7lib/third_party/prompt_toolkit/terminal/win32_output.py _coord_byvalrs& %''G#egg&66 77Fz!prompt-toolkit-windows-output.logc"eZdZdZfdZxZS)NoConsoleScreenBufferErrorz Raised when the application is not running inside a Windows Console, but the user tries to instantiate Win32Output. cdtjjddv}|rdtjdz}nd}tt||y)NxtermTERMzFound %s, while expecting a Windows console. Maybe try to run this program using "winpty" or run it in cmd.exe instead. Or otherwise, in case of Cygwin, use the Python executable that is compiled for Cygwin.z2No Windows console found. Are you running cmd.exe?)osenvirongetsuperr__init__)selfrmessage __class__s rr&z#NoConsoleScreenBufferError.__init__0sR2::>>&"55 69; 68J KG KG ($8Ar)__name__ __module__ __qualname____doc__r& __classcell__)r)s@rrr+s B BrrceZdZdZd$dZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZd%dZdZdZdZdZdZdZdZdZdZdZ dZ!d Z"d!Z#e$d"Z%y#)&rzW I/O abstraction for rendering to Windows consoles. (cmd.exe and similar.) c,||_g|_||_tjj t |_d|_t|_ |j}|r |jnd|_ trtt d|_yy)NFab)use_complete_width_bufferstdoutrkernel32 GetStdHandlerhconsole_in_alternate_screenColorLookupTablecolor_lookup_tableget_win32_screen_buffer_info wAttributes default_attrs_DEBUG_RENDER_OUTPUTopen_DEBUG_RENDER_OUTPUT_FILENAMELOG)r'r5r3infos rr&zWin32Output.__init__Ds{"4  445FG $)!"2"400215T--2 94@DH rc6|jjS)z Return file descriptor. )r5filenor's rrEzWin32Output.filenoVs{{!!##rc.|jjS)z" Return encoding used for stdout. )r5encodingrFs rrHzWin32Output.encodingZs{{###rc:|jj|yN)r4appendr'datas rwritezWin32Output.write^s D!rc&|j|y)z@ For win32, there is no difference between write and write_raw. N)rNrLs r write_rawzWin32Output.write_rawas 4rcddlm}|j}|jr|jj }n-|j j|j jz }|j j|j jz dz}|jj dz }t||}|||S)Nr)Size)rowscolumns) prompt_toolkit.layout.screenrRr<r3dwSizersrWindowRightLeftBottomTopmin)r'rRrCwidthheightmaxwidths rget_sizezWin32Output.get_sizees5002  " "KKMMEMM''$--*<*<b> ! 1F0K  d#T]]Aq QYYZabcc$ ds%6 DD"D'' F0AE>>Fc|jt}tjj |j t |}|r|St)z, Return Screen buffer info. )rhrrr6GetConsoleScreenBufferInfor8rr)r'sbinfosuccesss rr<z(Win32Output.get_win32_screen_buffer_infosD, +-//<rFs rreset_attributeszWin32Output.reset_attributess) V__<r;lookup_fg_colorlookup_bg_colorrqrr6rr8) r'attrsfgcolorbgcolorbold underlineitalicblinkreverses rset_attributeszWin32Output.set_attributessDIA$ 65'""  DLE T,,<QRE V__<">@QRF LLEEv N LL44dmm D"DM(-D % %rc:d}tjjt}t }|j tjj |t||j tjj||j|zyN rr6r7rr rqGetConsoleModer SetConsoleModevaluer'ENABLE_MOUSE_INPUTr original_modes renable_mouse_supportz Win32Output.enable_mouse_supportosi!--.>?  V__33VW]=ST V__33V]=P=PSe=efrc<d}tjjt}t }|j tjj |t||j tjj||j|zyrrrs rdisable_mouse_supportz!Win32Output.disable_mouse_supportwsl!--.>?  V__33VW]=ST V__33V]=P=PUgSg=ghrcyrJrrFs r hide_cursorzWin32Output.hide_cursor rcyrJrrFs r show_cursorzWin32Output.show_cursorrrctjj}d}tjj |ddt |y)a$ Call win32 API to refresh the whole Window. This is sometimes necessary when the application paints background for completion menus. When the menu disappears, it leaves traces due to a bug in the Windows Console. Sending a repaint request solves it. rSN)rr6GetConsoleWindowuser32 RedrawWindowr )clsrRDW_INVALIDATEs rwin32_refresh_windowz Win32Output.win32_refresh_windows8113 ""64vn7MNrN)F)rr)&r*r+r,r-r&rErHrNrPrarqr<r|r~rrrrrrrrrrrrrrhrrrrrrrr classmethodrrrrrr?sA$$$"0*d$ -D>;### +) T,  aa  a%8B_8- .gi   O Orrc0eZdZdZdZdZdZdZdZdZ dZ d Z y ) FOREGROUND_COLORrrSrN r*r+r,BLACKBLUEGREENCYANREDMAGENTAYELLOWGRAY INTENSITYrrrrr/EDEDCGFDIrrc0eZdZdZdZdZdZdZdZdZ dZ d Z y ) BACKROUND_COLORrr 0@P`pNrrrrrrrrrcid|jd|jd|j|jzd|jd|j|jzd|jd|jd|j d |j d |jd |jd |j|jzd |j|jzd|j |jzd|j |jzd|j|jzd|j|jzS)zJ Create a table that maps the 16 named ansi colors to their Windows code. ansidefault ansiblack ansidarkgray ansilightgray ansiwhite ansidarkred ansidarkgreen ansibrown ansidarkblue ansipurpleansitealansired ansigreen ansiyellowansiblue ansifuchsia ansiturquoise) rrrrrrrrr) color_clss r_create_ansi_color_dictrsx    9+>+>>     )*=*==   9==  9??  9++  9>>  9,,  9>>  )--)*=*==! " )//I,?,??# $ )**Y-@-@@% & )..9+>+>>' ( )++i.A.AA) * )..9+>+>>+ rc>eZdZdZdZedZdZdZdZ dZ y) r:z8 Inspired by pygments/formatters/terminal256.py c<|j|_i|_yrJ)_build_color_table _win32_colors best_matchrFs rr&zColorLookupTable.__init__s!446rct}t}ddd|j|jfddd|j|jfddd|j|jfddd|j |j fddd|j |j fddd|j|jfddd|j|jfddd|j|jfddd|j|jz|j|jzfddd|j|jz|j|jzfddd|j |jz|j |jzfddd|j |jz|j |jzfddd|j|jz|j|jzfddd|j|jz|j|jzfddd|j|jz|j|jzfddd|j|jz|j|jzfgS)z< Build an RGB-to-256 color conversion table rD) rrrrrrrrrrr)FGBGs rrz#ColorLookupTable._build_color_tables   4rxx 2 4rww 0 4rxx 2 4rww 0 4rvvrvv . 4rzz2:: 6 4ryy")) 4 4rww 0 4rww5rww7M N 4rxx",,62<<8O P 4rww5rww7M N 4rvv 4bffr||6K L 4rzzBLL8"**r||:S T 4ryy2<<7R\\9Q R 4rxx",,62<<8O P 4rww5rww7M N%  rcd}d}d}|jD]4\}}} } } ||z } ||z } || z }| | z| | zz||zz}||ks/| }| }|}6||fS)Nir)r)r'rgrdistancefg_matchbg_matchr_g_b_fg_bg_rdgdbdds r_closest_colorzColorLookupTable._closest_colors $($6$6 BBSRBRBRBR"r'!BG+A8|%7!!rc|jj|d}|M tt|d}|dz dz}|dz dz}|dz}|j |||}||j|<|S#t$rd}YFwxYw)Nrrr%r)r r$intstr ValueErrorr7)r'colorindexesrgbr)r*rs r_color_indexeszColorLookupTable._color_indexess//%%eT2 ? #e*b)d"AT!Ad A))!Q2G%,DOOE "  sA.. A<;A<cL|tvr t|S|j|dS)z Return the color for use in the `windll.kernel32.SetConsoleTextAttribute` API call. :param fg_color: Foreground as text. E.g. 'ffffff' or 'red' r)FG_ANSI_COLORSr?)r'fg_colors rrz ColorLookupTable.lookup_fg_color- ~ %!(+ +&&x03 3rcL|tvr t|S|j|dS)z Return the color for use in the `windll.kernel32.SetConsoleTextAttribute` API call. :param bg_color: Background as text. E.g. 'ffffff' or 'red' rS)BG_ANSI_COLORSr?)r'bg_colors rrz ColorLookupTable.lookup_bg_color!rCrN) r*r+r,r-r& staticmethodrr7r?rrrrrr:r:s4  8"$  4 4rr:)( __future__rctypesrrrrrr r r ctypes.wintypesr prompt_toolkit.rendererr prompt_toolkit.stylesrprompt_toolkit.win32_typesrrrrrr"rx__all__rr?rA ExceptionrrrrrrArEsetobjectr:rrrrRs'YYY!*2yy   8* DBB(SO&SOl    4))9:(9 >c"23333 >c"23333^4v^4r