File: //proc/1233/cwd/usr/lib/python3/dist-packages/dockerpty/__pycache__/pty.cpython-310.pyc
o
���V�+ � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ G dd� de �Z
G dd� de �Z G dd� de �Z dd
d �Z
G d d
� d
e �ZG dd� de �ZdS )� N)�SSLErrorc @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� WINCHHandlerz?
WINCH Signal handler to keep the PTY correctly sized.
c C s || _ d| _dS )z�
Initialize a new WINCH handler for the given PTY.
Initializing a handler has no immediate side-effects. The `start()`
method must be invoked for the signals to be trapped.
N)�pty�original_handler)�selfr � r �//usr/lib/python3/dist-packages/dockerpty/pty.py�__init__ s
zWINCHHandler.__init__c C s | � � | S )z5
Invoked on entering a `with` block.
)�start�r r r r � __enter__* s zWINCHHandler.__enter__c G s | � � dS )z4
Invoked on exiting a `with` block.
N)�stop)r �_r r r �__exit__2 s zWINCHHandler.__exit__c s � fdd�}t � t j|�� _dS )z�
Start trapping WINCH signals and resizing the PTY.
This method saves the previous WINCH handler so it can be restored on
`stop()`.
c s | t jkr � j�� d S d S �N)�signal�SIGWINCHr �resize)�signum�framer r r �handleA s
�z"WINCHHandler.start.<locals>.handleN)r r r )r r r r r r
9 s zWINCHHandler.startc C s" | j durt�tj| j � dS dS )zU
Stop trapping WINCH signals and restore the previous WINCH handler.
N)r r r r r r r r
G s
�zWINCHHandler.stopN) �__name__�
__module__� __qualname__�__doc__r r r r
r
r r r r r s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� Operationc K � t � �)z3
are we dealing with a tty or not?
��NotImplementedError�r �kwargsr r r �israwR � zOperation.israwc K r )�!
start execution
r r r r r r
X r" zOperation.startc K r )z0
if we have terminal, resize it
r �r �height�widthr r r r r ^ r" zOperation.resizec C r )zReturn sockets for streams.r r r r r �socketsd s zOperation.socketsN)r r r r! r
r r' r r r r r P s
r c @ sD e Zd ZdZddd�Zddd�Zdd � Zd
d � Zd d
� Zdd� Z dS )� RunOperationz6
class for handling `docker run`-like command
TNc C s^ |du r t �dt� d}|| _|| _d| _|| _|p tj| _|p"tj | _ |p(tj
| _
|| _ dS )�Y
Initialize the PTY using the docker.Client instance and container dict.
Nz�The default behaviour of dockerpty is changing. Please add logs=1 to your dockerpty.start call to maintain existing behaviour. See https://github.com/d11wtq/dockerpty/issues/51 for details.� ) �warnings�warn�DeprecationWarning�client� container�raw� interactive�sys�stdout�stderr�stdin�logs)r r. r/ r1 r3 r4 r5 r6 r r r r n s
zRunOperation.__init__c K s� |p| � � \}}}g }|r | jr |�tjt�| j�|dd�� |r0|�tj|t�| j�dd�� |rA|�tj|t�| j�dd�� | � � d d sT| j
j | j fi |�� |S )z�
Present the PTY of the container inside the current process.
This will take over the current process' TTY until the container's PTY
is closed.
F��wait_for_output��propagate_close�State�Running)
r' r1 �append�io�Pump�Streamr5 r3 r4 �_container_infor. r
r/ )r r' r � pty_stdin�
pty_stdout�
pty_stderr�pumpsr r r r
� s
zRunOperation.startc K s0 | j du r| �� }| j�� o|d d | _ | j S )z�
Returns True if the PTY should operate in raw mode.
If the container was not started with tty=True, this will return False.
N�Config�Tty)r0 rA r3 �isatty)r r �infor r r r! � s
zRunOperation.israwc s �� � � � �fdd�}t|d�S )z�
Returns a tuple of sockets connected to the pty (stdin,stdout,stderr).
If any of the sockets are not attached in the container, `None` is
returned in the tuple.
c s\ � d d� | �� � r,�j��j| dddd�ji�}t�|�}� d d r'|S t�|�S d S )NrF z Attach{0}r* �streamr6 rG ) �format�
capitalizer. �
attach_socketr/ r6 r> r@ �Demuxer)�key�socketrJ �rI r r r rM � s �
z+RunOperation.sockets.<locals>.attach_socket)r5 r3 r4 )rA �map)r rM r rQ r r' � s
zRunOperation.socketsc K � | j j| j||d� dS )z-
resize pty within container
�r% r& N)r. r r/ r$ r r r r � � zRunOperation.resizec C s | j �| j�S )zA
Thin wrapper around client.inspect_container().
)r. �inspect_containerr/ r r r r rA � s z RunOperation._container_info)TNNNNr )
r r r r r r
r! r' r rA r r r r r( i s
r( Tc C s | j ||||d�}|S )N)�ttyr5 )� exec_create)r. r/ �commandr1 �exec_idr r r rX � s rX c @ sL e Zd ZdZddd�Zddd�Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dS )�
ExecOperationz7
class for handling `docker exec`-like command
TNc C sF || _ || _d | _|| _|ptj| _|ptj| _|p tj| _d | _d S r ) rZ r. r0 r1 r2 r3 r4 r5 �_info)r r. rZ r1 r3 r4 r5 r r r r � s
zExecOperation.__init__c K sV |p| � � }g }| jr|�tjt�| j�|dd�� |�tj|t�| j�dd�� |S )r# Fr7 r9 )r' r1 r= r> r? r@ r5 r3 )r r' r rJ rE r r r r
� s zExecOperation.startc K s$ | j du r| j�� o
| �� | _ | j S )z�
Returns True if the PTY should operate in raw mode.
If the exec was not started with tty=True, this will return False.
N)r0 r3 rH �is_process_ttyr r r r r! � s
zExecOperation.israwc C s6 | j j| jd| jd�}t�|�}| �� r|S t�|�S )zL
Return a single socket which is processing all I/O to exec
T)rP rW )r. �
exec_startrZ r1 r> r@ r] rN )r rP rJ r r r r' s
zExecOperation.socketsc K rS )z1
resize pty of an execed process
rT N)r. � exec_resizerZ r$ r r r r rU zExecOperation.resizec C s | � � d d S )z9
does execed process have allocated tty?
�
ProcessConfigrW )�
_exec_infor r r r r] s z ExecOperation.is_process_ttyc C s | j du r
| j�| j�| _ | j S )z<
Caching wrapper around client.exec_inspect
N)r\ r. � exec_inspectrZ r r r r ra s
zExecOperation._exec_info)TNNNr ) r r r r r r
r! r' r r] ra r r r r r[ � s
r[ c @ s< e Zd ZdZdd� Zdd� Zd
dd�Zd
d d
�Zd d � ZdS )�PseudoTerminala!
Wraps the pseudo-TTY (PTY) allocated to a docker container.
The PTY is managed via the current process' TTY until it is closed.
Example:
import docker
from dockerpty import PseudoTerminal
client = docker.Client()
container = client.create_container(
image='busybox:latest',
stdin_open=True,
tty=True,
command='/bin/sh',
)
# hijacks the current tty until the pty is closed
PseudoTerminal(client, container).start()
Care is taken to ensure all file descriptors are restored on exit. For
example, you can attach to a running container from within a Python REPL
and when the container exits, the user will be returned to the Python REPL
without adverse effects.
c C s || _ || _dS )r) N)r. � operation)r r. rd r r r r < s
zPseudoTerminal.__init__c C s
| j �� S r )rd r' r r r r r' D s
zPseudoTerminal.socketsNc C s� | j j|d�}dd� |D �}z0t| ��
| �|� W d � n1 s#w Y W |r;t||�D ] \}}t�||� q0d S d S |rQt||�D ] \}}t�||� qFw w )N)r' c S s g | ]}|� d ��qS )F)� set_blocking��.0�pr r r �
<listcomp>J � z(PseudoTerminal.start.<locals>.<listcomp>)rd r
r � _hijack_tty�zipr> re )r r' rE �flags�pump�flagr r r r
G s"
�����zPseudoTerminal.startc C s` | j �� sdS |pt�| j j�}|dur.|\}}z | j j||d� W dS ty- Y dS w dS )z�
Resize the container's PTY.
If `size` is not None, it must be a tuple of (height,width), otherwise
it will be determined by the size of the current TTY.
NrT )rd r! rW �sizer3 r �IOError)r rp �rows�colsr r r r T s
��zPseudoTerminal.resizec C s� t j| jj| j�� d��` | �� dd� |D �}dd� |D �}tj||dd�\}}z |D ]}|�� q-|D ]}|� � q6t
dd� |D ��rHW n W n t yb } z
d |j vrX|�W Y d }~nd }~ww qW d � d S 1 sow Y d S )
N)r0 Tc S s g | ]}|j s|�qS r )�eofrf r r r ri l rj z.PseudoTerminal._hijack_tty.<locals>.<listcomp>c S s g | ]
}|j �� r|j �qS r )� to_stream� needs_writerf r r r ri m s �< )�timeoutc S s g | ]}|� � �qS r )�is_donerf r r r ri w s z The operation did not complete)
rW �Terminalrd r5 r! r r> �select�do_write�flush�allr �strerror) r rE �
read_pumps�
write_streams�
read_ready� write_ready� write_streamrn �er r r rk h s.
�
����
"�zPseudoTerminal._hijack_ttyr ) r r r r r r' r
r rk r r r r rc s
rc )T)r2 r r+ �sslr � dockerpty.ior> �
dockerpty.ttyrW �objectr r r( rX r[ rc r r r r �<module> s 6
fL