Format: application/x-python-code
1 o
2
3 1'hx+ � @ s� d dl Z d dlZd dlZddlmZmZmZmZmZ ddl m
4 Z
5 m Z m Z m
6 Z
7 ddlmZmZ dZ edur7e � Z G dd� de�ZG dd � d e�ZdS )
8 � N� )�AnsiFore�AnsiBack� AnsiStyle�Style�BEL)�enable_vt_processing�WinTerm�WinColor�WinStyle)�windll� winapi_testc @ s\ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
9 d � Zd d
10 � Z dd� Z
11 dd� Z e dd� �Z
12 dS )�
13 StreamWrapperz�
14 Wraps a stream (such as stdout), acting as a transparent proxy for all
15 attribute access apart from method 'write()', which is delegated to our
16 Converter instance.
17 c C s || _ || _d S �N)�_StreamWrapper__wrapped�_StreamWrapper__convertor)�self�wrapped� converter� r �JC:\Users\shubh\Videos\agent\nova\Lib\site-packages\colorama/ansitowin32.py�__init__ s
18 zStreamWrapper.__init__c C s t | j|�S r )�getattrr )r �namer r r � __getattr__ s zStreamWrapper.__getattr__c O � | j j|i |��S r )r � __enter__�r �args�kwargsr r r r s zStreamWrapper.__enter__c O r r )r �__exit__r r r r r % s zStreamWrapper.__exit__c C s
19 || _ d S r ��__dict__)r �stater r r � __setstate__( s
20 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 sT | j }dtjv r|d ur|tju s|tju rdS z|j}W |� S ty) Y dS w )NZPYCHARM_HOSTEDTF)r �os�environ�sys�
21 __stdout__�
22 __stderr__�isatty�AttributeError)r �streamZ
23 stream_isattyr r r r/ 1 s
24 ��zStreamWrapper.isattyc C s( | j }z|jW S ttfy Y dS w �NT)r �closedr0 �
25 ValueError)r r1 r r r r3 = s �zStreamWrapper.closedN)�__name__�
26 __module__� __qualname__�__doc__r r r r r$ r& r' r/ �propertyr3 r r r r r s r c @ s� e Zd ZdZe�d�Ze�d�Zd dd�Zdd � Z d
27 d � Z
28 d d
29 � Z dd� Z dd� Z
30 dd� Zdd� Zdd� Zdd� Zdd� Zd d � ZdS ) � AnsiToWin32z�
31 Implements a 'write()' method which, on Windows, will strip ANSI character
32 sequences from the text, and if outputting to a tty, will convert them into
33 win32 function calls.
34 z ?\[((?:\d|;)*)([a-zA-Z])?z?\]([^]*)()?NFc C s� || _ || _t|| �| _tjdk}|ot� }z|�� }W n ty' d}Y nw | p.t |�}| jj
35 o8| j� � } |o=| }
36 |d u rG|
37 pF| }|| _ |d u rR|
38 oQ| }|| _
39 | �� | _| j tju | _d S )N�nt�����)r � autoresetr r1 r* r r
40 �fileno� Exceptionr r3 r/ �strip�convert�get_win32_calls� win32_callsr, �stderr� on_stderr) r r rA r@ r= Z
41 on_windowsZconversion_supported�fdZsystem_has_native_ansiZhave_ttyZneed_conversionr r r r Q s*
42
43 �
44
45
46 zAnsiToWin32.__init__c C s | j p| jp| jS )aj
47 True if this class is actually needed. If false, then the output
48 stream will not be affected, nor will win32 calls be issued, so
49 wrapping stdout is not actually required. This will generally be
50 False on non-Windows platforms, unless optional functionality like
51 autoreset has been requested using kwargs to init()
52 )rA r@ r= r% r r r � should_wrapy s zAnsiToWin32.should_wrapc C s� | j �rCt�rCi tjtjf�tjtjtjf�tjtjtj f�tj tjtj f�t
53 j tj t
54 j f�t
55 jtj t
56 jf�t
57 jtj t
58 jf�t
59 jtj t
60 jf�t
61 jtj t
62 jf�t
63 jtj t
64 jf�t
65 jtj t
66 jf�t
67 jtj t
68 jf�t
69 jtj f�t
70 jtj t
71 j df�t
72 jtj t
73 jdf�t
74 jtj t
75 jdf�t
76 jtj t
77 jdf�i t
78 jtj t
79 jdf�t
80 j tj t
81 jdf�t
82 j tj t
83 jdf�t
84 j tj t
85 jdf�t j tj t
86 j f�t jtj t
87 jf�t jtj t
88 jf�t jtj t
89 jf�t jtj t
90 jf�t jtj t
91 jf�t jtj t
92 jf�t jtj t
93 jf�t jtj f�t jtj t
94 j df�t jtj t
95 jdf�t jtj t
96 jdf�t jtj t
97 jdf��t jtj t
98 jdft j tj t
99 jdft j tj t
100 jdft j tj t
101 jdfi�S t!� S r2 )"rA �wintermr � RESET_ALL� reset_allZBRIGHT�styler ZDIMZNORMALr ZBLACK�forer
102 ZREDZGREENZYELLOWZBLUEZMAGENTAZCYANZWHITEZGREY�RESETZ
103 LIGHTBLACK_EXZ LIGHTRED_EXZ
104 LIGHTGREEN_EXZLIGHTYELLOW_EXZ LIGHTBLUE_EXZLIGHTMAGENTA_EXZ LIGHTCYAN_EXZ
105 LIGHTWHITE_EXr �back�dictr% r r r rB � s�
106 �������� �
107 � � �
108
109 ��������������� � �
110 � � �!�"�#�(zAnsiToWin32.get_win32_callsc C sD | j s| jr | �|� n | j�|� | j�� | jr | �� d S d S r )r@ rA �write_and_convertr r' �flushr= rJ r( r r r r' � s
111 �zAnsiToWin32.writec C s> | j r | �dd� d S | js| jjs | j�tj� d S d S d S )N�m�r ) rA �
112 call_win32r@ r1 r3 r r' r rI r% r r r rJ � s
113 �zAnsiToWin32.reset_allc C sb d}| � |�}| j�|�D ]}|�� \}}| �|||� | j|�� � |}q
114 | �||t|�� dS )z�
115 Write the given text to our wrapped stream, stripping any ANSI
116 sequences from the text, and optionally converting them into win32
117 calls.
118 r N)� convert_osc� ANSI_CSI_RE�finditer�span�write_plain_text� convert_ansi�groups�len)r r) �cursor�match�start�endr r r rP � s
119 z AnsiToWin32.write_and_convertc C s. ||k r| j �|||� � | j �� d S d S r )r r' rQ )r r) r_ r` r r r rY � s �z AnsiToWin32.write_plain_textc C s& | j r| �||�}| �||� d S d S r )rA �extract_paramsrT )r � paramstring�command�paramsr r r rZ � s �zAnsiToWin32.convert_ansic C s� |dv r"t dd� |�d�D ��}t|�dk r |d }t|�dk s|S t dd� |�d�D ��}t|�dkrB|d v r<d
120 }|S |d v rBd}|S ) N�Hfc s s( � | ]}t |�d krt|�ndV qdS )r r N�r\ �int��.0�pr r r � <genexpr>� s �& z-AnsiToWin32.extract_params.<locals>.<genexpr>�;� )r c s s$ � | ]
121 }t |�d krt|�V qdS )r Nrf rh r r r rk � s �" r ZJKmrS �ABCD)�tuple�splitr\ )r rc rb rd r r r ra � s � � �zAnsiToWin32.extract_paramsc C s |dkr,|D ]#}|| j v r)| j | }|d }|dd � }t| jd�}||i |�� qd S |dv r<tj|d | jd� d S |dv rLtj|d | jd� d S |dv rZtj|| jd� d S |dv r�|d }d| fd|f|df| dfd �| \} }
122 tj| |
123 | jd� d S d S )
124 NrR r r )rE �J�Kre rn )�A�B�C�D)rC rO rE rH Z erase_screen�
125 erase_lineZset_cursor_positionZ
126 cursor_adjust) r rc rd �paramZ func_args�funcr r �n�x�yr r r rT � s*
127
128 ��*�zAnsiToWin32.call_win32c C s� | j �|�D ]7}|�� \}}|d |� ||d � }|�� \}}|tkr=|�d�dkr=|�d�}|d dv r=t�|d � q|S )Nrl r r Z02) � ANSI_OSC_RErW rX r[ r �countrp rH � set_title)r r) r^ r_ r` rb rc rd r r r rU s
129 �zAnsiToWin32.convert_oscc C s | j �� d S r )r rQ r% r r r rQ s zAnsiToWin32.flush)NNF)r5 r6 r7 r8 �re�compilerV r} r rG rB r' rJ rP rY rZ ra rT rU rQ r r r r r: H s
130
131
132 (
133 ,
134 r: )r� r, r* �ansir r r r r rH r r r
135 r �win32r r
136 �objectr r: r r r r �<module> s 8