Edit File: ansitowin32.cpython-36.pyc
3 �Pf�% � @ s� d dl Z d dlZd dlZddlmZmZmZmZ ddlm Z m Z m Z ddl m Z mZ dZe dk rfe � Zdd� Zdd � ZG d d � d e�ZG d d � d e�ZdS )� N� )�AnsiFore�AnsiBack� AnsiStyle�Style)�WinTerm�WinColor�WinStyle)�windll� winapi_testc C s t | d� p| jS )N�closed)�hasattrr )�stream� r �!/usr/lib/python3.6/ansitowin32.py�is_stream_closed s r c C s t | d�o| j� S )N�isatty)r r )r r r r �is_a_tty s r c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � StreamWrapperz� Wraps a stream (such as stdout), acting as a transparent proxy for all attribute access apart from method 'write()', which is delegated to our Converter instance. c C s || _ || _d S )N)�_StreamWrapper__wrapped�_StreamWrapper__convertor)�self�wrappedZ converterr r r �__init__ s zStreamWrapper.__init__c C s t | j|�S )N)�getattrr )r �namer r r � __getattr__$ s zStreamWrapper.__getattr__c C s | j j|� d S )N)r �write)r �textr r r r ' s zStreamWrapper.writeN)�__name__� __module__� __qualname__�__doc__r r r r r r r r s r c @ s~ e Zd ZdZejd�Zejd�Zd dd�Zdd � Z d d � Z d d � Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� ZdS ) � AnsiToWin32z� Implements a 'write()' method which, on Windows, will strip ANSI character sequences from the text, and if outputting to a tty, will convert them into win32 function calls. z ?\[((?:\d|;)*)([a-zA-Z])?z?\]((?:.|;)*?)()?NFc C s� || _ || _t|| �| _tjdk}|o*t� }|d krL|pJt|� oJt|� }|| _ |d krp|ont|� ont|�}|| _ | j � | _ | j t jk| _d S )N�nt)r � autoresetr r �osr r r r �strip�convert�get_win32_calls� win32_calls�sys�stderr� on_stderr)r r r( r' r% Z on_windowsZconversion_supportedr r r r 4 s zAnsiToWin32.__init__c C s | j p| jp| jS )aj True if this class is actually needed. If false, then the output stream will not be affected, nor will win32 calls be issued, so wrapping stdout is not actually required. This will generally be False on non-Windows platforms, unless optional functionality like autoreset has been requested using kwargs to init() )r( r' r% )r r r r � should_wrapU s zAnsiToWin32.should_wrapc N C s> | j ot�r8tjtjftjtjtjftjtjtj ftj tjtj ft j tj t j ft jtj t jft jtj t jft jtj t jft jtj t jft jtj t jft jtj t jft jtj t jft jtj ft jtj t j dft jtj t jdft jtj t jdft jtj t jdft jtj t jdft j tj t jdft j tj t jdft j tj t jdft j tj t j ft jtj t jft jtj t jft jtj t jft jtj t jft jtj t jft jtj t jft jtj t jft jtj ft jtj t j dft jtj t jdft jtj t jdft jtj t jdft jtj t jdft j tj t jdft j tj t jdft j tj t jdfi&S t!� S )NT)"r( �wintermr � RESET_ALL� reset_allZBRIGHTZstyler ZDIMZNORMALr ZBLACKZforer ZREDZGREENZYELLOWZBLUEZMAGENTAZCYANZWHITEZGREYZRESETZ LIGHTBLACK_EXZ LIGHTRED_EXZ LIGHTGREEN_EXZLIGHTYELLOW_EXZ LIGHTBLUE_EXZLIGHTMAGENTA_EXZ LIGHTCYAN_EXZ LIGHTWHITE_EXr Zback�dict)r r r r r) _ sP zAnsiToWin32.get_win32_callsc C s@ | j s | jr| j|� n| jj|� | jj� | jr<| j� d S )N)r' r( �write_and_convertr r �flushr% r1 )r r r r r r � s zAnsiToWin32.writec C s: | j r| jdd� n"| j r6t| j� r6| jjtj� d S )N�mr )r )r( � call_win32r' r r r r r0 )r r r r r1 � s zAnsiToWin32.reset_allc C sf d}| j |�}x@| jj|�D ]0}|j� \}}| j|||� | j|j� � |}q W | j||t|�� dS )z� Write the given text to our wrapped stream, stripping any ANSI sequences from the text, and optionally converting them into win32 calls. r N)� convert_osc� ANSI_CSI_RE�finditer�span�write_plain_text� convert_ansi�groups�len)r r Zcursor�match�start�endr r r r3 � s z AnsiToWin32.write_and_convertc C s* ||k r&| j j|||� � | j j� d S )N)r r r4 )r r r@ rA r r r r; � s z AnsiToWin32.write_plain_textc C s" | j r | j||�}| j||� d S )N)r( �extract_paramsr6 )r � paramstring�command�paramsr r r r< � s zAnsiToWin32.convert_ansic C s~ |dkr<t dd� |jd�D ��}xXt|�dk r8|d }q"W n>t dd� |jd�D ��}t|�dkrz|d krnd }n |d krzd }|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" | ]}t |�d krt|�V qdS )r N)r> rG )rH rI r r r rJ � s r ZJKm�ABCD)r )r )r )�tuple�splitr> )r rD rC rE r r r rB � s zAnsiToWin32.extract_paramsc C s� |dkrVx�|D ]B}|| j kr| j | }|d }|dd � }t| jd�}|||� qW n�|dkrttj|d | jd� n�|dkr�tj|d | jd� nf|dkr�tj|| jd� nL|dkr�|d }d| fd|f|df| dfd �| \} } tj| | | jd� d S ) Nr5 r r )r- �J�KrF rM )�A�B�C�D)r* r2 r- r/ Z erase_screenZ erase_lineZset_cursor_positionZ cursor_adjust) r rD rE ZparamZ func_args�func�args�kwargs�n�x�yr r r r6 � s$ *zAnsiToWin32.call_win32c C sv xp| j j|�D ]`}|j� \}}|d |� ||d � }|j� \}}|dkr|jd�}|d dkrtj|d � qW |S )N�rK r Z02r )� ANSI_OSC_REr9 r: r= rO r/ Z set_title)r r r? r@ rA rC rD rE r r r r7 � s zAnsiToWin32.convert_osc)NNF)r r r! r" �re�compiler8 r] r r. r) r r1 r3 r; r< rB r6 r7 r r r r r# + s ! , r# )r^ r+ r& �ansir r r r r/ r r r Zwin32r r r r �objectr r# r r r r �<module> s
Back to File Manager