U
    c                     @   sv   zd dl mZ W n ek
r,   dd ZY nX ddlmZ G dd deZG dd	 d	eZG d
d deZdd Z	dS )    )get_osfhandlec                 C   s   t dd S )NzThis isn't windows!)OSError)_ r   4/tmp/pip-unpacked-wheel-sbtr0pc4/colorama/winterm.pyr      s    r      )win32c                   @   s,   e Zd ZdZdZdZdZdZdZdZ	dZ
d	S )
WinColorr   r                     N)__name__
__module____qualname__BLACKBLUEGREENCYANREDMAGENTAYELLOWGREYr   r   r   r   r	      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_all2   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)r6   )r(   r'   r-   r   r   r5   )r/   forelightr6   r   r   r   r9   7   s    zWinTerm.forec                 C   sL   |d kr| j }|| _|r*|  jtjO  _n|  jtj M  _| j|d d S r8   )r*   r)   r-   r   r    r5   )r/   backr:   r6   r   r   r   r;   B   s    zWinTerm.backc                 C   s$   |d kr| j }|| _| j|d d S r8   )r,   r+   r5   )r/   styler6   r   r   r   r<   M   s    zWinTerm.stylec                 C   s0   |d kr|   }tj}|r tj}t|| d S N)r2   r   r#   STDERRSetConsoleTextAttribute)r/   r4   r6   handler   r   r   r5   S   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_position[   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/   rD   r6   r@   r   r   r   set_cursor_positionc   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>   rE   rC   rB   rF   )r/   xyr6   r@   rD   adjusted_positionr   r   r   cursor_adjustm   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"   dwSizerB   rC   rA   COORDFillConsoleOutputCharacterFillConsoleOutputAttributer2   rF   )	r/   moder6   r@   csbicells_in_screencells_before_cursor
from_coordcells_to_eraser   r   r   erase_screenu   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
   rM   )r   r#   r>   r"   rA   rN   rB   rO   rC   rP   rQ   r2   )r/   rR   r6   r@   rS   rV   rW   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   r0   r2   r&   r7   r9   r;   r<   r5   rE   rG   rL   rX   rY   r\   r   r   r   r   r!      s   








 
r!   c              	   C   sx   t jd kst  sdS zBt| }t |}t ||t jB  t |}|t j@ rVW dS W n ttfk
rr   Y dS X d S )NFT)	r   windllwinapi_testr   ZGetConsoleModeZSetConsoleModeZ"ENABLE_VIRTUAL_TERMINAL_PROCESSINGr   	TypeError)fdr@   rR   r   r   r   enable_vt_processing   s    



ra   N)
msvcrtr   ImportError r   objectr	   r   r!   ra   r   r   r   r   <module>   s    