o jx+@sddlZddlZddlZddlmZmZmZmZmZddl m Z m Z m Z m Z ddlmZmZdZ edur7e Z GdddeZGdd d eZdS) N)AnsiForeAnsiBack AnsiStyleStyleBEL)enable_vt_processingWinTermWinColorWinStyle)windll winapi_testc@s\eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ e ddZ 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. cCs||_||_dSN)_StreamWrapper__wrapped_StreamWrapper__convertor)selfwrapped converterr/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/venv/lib/python3.10/site-packages/colorama/ansitowin32.py__init__s zStreamWrapper.__init__cCs t|j|Sr)getattrr)rnamerrr __getattr__s zStreamWrapper.__getattr__cO|jj|i|Sr)r __enter__rargskwargsrrrrszStreamWrapper.__enter__cOrr)r__exit__rrrrr %szStreamWrapper.__exit__cCs ||_dSr__dict__)rstaterrr __setstate__(s zStreamWrapper.__setstate__cCs|jSrr!rrrr __getstate__+szStreamWrapper.__getstate__cCs|j|dSr)rwritertextrrrr'.szStreamWrapper.writecCsT|j}dtjvr|dur|tjus|tjurdSz|j}W|Sty)YdSw)NPYCHARM_HOSTEDTF)rosenvironsys __stdout__ __stderr__isattyAttributeError)rstream stream_isattyrrrr01s  zStreamWrapper.isattyc Cs(|j}z|jWSttfyYdSwNT)rclosedr1 ValueError)rr2rrrr5=s zStreamWrapper.closedN)__name__ __module__ __qualname____doc__rrrr r$r&r'r0propertyr5rrrrrs rc@seZdZdZedZedZdddZdd Z d d Z d d Z ddZ ddZ ddZddZddZddZddZddZdS) 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 Cs||_||_t|||_tjdk}|ot}z|}Wn ty'd}Ynw| p.t |}|jj o8|j } |o=| } |durG| pF| }||_ |durR| oQ| }||_ ||_|jtju|_dS)Nnt)r autoresetrr2r+rr fileno Exceptionrr5r0stripconvertget_win32_calls win32_callsr-stderr on_stderr) rrrCrBr? on_windowsconversion_supportedfdsystem_has_native_ansihave_ttyneed_conversionrrrrQs*        zAnsiToWin32.__init__cCs|jp|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() )rCrBr?r%rrr should_wrapyszAnsiToWin32.should_wrapc Cs|jrCtrCitjtjftjtjtjftjtjtj 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 jdfit jtj t jdft jtj t jdft jtj t jdft jtj t jdftj tj t j ftjtj t jftjtj t jftjtj t jftjtj t jftjtj t jftjtj t jftjtj t jftjtj ftjtj t j dftjtj t jdftjtj t jdftjtj t jdftjtj t jdftjtj t jdftjtj t jdftjtj t jdfiSt!Sr4)"rCwintermr RESET_ALL reset_allBRIGHTstyler DIMNORMALrBLACKforer REDGREENYELLOWBLUEMAGENTACYANWHITEGREYRESET LIGHTBLACK_EX LIGHTRED_EX LIGHTGREEN_EXLIGHTYELLOW_EX LIGHTBLUE_EXLIGHTMAGENTA_EX LIGHTCYAN_EX LIGHTWHITE_EXrbackdictr%rrrrDs        !"#(zAnsiToWin32.get_win32_callscCsD|js|jr ||n |j||j|jr |dSdSr)rBrCwrite_and_convertrr'flushr?rQr(rrrr's     zAnsiToWin32.writecCs>|jr |dddS|js|jjs|jtjdSdSdS)Nmr) rC call_win32rBr2r5rr'rrPr%rrrrQs zAnsiToWin32.reset_allcCsbd}||}|j|D]}|\}}|||||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. rN) convert_osc ANSI_CSI_REfinditerspanwrite_plain_text convert_ansigroupslen)rr)cursormatchstartendrrrrks  zAnsiToWin32.write_and_convertcCs.||kr|j||||jdSdSr)rr'rl)rr)rzr{rrrrtszAnsiToWin32.write_plain_textcCs&|jr|||}|||dSdSr)rCextract_paramsro)r paramstringcommandparamsrrrrus zAnsiToWin32.convert_ansicCs|dvr"tdd|dD}t|dkr |d}t|dks|Stdd|dD}t|dkrB|d vrs&z-AnsiToWin32.extract_params..;)rcss$|] }t|dkrt|VqdS)rNrrrrrrs"rJKmrnABCD)tuplesplitrw)rr~r}rrrrr|s    zAnsiToWin32.extract_paramsc Cs|dkr,|D]#}||jvr)|j|}|d}|dd}t|jd}||i|qdS|dvrs8