File: //lib/python3/dist-packages/twisted/conch/insults/__pycache__/helper.cpython-38.pyc
U
W[<8 � @ s d Z ddlmZ ddlZddlZddlmZ ddlmZ ddl m
Z
m Z m Z ddl
mZmZ ddlmZ dd lmZmZ dd
lmZ d Zd Zed
�\ ZZZ Z Z Z Z Z!Z"G dd� dej#�Z$e$Z%eedddd�ddd� eej&�G dd� de j'��Z(G dd� de)�Z*G dd� de(�Z+dddgZ,dS ) z<
Partial in-memory terminal emulator
@author: Jp Calderone
� )�print_functionN)� implementer)�Version)�defer�protocol�reactor)�log�_textattributes)� iterbytes)�
deprecated�deprecatedModuleAttribute)�insults� �( � c @ sR e Zd ZdZdZejddddeedfdd�Z e
e dddd ��d
d � �Z d d
� Z
dS )�_FormattingStatez�
Represents the formatting state/attributes of a single character.
Character set, intensity, underlinedness, blinkitude, video
reversal, as well as foreground and background colors made up a
character's attributes.
��charset�bold� underline�blink� reverseVideo�
foreground�
background� _subtractingFc C s4 || _ || _|| _|| _|| _|| _|| _|| _d S �Nr ) �selfr r r r r r r r � r �>/usr/lib/python3/dist-packages/twisted/conch/insults/helper.py�__init__, s z_FormattingState.__init__�Twisted�
� r c K s |� � \}}| �||�S )a>
Add a character attribute to a copy of this formatting state.
@param **kw: An optional attribute name and value can be provided with
a keyword argument.
@return: A formatting state instance with the new attribute.
@see: L{DefaultFormattingState._withAttribute}.
)�popitemZ_withAttribute)r �kw�k�vr r r �wantOne9 s z_FormattingState.wantOnec C s� g }| j r|�d� | jr&|�tj� | jr8|�tj� | jrJ|�tj� | j r\|�tj
� | j t krv|�t
| j � | jtkr�|�t| j � |r�dd�tt|�� d S dS )Nr z[�;�m� )r �appendr r
�BOLDr � UNDERLINEr �BLINKr �
REVERSE_VIDEOr �WHITE�
FOREGROUNDr �BLACK�
BACKGROUND�join�map�str)r Zattrsr r r �toVT102I s$
z_FormattingState.toVT102N)�__name__�
__module__� __qualname__�__doc__ZcompareAttributesr
�G0r0 r2 r r r r' r7 r r r r r s �
r r r! r" z=Use twisted.conch.insults.text.assembleFormattedText instead.z twisted.conch.insults.helper�CharacterAttributec @ s� e Zd ZdZdD ] Zed Ze�d�Zee� qdZ dZ
dZ dZ d Z
e� Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdddd �Zded d �Zdfd d �Zdgd!d"�Zd#d$� Z d%d&� Z d'd(� Z d)d*� Z d+d,� Z d-d.� Z!d/d0� Z"d1d2� Z#d3d4� Z$d5d6� Z%d7d8� Z&d9d:� Z'd;d<� Z(d=d>� Z)d?d@� Z*dAdB� Z+dCdD� Z,dEdF� Z-dGdH� Z.dIdJ� Z/dKdL� Z0dMdN� Z1dOdP� Z2dQdR� Z3dSdT� Z4dhdUdV�Z5didWdX�Z6djdYdZ�Z7d[d\� Z8d]d^� Z9d_d`� Z:dadb� Z;dcS )k�TerminalBufferz)
An in-memory terminal emulator.
)s UP_ARROWs
DOWN_ARROWs RIGHT_ARROWs
LEFT_ARROWs HOMEs INSERTs DELETEs ENDs PGUPs PGDNs F1s F2s F3s F4s F5s F6s F7s F8s F9s F10s F11s F12s = object()�ascii� � �P � � c C s | j | | S r )�lines)r �x�yr r r � getCharacter� s zTerminalBuffer.getCharacterc C s | � � d S r )�reset�r r r r �connectionMade� s z TerminalBuffer.connectionMadec C s$ t |�dd��D ]}| �|� qdS )z�
Add the given printable bytes to the terminal.
Line feeds in L{bytes} will be replaced with carriage return / line
feed pairs.
�
s
N)r
�replace�insertAtCursor)r �data�br r r �write� s zTerminalBuffer.writec C s t | jf| j�S r )r �
activeCharset�graphicRenditionrJ r r r �_currentFormattingState� s z&TerminalBuffer._currentFormattingStatec C s� |dkrd| _ n�|dkr"| �� n�|tj�d�kr�| j | jkrF| �� || �� f}| j� t
jj �r�|g| j | j
| j | j �< | j | j
�� n|| j | j
| j < | j d7 _ dS )a�
Add one byte to the terminal at the cursor and make consequent state
updates.
If b is a carriage return, move the cursor to the beginning of the
current row.
If b is a line feed, move the cursor to the next row or scroll down if
the cursor is already in the last row.
Otherwise, if b is printable, put it at the cursor position (inserting
or overwriting as dictated by the current mode) and move the cursor.
�
r rL r? r" N)rF � _scrollDown�stringZ printable�encode�width�nextLinerT �modes�getr
ZIRMrE rG �pop)r rP �chr r r rN � s
z TerminalBuffer.insertAtCursorc s � fdd�t |�D �S )Nc s g | ]}� j � �� f�qS r )�voidrT ��.0�irJ r r �
<listcomp>� s �z-TerminalBuffer._emptyLine.<locals>.<listcomp>)�range�r rY r rJ r �
_emptyLine� s
�zTerminalBuffer._emptyLinec C sH | j d7 _ | j | jkrD| j d8 _ | jd= | j�| �| j�� d S )Nr" r )rG �heightrE r+ rf rY rJ r r r rV � s
zTerminalBuffer._scrollDownc C s@ | j d8 _ | j dk r<d| _ | jd= | j�d| �| j�� d S )Nr" r ���)rG rE �insertrf rY rJ r r r � _scrollUp� s
zTerminalBuffer._scrollUpr" c C s t d| j| �| _d S �Nr )�maxrG �r �nr r r �cursorUp� s zTerminalBuffer.cursorUpc C s t | jd | j| �| _d S �Nr" )�minrg rG rm r r r �
cursorDown� s zTerminalBuffer.cursorDownc C s t d| j| �| _d S rk )rl rF rm r r r �cursorBackward� s z TerminalBuffer.cursorBackwardc C s t | j| j| �| _d S r )rq rY rF rm r r r �
cursorForward� s z TerminalBuffer.cursorForwardc C s || _ || _d S r �rF rG )r �column�liner r r �cursorPosition� s z TerminalBuffer.cursorPositionc C s | j j| _| j j| _d S r )�homerF rG rJ r r r �
cursorHome� s
zTerminalBuffer.cursorHomec C s | � � d S r )rV rJ r r r �index� s zTerminalBuffer.indexc C s | � � d S r )rj rJ r r r � reverseIndex� s zTerminalBuffer.reverseIndexc C s d| _ | �� dS )zW
Update the cursor position attributes and scroll down if appropriate.
r N)rF rV rJ r r r rZ � s zTerminalBuffer.nextLinec C s | j | jf| _d S r )rF rG � _savedCursorrJ r r r �
saveCursor� s zTerminalBuffer.saveCursorc C s | j \| _| _| ` d S r )r} rF rG rJ r r r �
restoreCursor� s z TerminalBuffer.restoreCursorc C s |D ]}d| j |< qd S )NT)r[ �r r[ r) r r r �setModes� s zTerminalBuffer.setModesc C s0 |D ]&}z | j |= W q tk
r( Y qX qd S r )r[ �KeyErrorr� r r r �
resetModes s
zTerminalBuffer.resetModesc C s |D ]}d| j |< qdS )aH
Enable the given modes.
Track which modes have been enabled so that the implementations of
other L{insults.ITerminalTransport} methods can be properly implemented
to respect these settings.
@see: L{resetPrivateModes}
@see: L{insults.ITerminalTransport.setPrivateModes}
TN)� privateModesr� r r r �setPrivateModes
s z TerminalBuffer.setPrivateModesc C s0 |D ]&}z | j |= W q tk
r( Y qX qdS )z�
Disable the given modes.
@see: L{setPrivateModes}
@see: L{insults.ITerminalTransport.resetPrivateModes}
N)r� r� r� r r r �resetPrivateModes s
z TerminalBuffer.resetPrivateModesc C s
d| _ d S )N�app�Z
keypadModerJ r r r �applicationKeypadMode' s z$TerminalBuffer.applicationKeypadModec C s
d| _ d S )NZnumr� rJ r r r �numericKeypadMode+ s z TerminalBuffer.numericKeypadModec C s || j |< d S r )�charsets)r ZcharSetZwhichr r r �selectCharacterSet/ s z!TerminalBuffer.selectCharacterSetc C s t j| _d S r )r
r< rR rJ r r r �shiftIn3 s zTerminalBuffer.shiftInc C s t j| _d S r )r
�G1rR rJ r r r �shiftOut7 s zTerminalBuffer.shiftOutc s. �j �tj�_ �j� � ��fdd�}|�_d S )Nc s � | � �` ��_d S r �rN rR �rP ��fZoldActiveCharsetr r r rN ? s z3TerminalBuffer.singleShift2.<locals>.insertAtCursor)rR r
�G2rN �r rN r r� r � singleShift2; s
zTerminalBuffer.singleShift2c s. �j �tj�_ �j� � ��fdd�}|�_d S )Nc s � | � �` ��_d S r r� r� r� r r rN J s z3TerminalBuffer.singleShift3.<locals>.insertAtCursor)rR r
�G3rN r� r r� r � singleShift3F s
zTerminalBuffer.singleShift3c G s( |D �] }|t jkr*ddddttd�| _q|t jkr@d| jd< q|t jkrVd| jd< q|t jkrld| jd< q|t jkr�d| jd< qz t |�}W n& t
k
r� t � dt
|� � Y qX t| kr�tt kr�n n|t | jd < qt| kr�tt k�rn n|t | jd
< qt � dt
|� � qd S ) NF�r r r r r r Tr r r r z%Unknown graphic rendition attribute: r r )r
ZNORMALr0 r2 rS r, r- r. r/ �int�
ValueErrorr �msg�reprr1 �N_COLORSr3 )r Z
attributes�ar& r r r �selectGraphicRenditionQ s4
�
z%TerminalBuffer.selectGraphicRenditionc C s | � | j�| j| j< d S r )rf rY rE rG rJ r r r � eraseLineq s zTerminalBuffer.eraseLinec C s, | j | j }| �|�| j| j | jd �< d S r )rY rF rf rE rG re r r r �eraseToLineEndu s z TerminalBuffer.eraseToLineEndc C s* | � | jd �| j| j d | jd �< d S rp )rf rF rE rG rJ r r r �eraseToLineBeginningz s z#TerminalBuffer.eraseToLineBeginningc s � fdd�t � j�D �� _d S )Nc s g | ]}� � � j��qS r �rf rY r` rJ r r rc s z/TerminalBuffer.eraseDisplay.<locals>.<listcomp>)rd rg rE rJ r rJ r � eraseDisplay~ s zTerminalBuffer.eraseDisplayc sB � � � � j� j d }� fdd�t|�D �� j� jd d �< d S )Nr" c s g | ]}� � � j��qS r r� r` rJ r r rc � s z4TerminalBuffer.eraseToDisplayEnd.<locals>.<listcomp>)r� rg rG rd rE )r rg r rJ r �eraseToDisplayEnd� s z TerminalBuffer.eraseToDisplayEndc s0 � � � � fdd�t� j�D �� jd � j�< d S )Nc s g | ]}� � � j��qS r r� r` rJ r r rc � s z:TerminalBuffer.eraseToDisplayBeginning.<locals>.<listcomp>)r� rd rG rE rJ r rJ r �eraseToDisplayBeginning� s z&TerminalBuffer.eraseToDisplayBeginningc C sD | j | j | j| j| �= | j | j �| �t| j| j |��� d S r )rE rG rF �extendrf rq rY rm r r r �deleteCharacter� s z TerminalBuffer.deleteCharacterc s6 � fdd�t |�D �� j� j� j�< � j� jd �= d S )Nc s g | ]}� � � j��qS r r� r` rJ r r rc � s z-TerminalBuffer.insertLine.<locals>.<listcomp>)rd rE rG rg rm r rJ r �
insertLine� s $zTerminalBuffer.insertLinec s6 � j � j� j| �= � j �� fdd�t|�D �� d S )Nc s g | ]}� � � j��qS r r� r` rJ r r rc � s z-TerminalBuffer.deleteLine.<locals>.<listcomp>)rE rG r� rd rm r rJ r �
deleteLine� s zTerminalBuffer.deleteLinec C s | j | jfS r ru rJ r r r �reportCursorPosition� s z#TerminalBuffer.reportCursorPositionc C s� t �dd�| _d | _| _i | _i | _| �t jjt jj g� d| _
t j | _ ddddt
td�| _t j t jt jt jt jt jt jt ji| _| �� d S )Nr r� Fr� )r
ZVectorry rF rG r[ r� r� Z AUTO_WRAPZ CURSOR_MODEZ
numericKeypadr< rR r0 r2 rS ZCS_USr� r� Z CS_ALTERNATEr� ZCS_ALTERNATE_SPECIALr� r� rJ r r r rI � s4
�� �zTerminalBuffer.resetc C s t dt|�� d S )NzCould not handle)�printr� )r �bufr r r �unhandledControlSequence� s z'TerminalBuffer.unhandledControlSequencec C sr g }| j D ]\}g }d}|D ]2\}}|| jk r@|�|� t|�}q|�| j� q|�d�|d |� �� q
d�|�S )Nr � rL )rE r_ r+ �len�fillr4 )r rE �Lr� Zlengthr^ �attrr r r � __bytes__� s
zTerminalBuffer.__bytes__N)r" )r" )r" )r" )r" )r" )r" )<r8 r9 r: r; ZkeyIDZ execBytes�decodeZexecStr�execZTABZ BACKSPACErY rg r� �objectr_ rH rK rQ rT rN rf rV rj ro rr rs rt rx rz r{ r| rZ r~ r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rI r� r� r r r r r> j sn
r> c @ s e Zd ZdS )�ExpectationTimeoutN)r8 r9 r: r r r r r� � s r� c @ sF e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd e fd
d�Z
d S )�ExpectableBufferr c C s t �| � g | _d S r )r> rK �
_expectingrJ r r r rK � s
z ExpectableBuffer.connectionMadec C s t �| |� | �� d S r )r> rQ �_checkExpected)r rO r r r rQ � s zExpectableBuffer.writec C s t �| � d| _d S rk )r> rz �_markrJ r r r rz � s
zExpectableBuffer.cursorHomec C s |� t� � | �� d S r )Zerrbackr� r� )r �dr r r �_timeoutExpected� s z!ExpectableBuffer._timeoutExpectedc C s� | � � | jd � }| jr�| jd \}}}|r>|�� s>| jd= q|�|�D ]H}|rX|�� | jd= | j|�� 7 _||�� d � }|�|� qqHd S qd S rk )r� r� r� Zactive�finditerZcancel�end�callback)r �s�expr�timerZdeferred�matchr r r r� � s
z ExpectableBuffer._checkExpectedNc C sD t �� }d }|r |�|| j|�}| j�t�|�||f� | �� |S r ) r ZDeferredZ callLaterr� r� r+ �re�compiler� )r Z
expressionZtimeoutZ schedulerr� r� r r r �expect� s zExpectableBuffer.expect) r8 r9 r: r� rK rQ rz r� r� r r� r r r r r� � s r� )-r; Z
__future__r r� rW Zzope.interfacer Z incrementalr Ztwisted.internetr r r Ztwisted.pythonr r Ztwisted.python.compatr
Ztwisted.python.deprecater r Ztwisted.conch.insultsr
r1 r3 rd r2 ZREDZGREENZYELLOWZBLUEZMAGENTAZCYANr0 r� Z_FormattingStateMixinr r= ZITerminalTransportZProtocolr> � Exceptionr� r� �__all__r r r r �<module> s< @ � c4 �