Re( @ s d d l Z d d l Z d d l Z d d l m Z m Z m Z m Z d d l m Z m
Z
m Z d d l m
Z
m Z d Z e
d k r e Z Gd d d e Z Gd d d e Z d S)
N )AnsiForeAnsiBack AnsiStyleStyle)WinTermWinColorWinStyle)windll winapi_testc @ sp e Z d Z d Z d d Z d d Z d d Z d d Z d
d Z d d
Z e
d d Z d S)
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)selfwrapped converter r /builddir/build/BUILDROOT/alt-python35-pip-20.2.4-5.el8.x86_64/opt/alt/python35/lib/python3.5/site-packages/pip/_vendor/colorama/ansitowin32.py__init__ s zStreamWrapper.__init__c C s t | j | S)N)getattrr
)r namer r r __getattr__ s zStreamWrapper.__getattr__c O s | j j | | S)N)r
__enter__)r argskwargsr r r r s zStreamWrapper.__enter__c O s | j j | | S)N)r
__exit__)r r r r r r r % s zStreamWrapper.__exit__c C s | j j | d S)N)r write)r textr r r r ( s zStreamWrapper.writec C sw | j } d t j k rF | d k rF | t j k sB | t j k rF d Sy
| j } Wn t k
rk d SYn X| Sd S)NPYCHARM_HOSTEDTF)r
osenvironsys
__stdout__
__stderr__isattyAttributeError)r stream
stream_isattyr r r r$ + s *
zStreamWrapper.isattyc C s1 | j } y | j SWn t k
r, d SYn Xd S)NT)r
closedr% )r r& r r r r( 7 s
zStreamWrapper.closedN) __name__
__module__ __qualname____doc__r r r r r r$ propertyr( r r r r r s r c @ s e Z d Z d Z e j d Z e j d Z d d d d d Z d d Z d
d Z
d d
Z d d Z d d Z
d d Z d d Z d d Z d d Z d d Z d S) 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 | | | _ t j d k } | o? t } | d k rq | pn | j j on | j j } | | _ | d k r | o | j j o | j j } | | _
| j | _ | j t
j k | _ d S)Nnt)r autoresetr r& r r r r( r$ stripconvertget_win32_calls win32_callsr! stderr on_stderr)r r r2 r1 r0
on_windowsconversion_supportedr r r r I s # " zAnsiToWin32.__init__c C s | j p | j p | j S)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()
)r2 r1 r0 )r r r r should_wrapj s zAnsiToWin32.should_wrapc N C sV | j rOt rOt j t j f t j t j t j f t j t j t j f t j t j t j f t
j t j t
j f t
j t j t
j f t
j t j t
j f t
j t j t
j f t
j t j t
j f t
j t j t
j f t
j t j t
j f t
j t j t
j f t
j t j f t
j t j t
j d f t
j t j t
j d f t
j t j t
j d f t
j t j t
j d f t
j t j t
j d f t
j t j t
j d f t
j t j t
j d f t
j t j t
j d f t j t j t
j f t j t j t
j f t j t j t
j f t j t j t
j f t j t j t
j f t j t j t
j f t j t j t
j f t j t j t
j f t j t j f t j t j t
j d f t j t j t
j d f t j t j t
j d f t j t j t
j d f t j t j t
j d f t j t j t
j d f t j t j t
j d f t j t j t
j d f i& St! S)NT)"r2 wintermr RESET_ALL reset_allBRIGHTstyler DIMNORMALr BLACKforer REDGREENYELLOWBLUEMAGENTACYANWHITEGREYRESET
LIGHTBLACK_EX LIGHTRED_EX
LIGHTGREEN_EXLIGHTYELLOW_EX LIGHTBLUE_EXLIGHTMAGENTA_EX LIGHTCYAN_EX
LIGHTWHITE_EXr backdict)r r r r r3 t sP zAnsiToWin32.get_win32_callsc C sV | j s | j r" | j | n | j j | | j j | j rR | j d S)N)r1 r2 write_and_convertr r flushr0 r< )r r r r r r s
zAnsiToWin32.writec C sJ | j r | j d d n* | j rF | j j rF | j j t j d S)Nmr )r ) r2
call_win32r1 r& r( r r r r; )r r r r r< s zAnsiToWin32.reset_allc C s d } | j | } xX | j j | D]D } | j \ } } | j | | | | j | j | } q( W| j | | t | d S)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 cursormatchstartendr r r rV s
z AnsiToWin32.write_and_convertc C s7 | | k r3 | j j | | | | j j d S)N)r r rW )r r rd re r r r r^ s z AnsiToWin32.write_plain_textc C s/ | j r+ | j | | } | j | | d S)N)r2 extract_paramsrY )r paramstringcommandparamsr r r r_ s zAnsiToWin32.convert_ansic C s | d k rT t d d | j d D } x~ t | d k rP | d } q1 Wn[ t d d | j d D } t | d k r | d k r d } n | d
k r d
} | S)NHfc s s3 | ]) } t | d k r' t | n d Vq d S)r r N)ra int).0pr r r s z-AnsiToWin32.extract_params..; r c s s- | ]# } t | d k r t | Vq d S)r N)ra rk )rl rm r r r rn s r JKmABCD)r )r )r )tuplesplitra )r rh rg ri r r r rf s "" zAnsiToWin32.extract_paramsc C sa | d k ru xN| D][ } | | j k r | j | } | d } | d d } t d | j } | | | q Wn | d k r t j | d d | j n | d k r t j | d d | j n | d k r t j | d | j nq | d k r]| d } d d | f d
d | f d | d f d | d f i | \ } }
t j | |
d | j d S)
NrX r r r6 JKrj rr ABCD)r4 rU r6 r: erase_screen
erase_lineset_cursor_position
cursor_adjust) r rh ri param func_argsfuncr r nxyr r r rY s$
BzAnsiToWin32.call_win32c C s x | j j | D] } | j \ } } | d | | | d } | j \ } } | d k r | j d } | d d k r t j | d q W| S)Nro r 02r ) ANSI_OSC_REr\ r] r` rt r: set_title)r r rc rd re rg rh ri r r r rZ s zAnsiToWin32.convert_osc)r) r* r+ r, recompiler[ r r r9 r3 r r< rV r^ r_ rf rY rZ r r r r r. @ s !
,
r. )r r! r ansir r r r r: r r r win32r
r objectr r. r r r r s " 0