Format: application/x-python-code
1 o
2
3 1'h� � @ st zd dl mZ W n
4 ey dd� ZY nw ddlmZ G dd� de�ZG dd � d e�ZG d
5 d � d e�Zd d
6 � Z dS )� )�
7 get_osfhandlec C s t d��)NzThis isn't windows!)�OSError)�_� r �FC:\Users\shubh\Videos\agent\nova\Lib\site-packages\colorama/winterm.pyr s r � )�win32c @ s, e Zd ZdZdZdZdZdZdZdZ dZ
8 d S )
9 �WinColorr r � � � � � � N) �__name__�
10 __module__� __qualname__ZBLACKZBLUEZGREENZCYANZREDZMAGENTAZYELLOWZGREYr r r r r s r c @ s e Zd ZdZdZdZdS )�WinStyler � � N)r r r ZNORMAL�BRIGHT�BRIGHT_BACKGROUNDr r r r r s r c @ s� e Zd Zdd� Zdd� Zdd� Zd dd �Zd!d d �Zd!d
11 d�Zd"dd�Z d"dd�Z
12 dd� Z d"dd�Z d#dd�Z
13 d$dd�Zd$d d �Zd d � ZdS )%�WinTermc C s> t �t j�j| _| �| j� | j| _| j| _ | j
14 | _ d| _ d S )Nr )
15 r �GetConsoleScreenBufferInfo�STDOUT� wAttributes�_default� set_attrs�_fore�
16 _default_fore�_back�
17 _default_back�_style�_default_style�_light��selfr r r �__init__ s
18 zWinTerm.__init__c C s | j | jd | j| jB S )N� )r r r" r$ r% r r r � get_attrs* s zWinTerm.get_attrsc C s. |d@ | _ |d? d@ | _|tjtjB @ | _d S )Nr r )r r r r r r" )r& �valuer r r r - s
19 zWinTerm.set_attrsNc C s$ | � | j� | j| jd� d| _d S )N)�attrsr )r r � set_consoler$ )r& � on_stderrr r r � reset_all2 s
20 zWinTerm.reset_allFc C �L |d u r| j }|| _|r| jtjO _n | jtj M _| j|d� d S �N)r- )r r r$ r r r, )r& �fore�lightr- r r r r1 7 � z WinTerm.forec C r/ r0 )r! r r$ r r r, )r& �backr2 r- r r r r4 B r3 z WinTerm.backc C s$ |d u r| j }|| _| j|d� d S r0 )r# r" r, )r& �styler- r r r r5 M s z
21 WinTerm.stylec C s0 |d u r| � � }tj}|rtj}t�||� d S �N)r) r r �STDERR�SetConsoleTextAttribute)r& r+ r- �handler r r r, S s zWinTerm.set_consolec C s, t �|�j}| jd7 _| jd7 _|S )Nr )r r �dwCursorPosition�X�Y)r& r9 �positionr r r � get_position[ s zWinTerm.get_positionc C s, |d u rd S t j}|rt j}t �||� d S r6 )r r r7 �SetConsoleCursorPosition)r& r= r- r9 r r r �set_cursor_positionc s zWinTerm.set_cursor_positionc C sB t j}|rt j}| �|�}|j| |j| f}t j||dd� d S )NF)�adjust)r r r7 r> r< r; r? )r& �x�yr- r9 r= Zadjusted_positionr r r �
22 cursor_adjustm s
23 zWinTerm.cursor_adjustr c C s� t j}|rt j}t �|�}|jj|jj }|jj|jj |jj }|dkr-|j}|| }n |dkr:t �dd�}|}n|dkrGt �dd�}|}nd S t � |d||� t �
24 || � � ||� |dkrgt � |d� d S d S )Nr r r
25 � )r r )
26 r r r7 r �dwSizer; r< r: �COORD�FillConsoleOutputCharacter�FillConsoleOutputAttributer) r? ) r& �moder- r9 �csbiZcells_in_screenZcells_before_cursor�
27 from_coord�cells_to_eraser r r � erase_screenu s*
28
29 �zWinTerm.erase_screenc C s� t j}|rt j}t �|�}|dkr |j}|jj|jj }n$|dkr.t �d|jj�}|jj}n|dkr?t �d|jj�}|jj}nd S t � |d||� t �
30 || � � ||� d S )Nr r r
31 rE ) r r r7 r r: rF r; rG r< rH rI r) )r& rJ r- r9 rK rL rM r r r �
32 erase_line� s
33
34
35 zWinTerm.erase_linec C s t �|� d S r6 )r �SetConsoleTitle)r& �titler r r � set_title� s zWinTerm.set_titler6 )NFF)NF)F)r F)r r r r' r) r r. r1 r4 r5 r, r> r@ rD rN rO rR r r r r r s
36
37
38
39
40
41
42
43
44
45 r c C st t jd u s t �� s dS z"t| �}t �|�}t �||t jB � t �|�}|t j@ r+W dS W d S ttfy9 Y dS w )NFT) r �windllZ winapi_testr �GetConsoleModeZSetConsoleMode�"ENABLE_VIRTUAL_TERMINAL_PROCESSINGr � TypeError)�fdr9 rJ r r r �enable_vt_processing� s
46 �
47
48 ��rX N)
49 �msvcrtr � ImportError� r �objectr r r rX r r r r �<module> s �