3
�Pf� � @ s@ d dl mZ G dd� de�ZG dd� de�ZG dd� de�ZdS ) � )�win32c @ s, e Zd ZdZdZdZdZdZdZdZ dZ
d S )
�WinColor� r � � � � � � N) �__name__�
__module__� __qualname__ZBLACKZBLUEZGREENZCYANZREDZMAGENTAZYELLOWZGREY� r r � /usr/lib/python3.6/winterm.pyr 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
d�Zd#dd�Z d$dd�Z
dd� Z d%dd�Z d&dd�Z
d'dd�Zd(d d �Zd d � ZdS ))�WinTermc C s> t jt j�j| _| j| j� | j| _| j| _ | j
| _ d| _ d S )Nr )
r �GetConsoleScreenBufferInfo�STDOUTZ wAttributes�_default� set_attrs�_fore�
_default_fore�_back�
_default_back�_style�_default_style�_light)�selfr r r �__init__ s 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
zWinTerm.set_attrsNc C s | j | j� | j| jd� d S )N)�attrs)r r � set_console)r! � on_stderrr r r � reset_all, s zWinTerm.reset_allFc C sL |d kr| j }|| _|r*| jtjO _n| jtj M _| j|d� d S )N)r( )r r r r r r' )r! �fore�lightr( r r r r* 0 s z WinTerm.forec C sL |d kr| j }|| _|r*| jtjO _n| jtj M _| j|d� d S )N)r( )r r r r r r' )r! �backr+ r( r r r r, ; s z WinTerm.backc C s$ |d kr| j }|| _| j|d� d S )N)r( )r r r' )r! �styler( r r r r- F s z
WinTerm.stylec C s0 |d kr| j � }tj}|r tj}tj||� d S )N)r$ r r �STDERRZSetConsoleTextAttribute)r! r& r( �handler r r r' L s zWinTerm.set_consolec C s, t j|�j}| jd7 _| jd7 _|S )Nr )r r �dwCursorPosition�X�Y)r! r/ �positionr r r � get_positionT s zWinTerm.get_positionc C s, |d kr d S t j}|r t j}t j||� d S )N)r r r. �SetConsoleCursorPosition)r! r3 r( r/ r r r �set_cursor_position\ s zWinTerm.set_cursor_positionc C sB t j}|rt j}| j|�}|j| |j| f}t j||dd� d S )NF)Zadjust)r r r. r4 r2 r1 r5 )r! �x�yr( r/ r3 Zadjusted_positionr r r �
cursor_adjustf s
zWinTerm.cursor_adjustr c C s� t j}|rt j}t j|�}|jj|jj }|jj|jj |jj }|dkrX|j}|| }|dkrrt jdd�}|}n|dkr�t jdd�}|}t j |d||� t j
|| j � ||� |dkr�t j |d� d S )Nr r r � )r r )
r r r. r �dwSizer1 r2 r0 �COORD�FillConsoleOutputCharacter�FillConsoleOutputAttributer$ r5 ) r! �moder( r/ �csbiZcells_in_screenZcells_before_cursor�
from_coord�cells_to_eraser r r � erase_screenn s&
zWinTerm.erase_screenc C s� t j}|rt j}t j|�}|dkr8|j}|jj|jj }|dkrZt jd|jj�}|jj}n |dkrzt jd|jj�}|jj}t j |d||� t j
|| j � ||� d S )Nr r r r: ) r r r. r r0 r; r1 r< r2 r= r> r$ )r! r? r( r/ r@ rA rB r r r �
erase_line� s
zWinTerm.erase_linec C s t j|� d S )N)r ZSetConsoleTitle)r! �titler r r � set_title� s zWinTerm.set_title)N)NFF)NFF)NF)NF)NF)F)r F)r F)r r r
r" r$ r r) r* r, r- r' r4 r6 r9 rC rD rF r r r r r s
r N)� r �objectr r r r r r r � s