File: //opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/__pycache__/nb_popen.cpython-310.pyc
o
�N�g� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZej dkZ
e
r?ddl Z ddl Z ddl
mZmZ ddlmZ e�e�ZG dd� dej�ZdS )a
:codeauthor: Pedro Algarvio (pedro@algarvio.me)
salt.utils.nb_popen
~~~~~~~~~~~~~~~~~~~
Non blocking subprocess Popen.
This functionality has been adapted to work on windows following the recipe
found on:
http://code.activestate.com/recipes/440554/
� N�win32)�ReadFile� WriteFile)�
PeekNamedPipec s� e Zd ZdZdZ� fdd�Zd dd�Zd dd �Zd d d �Zd
d� Z dd� Z
e r0dd� Z dd� Z
ndd� Z dd� Z
d dd�Zd � fdd� Z� ZS ) �NonBlockingPopenz$salt.utils.nb_popen.STDOUT.PID-{pid}z$salt.utils.nb_popen.STDERR.PID-{pid}c s� |� dd�| _|� dd�| _|� d| j�| _|� d| j�| _|� dd �}|�dd �}t� j|i |�� t� | j�| _
t � | jj
| jd ��| _|tju rT| j
| _| j| _nt� | j�| _t � | jj
| jd ��| _t�d
| j|d u ru|� d S |� d S ) N� stream_stdsF�max_size_in_memi � Zstdout_logger_nameZstderr_logger_name�logging_command�stderr)�pidz"Running command under pid %s: '%s')�popr r �_stdout_logger_name_�_stderr_logger_name_�get�super�__init__�tempfileZSpooledTemporaryFile� stdout_buff�logging� getLogger�formatr Z_stdout_logger�
subprocess�STDOUT� stderr_buffZ_stderr_logger�log�info)�self�args�kwargsr r
�� __class__� �G/opt/saltstack/salt/lib/python3.10/site-packages/salt/utils/nb_popen.pyr + s: �� �
�
��zNonBlockingPopen.__init__Nc C � | � d|�S )N�stdout��_recv�r �maxsizer! r! r" �recvZ � zNonBlockingPopen.recvc C r# )Nr
r% r' r! r! r" �recv_err] r* zNonBlockingPopen.recv_err� c C s | � |�| �|�| �|�fS �N)�sendr) r+ )r �inputr( r! r! r" � send_recv` s zNonBlockingPopen.send_recvc C s( |d u rd}n|dk r
d}t | |�|fS )Ni � )�getattr)r �whichr( r! r! r" �get_conn_maxsizec s
z!NonBlockingPopen.get_conn_maxsizec C s t | |��� t| |d � d S r- )r2 �close�setattr)r r3 r! r! r" �_closej s zNonBlockingPopen._closec
C s� | j sd S zt�| j �� �}t||�\}}W |S ty% | �d� Y S tjt fyI } z|j
d dt j fv rD| �d�W Y d }~S � d }~ww )N�stdinr �m )
r8 �msvcrt�
get_osfhandle�filenor �
ValueErrorr7 �
pywintypes�error� Exceptionr �errno� ESHUTDOWN)r r/ �x�errCode�written�whyr! r! r" r. p s ���zNonBlockingPopen.sendc
C s$ | � ||�\}}|d u rd S z#t�|�� �}t|d�\}}}||k r$|}|dkr0t||d �\}}W n2 ty? | �|� Y S tj t
fyc } z| j d dt j
fv r^| �|�W Y d } ~ S � d } ~ ww t| |� d���|� t| d|� d���|�� � | jr�tt|��|� | jr�| �|�}|S )Nr r9 �_buff�_�_logger)r4 r: r; r< r r r= r7 r> r? r@ r rA rB r2 �write�debug�rstripr �sys�universal_newlines�_translate_newlines)
r r3 r( �connrC �readZnAvailZnMessagerD rF r! r! r" r& � s4 � ��
zNonBlockingPopen._recvc
C s� | j sd S t�g | j gg d�d sdS z t�| j �� |�}W |S ty? } z|jd tjkr:| � d�W Y d }~S � d }~ww )Nr r1 r8 )
r8 �select�osrJ r< �OSErrorr rA ZEPIPEr7 )r r/ rE rF r! r! r" r. � s ���c C sH | � ||�\}}|d u rd S dd l}|�||j�}|js'|�||j|tjB � zot�|gg g d�d sCW |jsA|�||j|� dS dS |�|�}|s\| � |�W |js[|�||j|� S S | j
rd| � |�}t | |� d���
|� t | d|� d���|�� � | jr�t t|��
|� |W |js�|�||j|� S S |js�|�||j|� w w )Nr r, rG rH rI )r4 �fcntl�F_GETFL�closed�F_SETFLrS �
O_NONBLOCKrR rQ r7 rN rO r2 rJ rK rL r rM )r r3 r( rP rU �flagsZbuffr! r! r" r& � s> �
�
��
���{�G�z�?c C s� d}| � � d u rI| jd urd}| �� | jd ur d}| �� |d7 }|dkr<d}| �� \}}|r5t�|� |r<t�|� t �
|� | � � d u sd S d S )Nr r1 �d ) �pollr$ r) r
r+ � communicater rK r? �time�sleep)r �intervalZsilent_iterationsZ
stdoutdataZ
stderrdatar! r! r" �poll_and_read_until_finish� s$
�z+NonBlockingPopen.poll_and_read_until_finishc sL t � �|� | j�� | j�d� | j�� | j�d� | j�� | j�� fS )Nr )r r^ r �flush�seekr rQ )r r/ r r! r" r^ � s
z NonBlockingPopen.communicater- )r, N)r[ )�__name__�
__module__� __qualname__r
r r r) r+ r0 r4 r7 � mswindowsr. r& rb r^ �
__classcell__r! r! r r" r % s
/
r )�__doc__rA r rS rR r rM r r_ �platformrh r: r> Z win32filer r Z win32piper r re r �Popenr r! r! r! r" �<module> s"