U
    }Ha&                     @   s@   d dl mZ G dd deZG dd deZG dd deZdS )	   )win32c                   @   s,   e Zd ZdZdZdZdZdZdZdZ	dZ
d	S )
WinColor    r                     N)__name__
__module____qualname__BLACKBLUEGREENCYANREDMAGENTAYELLOWGREY r   r   4/tmp/pip-unpacked-wheel-25st9mmh/colorama/winterm.pyr      s   r   c                   @   s   e Zd ZdZdZdZdS )WinStyler         N)r   r   r   NORMALBRIGHTBRIGHT_BACKGROUNDr   r   r   r   r      s   r   c                   @   s   e Zd Zdd Zdd Zdd Zd dd	Zd!ddZd"ddZd#ddZ	d$ddZ
dd Zd%ddZd&ddZd'ddZd(ddZdd ZdS ))WinTermc                 C   s>   t t jj| _| | j | j| _| j| _	| j
| _d| _d S )Nr   )r   GetConsoleScreenBufferInfoSTDOUTwAttributes_default	set_attrs_fore_default_fore_back_default_back_style_default_style_lightselfr   r   r   __init__   s    zWinTerm.__init__c                 C   s   | j | jd  | j| jB  S )N   )r$   r&   r(   r*   r+   r   r   r   	get_attrs$   s    zWinTerm.get_attrsc                 C   s.   |d@ | _ |d? d@ | _|tjtjB @ | _d S )Nr
   r   )r$   r&   r   r   r   r(   )r,   valuer   r   r   r#   '   s    
zWinTerm.set_attrsNc                 C   s$   |  | j | j| jd d| _d S )N)attrsr   )r#   r"   set_consoler*   )r,   	on_stderrr   r   r   	reset_all,   s    zWinTerm.reset_allFc                 C   sL   |d kr| j }|| _|r*|  jtjO  _n|  jtj M  _| j|d d S N)r3   )r%   r$   r*   r   r   r2   )r,   forelightr3   r   r   r   r6   1   s    zWinTerm.forec                 C   sL   |d kr| j }|| _|r*|  jtjO  _n|  jtj M  _| j|d d S r5   )r'   r&   r*   r   r   r2   )r,   backr7   r3   r   r   r   r8   <   s    zWinTerm.backc                 C   s$   |d kr| j }|| _| j|d d S r5   )r)   r(   r2   )r,   styler3   r   r   r   r9   G   s    zWinTerm.stylec                 C   s0   |d kr|   }tj}|r tj}t|| d S N)r/   r   r    STDERRSetConsoleTextAttribute)r,   r1   r3   handler   r   r   r2   M   s    zWinTerm.set_consolec                 C   s,   t |j}| jd7  _| jd7  _|S )Nr   )r   r   dwCursorPositionXY)r,   r=   positionr   r   r   get_positionU   s    zWinTerm.get_positionc                 C   s,   |d krd S t j}|rt j}t || d S r:   )r   r    r;   SetConsoleCursorPosition)r,   rA   r3   r=   r   r   r   set_cursor_position]   s    zWinTerm.set_cursor_positionc                 C   sB   t j}|rt j}| |}|j| |j| f}t j||dd d S )NF)adjust)r   r    r;   rB   r@   r?   rC   )r,   xyr3   r=   rA   adjusted_positionr   r   r   cursor_adjustg   s    
zWinTerm.cursor_adjustr   c           	      C   s   t j}|rt j}t |}|jj|jj }|jj|jj |jj }|dkrZ|j}|| }n8|dkrtt dd}|}n|dkrt dd}|}nd S t 	|d|| t 
||  || |dkrt |d d S )Nr   r   r    )r   r   )r   r    r;   r   dwSizer?   r@   r>   COORDFillConsoleOutputCharacterFillConsoleOutputAttributer/   rC   )	r,   moder3   r=   csbicells_in_screencells_before_cursor
from_coordcells_to_eraser   r   r   erase_screeno   s(    

zWinTerm.erase_screenc                 C   s   t j}|rt j}t |}|dkr:|j}|jj|jj }nH|dkr\t d|jj}|jj}n&|dkr~t d|jj}|jj}nd S t 	|d|| t 
||  || d S )Nr   r   r   rJ   )r   r    r;   r   r>   rK   r?   rL   r@   rM   rN   r/   )r,   rO   r3   r=   rP   rS   rT   r   r   r   
erase_line   s     


zWinTerm.erase_linec                 C   s   t | d S r:   )r   SetConsoleTitle)r,   titler   r   r   	set_title   s    zWinTerm.set_title)N)NFF)NFF)NF)NF)NF)F)r   F)r   F)r   r   r   r-   r/   r#   r4   r6   r8   r9   r2   rB   rD   rI   rU   rV   rY   r   r   r   r   r      s   








 
r   N) r   objectr   r   r   r   r   r   r   <module>   s   