U
|
Þ=®dx+ ã @ s d dl Z d dlZd dlZddlmZmZmZmZmZ ddl m
|
Z
|
m Z m Z m Z ddlmZmZ dZ edk rne Z G dd deZG dd d eZdS )
|
é Né )ÚAnsiForeÚAnsiBackÚ AnsiStyleÚStyleÚBEL)Úenable_vt_processingÚWinTermÚWinColorÚWinStyle)ÚwindllÚ winapi_testc @ s\ e Zd ZdZdd Zdd Zdd Zdd Zd
|
d Zd d Z dd Z
|
dd Z e dd Z dS )Ú StreamWrapperz´
|
Wraps a stream (such as stdout), acting as a transparent proxy for all
|
attribute access apart from method 'write()', which is delegated to our
|
Converter instance.
|
c C s || _ || _d S ©N)Ú_StreamWrapper__wrappedÚ_StreamWrapper__convertor)ÚselfÚwrappedÚ converter© r úKd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\colorama/ansitowin32.pyÚ__init__ s zStreamWrapper.__init__c C s t | j|S r )Úgetattrr )r Únamer r r Ú __getattr__ s zStreamWrapper.__getattr__c O s | j j||S r )r Ú __enter__©r ÚargsÚkwargsr r r r s zStreamWrapper.__enter__c O s | j j||S r )r Ú__exit__r r r r r % s zStreamWrapper.__exit__c C s
|
|| _ d S r ©Ú__dict__)r Ústater r r Ú __setstate__( s zStreamWrapper.__setstate__c C s | j S r r ©r r r r Ú __getstate__+ s zStreamWrapper.__getstate__c C s | j |¡ d S r )r Úwrite©r Útextr r r r& . s zStreamWrapper.writec C s\ | j }dtjkr0|d k r0|tjks,|tjkr0dS z
|
|j}W n tk
|
rP Y dS X | S d S )NÚPYCHARM_HOSTEDTF)r ÚosÚenvironÚsysÚ
|
__stdout__Ú
|
__stderr__ÚisattyÚAttributeError)r ÚstreamÚ stream_isattyr r r r/ 1 s
|
|
zStreamWrapper.isattyc C s. | j }z|jW S ttfk
|
r( Y dS X d S ©NT)r Úclosedr0 Ú
|
ValueError)r r1 r r r r4 = s
|
zStreamWrapper.closedN)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r r r r r# r% r& r/ Úpropertyr4 r r r r r s r c @ s e Zd ZdZe d¡Ze d¡Zd ddZdd Z d
|
d Z
|
d d Z dd Z dd Z dd Zdd Zdd Zdd Zdd Zd d ZdS ) Ú AnsiToWin32z¾
|
Implements a 'write()' method which, on Windows, will strip ANSI character
|
sequences from the text, and if outputting to a tty, will convert them into
|
win32 function calls.
|
z ?\[((?:\d|;)*)([a-zA-Z])?z?\]([^]*)()?NFc C sÈ || _ || _t|| | _tjdk}|o*t }z | ¡ }W n tk
|
rP d}Y nX | p^t |}| jj
|
or| j ¡ } |o|| }
|
|d kr|
|
p| }|| _ |d kr¦|
|
o¤| }|| _ | ¡ | _| j tjk| _d S )NÚntéÿÿÿÿ)r Ú autoresetr r1 r* r r ÚfilenoÚ Exceptionr r4 r/ ÚstripÚconvertÚget_win32_callsÚ win32_callsr, ÚstderrÚ on_stderr) r r rB rA r> Ú
|
on_windowsÚconversion_supportedÚfdZsystem_has_native_ansiZhave_ttyZneed_conversionr r r r Q s(
|
|
|
|
|
|
zAnsiToWin32.__init__c C s | j p| jp| jS )aj
|
True if this class is actually needed. If false, then the output
|
stream will not be affected, nor will win32 calls be issued, so
|
wrapping stdout is not actually required. This will generally be
|
False on non-Windows platforms, unless optional functionality like
|
autoreset has been requested using kwargs to init()
|
)rB rA r> r$ r r r Ú should_wrapy s zAnsiToWin32.should_wrapc N C s@ | j r:tr:tjtjftjtjtjftjtjtj ftj tjtj ft
|
j tj t j ft
|
jtj t jft
|
jtj t jft
|
jtj t jft
|
jtj t jft
|
jtj t jft
|
jtj t jft
|
jtj t jft
|
jtj ft
|
jtj t j dft
|
jtj t jdft
|
jtj t jdft
|
jtj t jdft
|
jtj t jdft
|
j tj t jdft
|
j tj t jdft
|
j tj t jdft j tj t j ft jtj t jft jtj t jft jtj t jft jtj t jft jtj t jft jtj t jft jtj t jft jtj ft jtj t j dft jtj t jdft jtj t jdft jtj t jdft jtj t jdft j tj t jdft j tj t jdft j tj t jdfi&S t! S r3 )"rB Úwintermr Ú RESET_ALLÚ reset_allÚBRIGHTÚstyler ÚDIMÚNORMALr ÚBLACKÚforer
|
ÚREDÚGREENÚYELLOWÚBLUEÚMAGENTAÚCYANÚWHITEÚGREYÚRESETÚ LIGHTBLACK_EXÚ LIGHTRED_EXÚ LIGHTGREEN_EXÚLIGHTYELLOW_EXÚ LIGHTBLUE_EXÚLIGHTMAGENTA_EXÚ LIGHTCYAN_EXÚ LIGHTWHITE_EXr ÚbackÚdictr$ r r r rC s
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Ú(zAnsiToWin32.get_win32_callsc C s@ | j s | jr| |¡ n| j |¡ | j ¡ | jr<| ¡ d S r )rA rB Úwrite_and_convertr r& Úflushr> rM r' r r r r& ¯ s
|
zAnsiToWin32.writec C s4 | j r| dd¡ n | js0| jjs0| j tj¡ d S )NÚm©r ) rB Ú
|
call_win32rA r1 r4 r r& r rL r$ r r r rM ¹ s zAnsiToWin32.reset_allc C sb d}| |¡}| j |¡D ]0}| ¡ \}}| |||¡ | j| ¡ |}q| ||t|¡ dS )zª
|
Write the given text to our wrapped stream, stripping any ANSI
|
sequences from the text, and optionally converting them into win32
|
calls.
|
r N)Ú convert_oscÚ ANSI_CSI_REÚfinditerÚspanÚwrite_plain_textÚ convert_ansiÚgroupsÚlen)r r( ÚcursorÚmatchÚstartÚendr r r rg À s
|
z AnsiToWin32.write_and_convertc C s* ||k r&| j |||
¡ | j ¡ d S r )r r& rh )r r( rv rw r r r rp Ð s z AnsiToWin32.write_plain_textc C s" | j r | ||¡}| ||¡ d S r )rB Úextract_paramsrk )r Ú paramstringÚcommandÚparamsr r r rq Ö s zAnsiToWin32.convert_ansic C sz |dkr8t dd | d¡D }t|dk rv|d }q n>t dd | d¡D }t|dkrv|d krjd
|
}n |d krvd}|S ) NÚHfc s s& | ] }t |d krt|ndV qdS )r r N©rs Úint©Ú.0Úpr r r Ú <genexpr>Þ s z-AnsiToWin32.extract_params.<locals>.<genexpr>ú;é )r c s s" | ]}t |d krt|V qdS )r Nr} r r r r r ã s r ÚJKmrj ÚABCD)ÚtupleÚsplitrs )r rz ry r{ r r r rx Ü s zAnsiToWin32.extract_paramsc C sø |dkrR|D ]B}|| j kr | j | }|d }|dd
}t| jd}||| q n¢|dkrptj|d | jd n|dkrtj|d | jd nf|dkr¨tj|| jd nL|dkrô|d }d| fd|f|df| dfd | \} }
|
tj| |
|
| jd d S )
|
Nri r r )rF ÚJÚKr| r )ÚAÚBÚCÚD)rD rf rF rK Ú erase_screenÚ
|
erase_lineÚset_cursor_positionÚ cursor_adjust) r rz r{ ÚparamÚ func_argsÚfuncr r ÚnÚxÚyr r r rk î s$
|
|
*zAnsiToWin32.call_win32c C s | j |¡D ]n}| ¡ \}}|d |
||d
}| ¡ \}}|tkr | d¡dkr | d¡}|d dkr t |d ¡ q |S )Nr r r Ú02) Ú ANSI_OSC_RErn ro rr r Úcountr rK Ú set_title)r r( ru rv rw ry rz r{ r r r rl s
|
zAnsiToWin32.convert_oscc C s | j ¡ d S r )r rh r$ r r r rh s zAnsiToWin32.flush)NNF)r6 r7 r8 r9 ÚreÚcompilerm r r rJ rC r& rM rg rp rq rx rk rl rh r r r r r; H s
|
|
|
(
|
,
|
r; )r r, r* Úansir r r r r rK r r r
|
r Úwin32r r Úobjectr r; r r r r Ú<module> s 8
|
