File: //lib/python3/dist-packages/twisted/conch/__pycache__/manhole.cpython-38.pyc
U
W[8. � @ s� d Z ddlZddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z m Z ddl
mZ G dd � d �ZG d
d � d ej�Zd Zd
ZdZdZdZdZG dd� dej�ZG dd� d�Zdd� ZG dd� de�Z dS )aO
Line-input oriented interactive interpreter loop.
Provides classes for handling Python source input and arbitrary output
interactively from a Twisted application. Also included is syntax coloring
code with support for VT102 terminals, control code handling (^C, ^D, ^Q),
and reasonable handling of Deferreds.
@author: Jp Calderone
� N)�BytesIO)�recvline)�defer)� _tokenize�_get_async_param)� TokenPrinterc @ s8 e Zd ZdZdZdZdd� Zdd� Zdd � Zd
d � Z d S )
� FileWrapperz�
Minimal write-file-like object.
Writes are translated into addOutput calls on an object passed to
__init__. Newlines are also converted from network to local style.
r Znormalc C s
|| _ d S �N)�o)�selfr
� r �7/usr/lib/python3/dist-packages/twisted/conch/manhole.py�__init__& s zFileWrapper.__init__c C s d S r r �r r r r
�flush* s zFileWrapper.flushc C s | j �|�dd�� d S )Nz
�
)r
� addOutput�replace)r �datar r r
�write. s zFileWrapper.writec C s | � d�|�� d S )N� )r �join)r �linesr r r
�
writelines2 s zFileWrapper.writelinesN)
�__name__�
__module__� __qualname__�__doc__Z softspace�stater r r r r r r r
r s r c @ sX e Zd ZdZdZddd�Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z ddd�Z dS )�ManholeInterpreteraf
Interactive Interpreter with special output and Deferred support.
Aside from the features provided by L{code.InteractiveInterpreter}, this
class captures sys.stdout output and redirects it to the appropriate
location (the Manhole protocol instance). It also treats Deferreds
which reach the top-level specially: each is formatted to the user with
a unique identifier and a new callback and errback added to it, each of
which will format the unique identifier and the result with which the
Deferred fires and then pass it on to the next participant in the
callback chain.
r N� <console>c C s, t j�| |� i | _|| _|| _| �� d S r )�code�InteractiveInterpreterr �_pendingDeferreds�handler�filename� resetBuffer)r r$ �localsr% r r r
r F s
zManholeInterpreter.__init__c C s
g | _ dS )z)
Reset the input buffer.
N)�bufferr r r r
r& N s z ManholeInterpreter.resetBufferc C s@ | j �|� d�| j �}|�d�}| �|| j�}|s<| �� |S )a
Push a line to the interpreter.
The line should not have a trailing newline; it may have
internal newlines. The line is appended to a buffer and the
interpreter's runsource() method is called with the
concatenated contents of the buffer as source. If this
indicates that the command was executed or invalid, the buffer
is reset; otherwise, the command is incomplete, and the buffer
is left as it was after the line was appended. The return
value is 1 if more input is required, 0 if the line was dealt
with in some way (this is the same as runsource()).
@param line: line of text
@type line: L{bytes}
@return: L{bool} from L{code.InteractiveInterpreter.runsource}
�
�utf-8)r( �appendr �decodeZ runsourcer% r& )r �line�source�morer r r
�pushU s
zManholeInterpreter.pushc O sX t j| j }t _z:t jt| j� }t _ztjj| f|�|� W 5 |t _X W 5 |t _X d S r )�sys� displayhook�stdoutr r$ r! r" �runcode)r �a�kwZorighookZorigoutr r r
r4 p s zManholeInterpreter.runcodec C s� || j d< t|tj�r�t|d�r0| �t|�� q�t|�| jkr^| �d| jt|� d f � q�| j}| j }||f|t|�< | j d7 _ |j
| j | j ||f||fd� | �d|f � n|d k r�| �t|�� d S )N�_�resultz<Deferred #%d>r � )Z callbackArgsZ errbackArgs)
r' �
isinstancer ZDeferred�hasattrr �repr�idr# � numDeferredsZ addCallbacks�_cbDisplayDeferred�_ebDisplayDeferred)r �obj�d�kr r r
r2 | s"
�z ManholeInterpreter.displayhookc C s$ | � d||f d� | jt|�= |S )Nz Deferred #%d called back: %rT)r r# r= )r r8 rC rA r r r
r? � s z%ManholeInterpreter._cbDisplayDeferredc C s( | � d||�� f d� | jt|�= |S )NzDeferred #%d failed: %rT)r ZgetErrorMessager# r= )r ZfailurerC rA r r r
r@ � s z%ManholeInterpreter._ebDisplayDeferredc K s t |f|�}| j�||� d S r )r r$ r )r r �isAsync�kwargsr r r
r � s zManholeInterpreter.write)Nr )N)
r r r r r> r r& r0 r4 r2 r? r@ r r r r r
r 7 s
r � � � � � � c @ s` e Zd ZdZdZddd�Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z ddd�Z dd� Z
dS )�Manholea�
Mediator between a fancy line source and an interactive interpreter.
This accepts lines from its transport and passes them on to a
L{ManholeInterpreter}. Control commands (^C, ^D, ^\) are also handled
with something approximating their normal terminal-mode behavior. It
can optionally be constructed with a dict which will be used as the
local namespace for any code executed.
Nc C s" t j�| � |d k r |�� | _d S r )r �HistoricRecvLiner �copy� namespace)r rO r r r
r � s zManhole.__init__c C sf t j�| � t| | j�| _| j| jt< | j | jt
< | j | jt < | j
| jt< | j| jt< | j| jt< d S r )r rM �connectionMader rO � interpreter�
handle_INTZ keyHandlers�CTRL_C�
handle_EOF�CTRL_D� handle_FF�CTRL_LZ handle_HOME�CTRL_AZ
handle_END�CTRL_E� handle_QUIT�CTRL_BACKSLASHr r r r
rP � s zManhole.connectionMadec C sT d| _ g | _d| _| j�� | j�� | j�d� | j�� | j�| j| j � dS )z~
Handle ^C as an interrupt keystroke by resetting the current input
variables to their initial state.
r s KeyboardInterruptN) �pn�
lineBuffer�lineBufferIndexrQ r& �terminal�nextLiner �psr r r r
rR � s
zManhole.handle_INTc C s | j r| j�d� n| �� d S )N� )r] r_ r rZ r r r r
rT � s zManhole.handle_EOFc C s | j �� | j �� | �� dS )zh
Handle a 'form feed' byte - generally used to request a screen
refresh/redraw.
N)r_ Z eraseDisplayZ
cursorHomeZ
drawInputLiner r r r
rV � s
zManhole.handle_FFc C s | j �� d S r )r_ ZloseConnectionr r r r
rZ � s zManhole.handle_QUITc C s | j j}|�d� o |�d� S )Nr) s E)r_ Z lastWrite�endswith)r �wr r r
�
_needsNewline� s zManhole._needsNewlinec K s� t |f|�}|r<| j�� | j�t| j�t| j| j � � | j�|� |r�| � � r^| j�
� | j�| j| j � | jr�| j}g | _d| _ | � |� d S )Nr )
r r_ � eraseLine�cursorBackward�lenr] ra r\ r re r` r^ Z_deliverBuffer)r r rD rE Z oldBufferr r r
r � s
�
zManhole.addOutputc C s@ | j �|�}t|�| _| �� r(| j�� | j�| j| j � d S r ) rQ r0 �boolr\ re r_ r` r ra )r r- r/ r r r
� lineReceived
s
zManhole.lineReceived)N)N)r r r r rO r rP rR rT rV rZ re r rj r r r r
rL � s
rL c @ sV e Zd ZdZdddddddd �Zd
Zd d � Zd
d� Zddd�Zdd� Z e
e krRe Z dS )� VT102Writerz�
Colorizer for Python tokens.
A series of tokens are written to instances of this object. Each is
colored in a particular way. The final line of the result of this is
generally added to the output.
s [31ms [32ms [33ms [1;33ms [35ms [36ms [37m)Z
identifier�keywordZ parameterZvariable�stringZnumber�ops [0mc C s
g | _ d S r )�writtenr r r r
r ' s zVT102Writer.__init__c C s | j �|d�}|S )N� )� typeToColor�get)r �type�rr r r
�color+ s zVT102Writer.colorNc C sH |rD|dkrD| � |�}|r&| j�|� | j�|� |rD| j�| j� d S )N�
)ru ro r+ � normalColor)r �tokenrs �cr r r
r 0 s
zVT102Writer.writec C s d� | j�}|�d��� d S )Nrp r) ���)r ro �strip�
splitlines)r �sr r r
� __bytes__: s zVT102Writer.__bytes__)N)
r r r r rq rw r ru r r~ �bytes�str�__str__r r r r
rk s �
rk c
C sd t | t�s| �d�} t� }t|j�j}t| �}t|j �D ]"}|\}}}}} |||||| � q8t|�S )z�
Tokenize and colorize the given Python source.
Returns a VT102-format colorized version of the last line of C{source}.
@param source: Python source code
@type source: L{str} or L{bytes}
@return: L{bytes} of colorized source
r* )
r: r �encoderk r r Z
printtokenr r �readline)
r. rd �pr} rx Z tokenTyperm �start�endr- r r r
�lastColorizedLineD s
r� c @ s e Zd ZdZdd� Zdd� ZdS )�ColoredManholez<
A REPL which syntax colors input as users type it.
c C s d� | jj�d d� | j� S )z�
Return a string containing the currently entered source.
This is only the code which will be considered for execution
next.
r) rp )r rQ r( r] r r r r
� getSourcea s
�
�zColoredManhole.getSourcec C s | j dkr | j�| j|� n|g| j| j| jd �< | jd7 _|rJd S |dkrb| j�|� d S | �� }z t|�}W n" tj k
r� | j�|� Y nlX | j�
� | j� t | j�t | j
| j � d � | j�| j
| j | � t | j�| j }|�r| j� |� d S )N�insertr9 � )�moder] r� r^ r_ r r� r� �tokenize�
TokenErrorrf rg rh ra r\ )r ZchZmoreCharactersComingr. Z coloredLine�nr r r
�characterReceivedm s(
&z ColoredManhole.characterReceivedN)r r r r r� r� r r r r
r� \ s r� ) r r! r1 r� �ior Z
twisted.conchr Ztwisted.internetr Ztwisted.python.compatr r Ztwisted.python.htmlizerr r r" r rS rU r[ rW rX rY rM rL rk r� r� r r r r
�<module> s$ kh1