3
.�]�5 � @ s� d Z ddlZddlZddlZddlZejddd� dZdZdZdZ d Z
d
Z d Z d Z
d
ZdZdZdZdZdZdZdZdZdZejd dkZer�eZdd� Z G dd� d�Z dS ) a� This implements a virtual screen. This is used to support ANSI terminal
emulation. The screen representation and state is implemented in this class.
Most of the methods are inspired by ANSI screen control codes. The
:class:`~pexpect.ANSI.ANSI` class extends this class to add parsing of ANSI
escape codes.
PEXPECT LICENSE
This license is approved by the OSI and FSF as GPL-compatible.
http://opensource.org/licenses/isc-license.txt
Copyright (c) 2012, Noah Spurrier <noah@noah.org>
PERMISSION TO USE, COPY, MODIFY, AND/OR DISTRIBUTE THIS SOFTWARE FOR ANY
PURPOSE WITH OR WITHOUT FEE IS HEREBY GRANTED, PROVIDED THAT THE ABOVE
COPYRIGHT NOTICE AND THIS PERMISSION NOTICE APPEAR IN ALL COPIES.
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
� Nz�pexpect.screen and pexpect.ANSI are deprecated. We recommend using pyte to emulate a terminal screen: https://pypi.python.org/pypi/pyte� )�
stacklevel� � � � �
� � �
� � � � � � � � � � c C s | |k r |S | |kr|S | S )z@This returns a number, n constrained to the min and max bounds. � )�n�min�maxr r � /usr/lib/python3.6/screen.py� constrain< s
r c @ s� e Zd ZdZdbdd�Zdd � Zd
d � Zer0eZn eZ d d
� Zdd� Z
dd� Z e fdd�Z
e fdd�Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zdcd/d0�Zddd1d2�Z ded3d4�Z dfd5d6�Z dgd7d8�Z d9d:� Z d;d<� Z!d=d>� Z"d?d@� Z#dAdB� Z$dCdD� Z%dEdF� Z&dGdH� Z'dIdJ� Z(dKdL� Z)dMdN� Z*dOdP� Z+dQdR� Z,dSdT� Z-dUdV� Z.dWdX� Z/dYdZ� Z0d[d\� Z1d]d^� Z2d_d`� Z3daS )h�screena5 This object maintains the state of a virtual text screen as a
rectangular array. This maintains a virtual cursor position and handles
scrolling as characters are added. This supports most of the methods needed
by an ANSI text screen. Row and column indexes are 1-based (not zero-based,
like arrays).
Characters are represented internally using unicode. Methods that accept
input characters, when passed 'bytes' (which in Python 2 is equivalent to
'str'), convert them from the encoding specified in the 'encoding'
parameter to the constructor. Methods that return screen contents return
unicode strings, with the exception of __str__() under Python 2. Passing
``encoding=None`` limits the API to only accept unicode input, so passing
bytes in will raise :exc:`TypeError`.
r �P �latin-1�replacec s| |� _ |� _|� _|� _|dk r2tj|�|�� _nd� _d� _d� _d� _ d� _
d� _ � j � _ � fdd�t
� j �D �� _dS )z8This initializes a blank screen of the given dimensions.N� c s g | ]}t g� j �qS r )�SPACE�cols)�.0�_)�selfr r �
<listcomp>f s z#screen.__init__.<locals>.<listcomp>)�rowsr" �encoding�encoding_errors�codecs�getincrementaldecoder�decoder�cur_r�cur_c� cur_saved_r� cur_saved_c�scroll_row_start�scroll_row_end�range�w)r% �r�cr( r) r )r% r �__init__U s zscreen.__init__c C s" | j dk r| j j|�S td��dS )ztThis converts from the external coding system (as passed to
the constructor) to the internal one (unicode). NzLThis screen was constructed with encoding=None, so it does not handle bytes.)r, �decode� TypeError)r% �sr r r �_decodeh s
zscreen._decodec C s dj dd� | jD ��S )z�This returns a printable representation of the screen as a unicode
string (which, under Python 3.x, is the same as 'str'). The end of each
screen line is terminated by a newline.�
c S s g | ]}d j |��qS )� )�join)r# r6 r r r r& v s z#screen._unicode.<locals>.<listcomp>)r> r4 )r% r r r �_unicodeq s zscreen._unicodec C s | j pd}| j� j|d�S )z{This returns a printable representation of the screen. The end of
each screen line is terminated by a newline. �asciir )r( r? �encode)r% r( r r r �__str__} s
zscreen.__str__c C s dj dd� | jD ��S )z�This returns a copy of the screen as a unicode string. This is similar to
__str__/__unicode__ except that lines are not terminated with line
feeds.r= c S s g | ]}d j |��qS )r= )r> )r# r6 r r r r& � s z screen.dump.<locals>.<listcomp>)r> r4 )r% r r r �dump� s z screen.dumpc C s< dd| j d }|djdd� t| �jd�D �� d | S )z�This returns a copy of the screen as a unicode string with an ASCII
text box around the screen border. This is similar to
__str__/__unicode__ except that it adds a box.�+�-z+
r< c S s g | ]}d | d �qS )�|r )r# �liner r r r& � s z!screen.pretty.<locals>.<listcomp>)r" r> �unicode�split)r% Ztop_botr r r �pretty� s z
screen.prettyc C s. t |t�r| j|�}| jdd| j| j|� d S )Nr )�
isinstance�bytesr; � fill_regionr' r" )r% �chr r r �fill� s
z screen.fillc C s� t |t�r| j|�}t|d| j�}t|d| j�}t|d| j�}t|d| j�}||kr^|| }}||krp|| }}x<t||d �D ]*}x$t||d �D ]}| j|||� q�W q�W d S )Nr )rK rL r; r r' r" r3 �put_abs)r% �rs�cs�re�cerN r5 r6 r r r rM � s
zscreen.fill_regionc C s | j | jd� dS )zKThis moves the cursor to the beginning (col 1) of the current row.
r N)� cursor_homer- )r% r r r �cr� s z screen.crc C s, | j }| j� || j kr(| j� | j� dS )z3This moves the cursor down with scrolling.
N)r- � cursor_down� scroll_up�
erase_line)r% �old_rr r r �lf� s
z screen.lfc C s | j � | j� dS )ztThis advances the cursor with CRLF properties.
The cursor will line wrap and the screen may scroll.
N)rV r[ )r% r r r �crlf� s z screen.crlfc C s | j � dS )z%This is an alias for crlf().
N)r\ )r% r r r �newline� s zscreen.newlinec C sX t |d| j�}t |d| j�}t|t�r6| j|�d }n|d }|| j|d |d < dS )z Screen array starts at 1 index.r r N)r r' r" rK rL r; r4 )r% r5 r6 rN r r r rP � s
zscreen.put_absc C s* t |t�r| j|�}| j| j| j|� dS )z?This puts a characters at the current cursor position.
N)rK rL r; rP r- r. )r% rN r r r �put� s
z
screen.putc C st t |t�r| j|�}t|d| j�}t|d| j�}x0t| j|d�D ] }| j||| j||d �� q@W | j|||� dS )z�This inserts a character at (r,c). Everything under
and to the right is shifted right one character.
The last character of the line is lost.
r N���) rK rL r; r r' r" r3 rP �get_abs)r% r5 r6 rN Zcir r r �
insert_abs� s
zscreen.insert_absc C s* t |t�r| j|�}| j| j| j|� d S )N)rK rL r; ra r- r. )r% rN r r r �insert� s
z
screen.insertc C s2 t |d| j�}t |d| j�}| j|d |d S )Nr )r r' r" r4 )r% r5 r6 r r r r` � s zscreen.get_absc C s | j | j| j� d S )N)r` r- r. )r% r r r �get� s z
screen.getc
C s� t |d| j�}t |d| j�}t |d| j�}t |d| j�}||krJ|| }}||kr\|| }}g }xPt||d �D ]>}d}x*t||d �D ]}| j||�} || }q�W |j|� qpW |S )z>This returns a list of lines representing the region.
r r= )r r' r" r3 r` �append)
r% rQ rR rS rT Zscr5 rG r6 rN r r r �
get_region� s
zscreen.get_regionc C s( t | jd| j�| _t | jd| j�| _dS )z6This keeps the cursor within the screen area.
r N)r r- r' r. r" )r% r r r �cursor_constrain s zscreen.cursor_constrainr c C s || _ || _| j� d S )N)r- r. rf )r% r5 r6 r r r rU s zscreen.cursor_homec C s | j | | _ | j� d S )N)r. rf )r% �countr r r � cursor_back s zscreen.cursor_backc C s | j | | _ | j� d S )N)r- rf )r% rg r r r rW # s zscreen.cursor_downc C s | j | | _ | j� d S )N)r. rf )r% rg r r r �cursor_forward( s zscreen.cursor_forwardc C s | j | | _ | j� d S )N)r- rf )r% rg r r r � cursor_up- s zscreen.cursor_upc C s$ | j }| j� || j kr | j� d S )N)r- rj rX )r% rZ r r r �cursor_up_reverse2 s
zscreen.cursor_up_reversec C s | j ||� dS )zIdentical to Cursor Home.N)rU )r% r5 r6 r r r �cursor_force_position9 s z screen.cursor_force_positionc C s | j � dS )z Save current cursor position.N)�cursor_save_attrs)r% r r r � cursor_save> s zscreen.cursor_savec C s | j � dS )z-Restores cursor position after a Save Cursor.N)�cursor_restore_attrs)r% r r r �
cursor_unsaveC s zscreen.cursor_unsavec C s | j | _| j| _dS )z Save current cursor position.N)r- r/ r. r0 )r% r r r rm H s zscreen.cursor_save_attrsc C s | j | j| j� dS )z-Restores cursor position after a Save Cursor.N)rU r/ r0 )r% r r r ro N s zscreen.cursor_restore_attrsc C s( | j dkrd| _ | j| jkr$| j| _dS )z6This keeps the scroll region within the screen region.r r N)r1 r2 r' )r% r r r �scroll_constrainS s
zscreen.scroll_constrainc C s d| _ | j| _dS )z$Enable scrolling for entire display.r N)r1 r' r2 )r% r r r �
scroll_screen[ s zscreen.scroll_screenc C s || _ || _| j� dS )z/Enable scrolling from row {start} to row {end}.N)r1 r2 rq )r% rQ rS r r r �scroll_screen_rowsa s zscreen.scroll_screen_rowsc C s> | j d }| jd }tj| j||� �| j|d |d �< dS )z Scroll display down one line.r N)r1 r2 �copy�deepcopyr4 )r% r: �er r r � scroll_downh s
zscreen.scroll_downc C s> | j d }| jd }tj| j|d |d � �| j||�< dS )zScroll display up one line.r N)r1 r2 rt ru r4 )r% r: rv r r r rX p s
zscreen.scroll_upc C s | j | j| j| j| j� dS )zOErases from the current cursor position to the end of the current
line.N)rM r- r. r" )r% r r r �erase_end_of_linex s zscreen.erase_end_of_linec C s | j | jd| j| j� dS )zQErases from the current cursor position to the start of the current
line.r N)rM r- r. )r% r r r �erase_start_of_line~ s zscreen.erase_start_of_linec C s | j | jd| j| j� dS )z Erases the entire current line.r N)rM r- r" )r% r r r rY � s zscreen.erase_linec C s& | j � | j| jd d| j| j� dS )zQErases the screen from the current line down to the bottom of the
screen.r N)rx rM r- r' r" )r% r r r �
erase_down� s zscreen.erase_downc C s$ | j � | j| jd dd| j� dS )zLErases the screen from the current line up to the top of the
screen.r N)ry rM r- r" )r% r r r �erase_up� s zscreen.erase_upc C s | j � dS )z,Erases the screen with the background color.N)rO )r% r r r � erase_screen� s zscreen.erase_screenc C s dS )z#Sets a tab at the current position.Nr )r% r r r �set_tab� s zscreen.set_tabc C s dS )z#Clears tab at the current position.Nr )r% r r r � clear_tab� s zscreen.clear_tabc C s dS )zClears all tabs.Nr )r% r r r �clear_all_tabs� s zscreen.clear_all_tabsN)r r r r )r r )r )r )r )r )4�__name__�
__module__� __qualname__�__doc__r7 r; r? �PY3rB Z __unicode__rC rJ r! rO rM rV r[ r\ r] rP r^ ra rb r` rc re rf rU rh rW ri rj rk rl rn rp rm ro rq rr rs rw rX rx ry rY rz r{ r| r} r~ r r r r r r F sb
r ) r� r* rt �sys�warnings�warnZNULZENQZBELZBSZHTZLFZVTZFFZCR�SOZSIZXONZXOFFZCANZSUBZESCZDELr! � version_infor� �strrH r r r r r r �<module> s: