3
Pf% @ s d dl Z d dlZd dlZddlmZmZmZmZ ddlm Z m
Z
m Z ddl m
Z
mZ dZe
dk rfe Zdd Zdd ZG d
d d eZG d d
d
eZdS ) N )AnsiForeAnsiBack AnsiStyleStyle)WinTermWinColorWinStyle)windll winapi_testc C s t | d p| jS )Nclosed)hasattrr )stream r !/usr/lib/python3.6/ansitowin32.pyis_stream_closed s r c C s t | do| j S )Nisatty)r
r )r r r r is_a_tty s r c @ s( e Zd ZdZdd Zdd Zdd ZdS )
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)selfwrappedZ converterr r r __init__ s zStreamWrapper.__init__c C s t | j|S )N)getattrr )r namer r r __getattr__$ s zStreamWrapper.__getattr__c C s | j j| d S )N)r write)r textr r r r ' s zStreamWrapper.writeN)__name__
__module__ __qualname____doc__r r r r r r r r s r c @ s~ e Zd ZdZejdZejdZd ddZdd Z d
d Z
d d
Z dd Z dd Z
dd Zdd Zdd Zdd Zdd 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 }|d krL|pJt| oJt| }|| _ |d krp|ont| ont|}|| _
| j | _ | j t
jk| _d S )Nnt)r autoresetr r osr r r r stripconvertget_win32_calls win32_callssysstderr on_stderr)r r r( r' r% Z
on_windowsZconversion_supportedr r r r 4 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()
)r( r' r% )r r r r should_wrapU s zAnsiToWin32.should_wrapc N C s> | j otr8tjtjftjtjtjftjtjtj 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 )NT)"r( wintermr RESET_ALL reset_allZBRIGHTZstyler ZDIMZNORMALr ZBLACKZforer ZREDZGREENZYELLOWZBLUEZMAGENTAZCYANZWHITEZGREYZRESETZ
LIGHTBLACK_EXZ LIGHTRED_EXZ
LIGHTGREEN_EXZLIGHTYELLOW_EXZ LIGHTBLUE_EXZLIGHTMAGENTA_EXZ LIGHTCYAN_EXZ
LIGHTWHITE_EXr Zbackdict)r r r r r) _ sP
zAnsiToWin32.get_win32_callsc C s@ | j s | jr| j| n| jj| | jj | jr<| j d S )N)r' r( write_and_convertr r flushr% r1 )r r r r r r s
zAnsiToWin32.writec C s: | j r| jdd n"| j r6t| j r6| jjtj d S )Nmr )r )r(
call_win32r' r r r r r0 )r r r r r1 s zAnsiToWin32.reset_allc C sf d}| j |}x@| jj|D ]0}|j \}}| j||| | j|j |}q W | j||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_REfinditerspanwrite_plain_text convert_ansigroupslen)r r Zcursormatchstartendr r r r3 s
z AnsiToWin32.write_and_convertc C s* ||k r&| j j||| | j j d S )N)r r r4 )r r r@ rA r r r r; s z AnsiToWin32.write_plain_textc C s" | j r | j||}| j|| d S )N)r( extract_paramsr6 )r paramstringcommandparamsr r r r< s zAnsiToWin32.convert_ansic C s~ |dkrt dd |jdD }t|dkrz|d krnd }n |d
krzd
}|S )NHfc s s& | ] }t |d krt|ndV qdS )r r N)r> int).0pr r r s z-AnsiToWin32.extract_params..; r c s s" | ]}t |d krt|V qdS )r N)r> rG )rH rI r r r rJ s r ZJKmABCD)r )r )r )tuplesplitr> )r rD rC rE r r r rB s zAnsiToWin32.extract_paramsc C s |dkrVx|D ]B}|| j kr| j | }|d }|dd }t| jd}||| qW n|dkrttj|d | jd n|dkrtj|d | jd nf|dkrtj|| jd nL|dkr|d }d| fd|f|df| dfd | \} }
tj| |
| jd d S )
Nr5 r r )r- JKrF rM )ABCD)r* r2 r- r/ Z erase_screenZ
erase_lineZset_cursor_positionZ
cursor_adjust) r rD rE ZparamZ func_argsfuncargskwargsnxyr r r r6 s$
*zAnsiToWin32.call_win32c C sv xp| j j|D ]`}|j \}}|d | ||d }|j \}}|dkr|jd}|d dkrtj|d qW |S )NrK r Z02r ) ANSI_OSC_REr9 r: r= rO r/ Z set_title)r r r? r@ rA rC rD rE r r r r7 s
zAnsiToWin32.convert_osc)NNF)r r r! r" recompiler8 r] r r. r) r r1 r3 r; r< rB r6 r7 r r r r r# + s
!
,
r# )r^ r+ r& ansir r r r r/ r r r Zwin32r
r r r objectr r# r r r r s