File: //lib/python3/dist-packages/twisted/conch/ui/__pycache__/ansi.cpython-38.pyc
U
W[
� @ sp d Z ddlZddlmZ G dd� d�ZG dd� d�Zd�edjej��Z i Z
ejD ]Z e D ] Z e e
e < qZqR[ dS ) zGModule to parse ANSI escape sequences
Maintainer: Jean-Paul Calderone
� N)�logc @ sN e Zd ZdZdZedd� eD ��Zeee��\Z Z
Z Z Z
ZZZdZdd� ZdS ) � ColorTextzb
Represents an element of text along with the texts colors and
additional attributes.
)�b�r�g�y�l�m�c�wc C s g | ] }|� � �qS � )�upper)�.0�xr r �7/usr/lib/python3/dist-packages/twisted/conch/ui/ansi.py�
<listcomp> s zColorText.<listcomp>)ZBlackZRedZGreenZYellowZBlueZMagentaZCyanZWhitec C sP ||| | _ | _| _|| _|| _|| _|| _|| _| jrL| j| j | _| _d S �N)�text�fg�bg�display�bold� underline�flash�reverse) �selfr r r r r r r r r r r �__init__ s zColorText.__init__N)�__name__�
__module__� __qualname__�__doc__�COLORS�tuple� BOLD_COLORS�range�lenZBLACKZREDZGREENZYELLOWZBLUEZMAGENTAZCYANZWHITEZ COLOR_NAMESr r r r r r s r c @ s� e Zd ZdZdZdZdZdZdZeeeeefZ dd� Z
d d
� Z d d � Z d
d� Z
dd� Zdd� Zdd� Zejfdd�Zdd� ZdS )�
AnsiParserz&
Parser class for ANSI codes.
) �H�f�A�B�C�D�R�s�u�d�G)�J�K�P)�hr )�p)r c C sD || | _ | _| j | j | _| _d\| _| _| _| _d| _d| _ d S )N�r r r r � � )
� defaultFG� defaultBG� currentFG� currentBGr r r r r �prepend)r r: r; r r r r @ s
zAnsiParser.__init__c C s� d}d}d}t |�}||k r�|dkr6|| tkr6d}n@|rv|�d|�}|dkr^|||d� S ||||� }|}d}|d }q|S )zF
Remove all ANSI color escapes from the given string.
r9 r8 r ����N)r% �_sets�find)r �string�resultZshow�i�L�nr r r � stripEscapesH s
zAnsiParser.stripEscapesc C s d S r r )r Zcolorstrr r r � writeString^ s zAnsiParser.writeStringc C s| | j r| j | }d| _ |�d�}t|�dkrD| �| �|d �� �n4| �| �|d �� |dd� D �]}t|�}d}d}||k r�|| tjd kr�q�|d7 }qz|s�d| _ dS |d dkr�| �| �||d d� �� qdn|dd� }|d8 }||d k�r
d| _ dS t�|| d�}|dk�r&qd|t j
k�rj| � |d|d � � ||d d� }| �| �|�� qd|t j k�r�|d|d � ||d d� }}| �
|� | �| �|�� qd|t jk�r�|d|d � ||d d� }}| �|� | �| �|�� qd|t jk�r,||d d� }| �| �|�� qd||k�rBd| | _ qdt�d || f � ||d d� }| �| �|�� qddS )
zK
Turn a string input into a list of L{ColorText} elements.
r9 r? r8 r Nz[;?�[z[z Unhandled ANSI control type: %c)r> �splitr% rI �
formatTextrC Zdigits�_setmap�getr& � COLOR_SET�
parseColor�
CURSOR_SET� parseCursor� ERASE_SET�
parseErase�MODE_SETr �msg) r �str�partsr. rF rE �type�cursor�eraser r r � parseStringa sb
"
"
zAnsiParser.parseStringc C s� |dd� }|sd}zt t|�d��}W n> tk
rf t�dt|�|f � | j| j | _ | _
Y dS X |D �]p}|dkr�| j| j | _ | _
d\| _ | _ | _
| _d| _ql|dkr�d| _ qld | kr�d
kr�n n |d | _ qld | k� r�d k�rn n |d | _
ql|d
k�r| j| _ ql|dk�r,| j| _
ql|dk�r>d| _
ql|dk�rPd| _ ql|dk�rbd| _ql|dk�rtd| _ql|dk�r�d| _ ql|dk�r�d| _
ql|dk�r�d| _ql|dk�r�d| _ql|dk�r�d| _qlt�d|f � qldS )z5
Handle a single ANSI color sequence
Nr@ �0�;z$Invalid ANSI color sequence (%d): %sr r7 r8 � �% �( �/ �' �1 � � � � � � � � � z#Unrecognised ANSI color command: %d)�map�intrK �
ValueErrorr rV r% r: r; r< r= r r r r r Zblink)r rW rX r r r r rP � sV
zAnsiParser.parseColorc C s d S r r )r rZ r r r rR � s zAnsiParser.parseCursorc C s d S r r )r r[ r r r rT � s zAnsiParser.parseErasec C s* |rt j| S | jr || p$t j| S d S r )r r! r )r �value�modeZBOLDr r r � pickColor� s
zAnsiParser.pickColorc C s4 t || �| jd�| �| jd�| j| j| j| j| j�S )Nr r8 ) r rs r< r= r r r r r )r r r r r rL � s �zAnsiParser.formatTextN)r r r r rQ rS rU Z
ASSIGN_SETrO �SETSr rH rI r\ rP rR rT r r# rs rL r r r r r& * s =5r& r9 )
r rC Ztwisted.pythonr r r& �joinrn rt rA rM r. r r r r r �<module> s A