File: //usr/lib/python3/dist-packages/mercurial/utils/__pycache__/procutil.cpython-310.pyc
o
�]Lb�l � @ s2 d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
ddl m Z ddl
mZmZ ddlmZmZmZm
Z
ddlmZ e�d�Ze
jrYdd lmZ ndd
lmZ d d � Z G d
d� dej �Z G dd� de �Z ej!�"e � dd� Z#dd� Z$G dd� de �Z%ej&�"e%� dd� Z'e
j(r�ej)du r�e � Z)nej)j*Z)ej+du r�e � Z+ne'ej+j*�Z+ej,du r�e � Z,ne'ej,j*�Z,e
jr�e�-e+�Z+e�-e,�Z,e e+�r�e#e+�Z+n,ej)Z)ej+Z+ej,Z,e
jr�e�-e+�Z+e�-e,�Z,e e+��re
jr�e#e+�Z+n e�.e+�/� dd�Z+ej0Z0ej1Z2ej3Z3ej4Z4ej5Z5ej6Z6ej7Z7ej8Z8ej9Z9ej:Z:ej;Z;ej<Z<ej=Z=zej>Z>W n
e?�y: Y nw zej@Z@W n
e?�yJ Y nw e
jAZBdd � ZCG d d � d e �ZDd^d!d"�ZEd#d$� ZFd%d&� ZGd_d'd(�ZHd_d)d*�ZId`d+d,�ZJd-d.� ZKd/d0� ZLeLeKd1�ZMd2d3� ZNdaOd4d5� ZPd6d7� ZQd8d9� ZRd:d;� ZSd<d=� ZTd>d?� ZUd@dA� ZVd_dBdC�ZWe
j�r�dDdE� ZXejYZZndFdE� ZXe
j[ZZdGdH� Z\dadIdJ�Z]da^dKdL� Z_dMdN� Z`dOdP� ZadQdR� ZbejcdSdT� �Zde
j�r�dUZeeeejfB Zg V W dbdXdY�ZhdS V W dbdZd[�Zi V W dbd\d]�Zje
j(�reiZhdS ejZhdS )c� )�absolute_importN� ��_)�getattr�open)�encoding�error�policy�pycompat)� resourceutil�osutil)�windows)�posixc C s z| � � W S ty Y dS w )NF)�isatty�AttributeError)�fp� r �:/usr/lib/python3/dist-packages/mercurial/utils/procutil.pyr . s
�r c @ s e Zd ZdZdd� Zdd� ZdS )�BadFilez3Dummy file object to simulate closed stdio behaviorc C � t tjd���NzBad file descriptor��IOError�errno�EBADF��self�br r r �readinto8 � zBadFile.readintoc C r r r r r r r �write; r z
BadFile.writeN)�__name__�
__module__� __qualname__�__doc__r r! r r r r r 5 s r c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�LineBufferedWrapperc C �
|| _ d S �N��orig�r r+ r r r �__init__@ �
z LineBufferedWrapper.__init__c C � t | j|�S r) �r r+ �r �attrr r r � __getattr__C r z LineBufferedWrapper.__getattr__c C s& | j }|�|�}|�d�r|�� |S )N�
)r+ r! �endswith�flush)r �sr+ �resr r r r! F s
zLineBufferedWrapper.writeN�r" r# r$ r- r3 r! r r r r r' ? � r' c C s, t jr t| tj�s | S t| t�r| S t| �S r) )r �ispy3�
isinstance�io�BufferedIOBaser' ��streamr r r �make_line_bufferedS s
rA c C s$ t | t�rt | jt�r
J �| jS | S r) )r< r' r+ r? r r r �unwrap_line_buffered_ s
rB c @ r& )�WriteAllWrapperc C r( r) r* r, r r r r- g r. zWriteAllWrapper.__init__c C r/ r) r0 r1 r r r r3 j r zWriteAllWrapper.__getattr__c C sD | j j}t|�}t|�}d}||k r ||||d � �7 }||k s|S �Nr )r+ r! �
memoryview�len)r r7 �write1�m�total_to_write�
total_writtenr r r r! m s �zWriteAllWrapper.writeNr9 r r r r rC f r: rC c C s0 t jsJ �t| t�r | S t| tj�r| S t| �S r) )r r; r< rC r= r> r? r r r �_make_write_all| s
rK �wb� c C s" | dkr
t d�| S t d�| S )zpreturn a message describing a subprocess status
(codes from kill are negative - not os.system/wait encoding)r s exited with status %ds killed by signal %dr )�coder r r � explainexit� s rO c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�_pfilez;File-like wrapper for a stream opened by subprocess.Popen()c C s || _ || _d S r) )�_proc�_fp)r �procr r r r r- � s
z_pfile.__init__c C s | j �� | j�� S r) )rR �closerQ �wait�r r r r rT � s
z _pfile.closec C s
t | j�S r) )�iterrR rV r r r �__iter__� r. z_pfile.__iter__c C r/ r) )r rR r1 r r r r3 � r z_pfile.__getattr__c C � | S r) r rV r r r � __enter__� � z_pfile.__enter__c C s | � � d S r) )rT )r �exc_type� exc_value�exc_tbr r r �__exit__� r z_pfile.__exit__N)
r" r# r$ r% r- rT rX r3 rZ r_ r r r r rP � s rP � rb���c C s2 |dkr t | |�S |dkrt| |�S t�d| ��)Nr` s wbs unsupported mode: %r)� _popenreader� _popenwriterr �ProgrammingError)�cmd�mode�bufsizer r r �popen� s
rh c C �& t jt| �d|tt jd�}t||j�S )NT)�shellrg � close_fds�stdout)�
subprocess�Popen� tonativestr�closefds�PIPErP rl �re rg �pr r r rb � � � rb c C ri )NT)rj rg rk �stdin)rm rn ro rp rq rP ru rr r r r rc rt rc c C s0 t jt| �ddtt jt jt|�d�}|j|jfS )NTra )rj rg rk ru rl �env)rm rn ro rp rq � tonativeenvru rl )re rv rs r r r �popen2 s � rx c C s t | |�\}}}}|||fS r) )�popen4)re rv ru rl �stderrrs r r r �popen3! s
r{ c
C s: t jt| �d|tt jt jt jt|�d�}|j|j|j|fS )NT)rj rg rk ru rl rz rv ) rm rn ro rp rq rw ru rl rz )re rv rg rs r r r ry & s �
ry c C s. t jt|�dtt jt jd�}|�| �\}}|S )z9filter string S through command CMD, returning its outputT)rj rk ru rl )rm rn ro rp rq � communicate)r7 re rs �pout�perrr r r �
pipefilter4 s �r c C s� d\}}z�t jdd�\}}t�|d�}|�| � |�� t jdd�\}}t�|� |�d|�}|�d|�}t|�}t jdkrD|d @ rDd
}|rSt �
t d �|t |�f ��t
|d ��4}|�� W d
� W z |rlt�|� W n tyv Y nw z |r�t�|� W S W S ty� Y S w 1 s�w Y W z |r�t�|� W n ty� Y nw z
|r�t�|� W d
S W d
S ty� Y d
S w z |r�t�|� W n ty� Y nw z |r�t�|� W w W w ty� Y w w )z�filter string S through a pair of temporary files with CMD.
CMD is used as a template to create the real command to be run,
with the strings INFILE and OUTFILE replaced by the real names of
the temporary files generated.)NNs
hg-filter-in-)�prefixrL s hg-filter-out-s INFILEs OUTFILE� OpenVMSrM r s command '%s' failed: %sr` N)r �mkstemp�os�fdopenr! rT �replace�system� sysplatformr �Abortr rO r �read�unlink�OSError)r7 re �inname�outname�infdr �outfdrN r r r �
tempfilterA s~
� �
� �� ��
� �� ��
� �� �r� )s tempfile:s pipe:c C sF t �t�D ]\}}|�|�r || |t|�d� �� � S qt| |�S )zMfilter a string through a command that transforms its input to its
outputN)r � iteritems� _filtertable�
startswithrF �lstripr )r7 re �name�fnr r r �filterl s
�
r� c C s� t du r[tj�d�} tjd }| rt| � t S t�� r3t tdd�dkr,ttjd � t S tt
j � t S t
j sLt
j�t |dd��d krLtt
�|j�� t S ttd
�pYt
j�t
jd �� t S ) z]return location of the 'hg' executable.
Defaults to $HG or 'hg' in the search path.
N� HG�__main__�frozen�
macosx_app� EXECUTABLEPATH�__file__� �hgs hgr )�
_hgexecutabler �environ�get�sys�modules�_sethgexecutabler �
mainfrozenr r �
sysexecutable� iswindowsr� �path�basename�fsencoder� �findexe�sysargv)r� �mainmodr r r � hgexecutablex s*
�
�
����r� c C s | a dS )z#set location of the 'hg' executableN)r� )r� r r r r� � s r� c C s8 t | dd �}z
|o|� |�� kW S tjy Y dS w )N�filenoF)r r� r= �UnsupportedOperation)�f�stdfr� r r r � _testfileno� s �r� c C � t | tj�S r) )r� r� � __stdin__�r� r r r �isstdin� r r� c C r� r) )r� r� �
__stdout__r� r r r �isstdout� r r� c C s� |� � | |}}t| t�r0t�| �� �}t�tjtj�}t� || �� � t�
|� t� |d�}t|t �rLt�|�� �}t� t
�� |�� � t� |d�}||fS )a� Duplicate streams and redirect original if (uin, uout) are stdio
If uin is stdin, it's redirected to /dev/null. If uout is stdout, it's
redirected to stderr so the output is still readable.
Returns (fin, fout) which point to the original (uin, uout) fds, but
may be copy of (uin, uout). The returned streams can be considered
"owned" in that print(), exec(), etc. never reach to them.
�rbrL )r6 r� ru r� �dupr� r �devnull�O_RDONLY�dup2rT r� rl rz )�uin�uout�fin�fout�newfd�nullfdr r r � protectstdio� s
r� c C sJ |� � || f||ffD ]\}}||ur"t�|�� |�� � |�� q dS )z@Restore (uin, uout) streams from possibly duplicated (fin, fout)N)r6 r� r� r� rT )r� r� r� r� r� �uifr r r � restorestdio� s ��r� c sB dd� � t tj�}| r|�� fdd�t�| �D �� t� |d< |S )z>return environ with optional override, useful for shelling outc S s* | du s| du r
dS | du rdS t �| �S )z9convert python object into string that is useful to shellNF� 0T� 1)r �bytestr)�valr r r �py2shell� s
z shellenviron.<locals>.py2shellc 3 s � | ] \}}|� |�fV qd S r) r )�.0�k�v�r� r r � <genexpr>� s � z shellenviron.<locals>.<genexpr>r� )�dictr r� �updater r� r� )r� rv r r� r � shellenviron� s
r� c C s t �| t|��S r) )�platform�
shelltocmdexer� �re rv r r r �
shelltonative� s �r� c C rY r) r r� r r r r� � r[ c C s t �t| �S )zHconvert the environment from bytes to strings suitable for Popen(), etc.)r �rapplyro )rv r r r rw � s rw c C s� zt �� W n ty Y nw t|�}|du s t|�r/tjt| �dtt |�t
� t|�d�}n,tj t| �dtt |�t
� t|�tj
tjd�}t|j jd�D ]}|�|� qL|�� |j}t
jdkrf|d@ rfd}|S ) z�enhanced shell command execution.
run with environment maybe modified, maybe in different dir.
if out is specified, it is assumed to be a file-like object that has a
write() method. stdout and stderr will be redirected to out.NT)rj rk rv �cwd)rj rk rv r� rl rz � r� rM r )rl r6 � Exceptionr� r� rm �callro rp rw r r� rn rq �STDOUTrW �readliner! rU �
returncoder� )re r� r� �outrv �rcrS �liner r r r� � s< �
�
� r� c C sJ t jrdtjv r
dS ttdd�rt�� S dS t jp$tj�d�p$tj�d�S )zAre we running in a GUI?s SSH_CONNECTIONF�isguiNTs DISPLAYs WAYLAND_DISPLAY) r �isdarwinr r� r r
r� r� r� r r r r �_gui s
�
�r� c C s t d u rt� a t S r) )�_is_guir� r r r r �gui0 s r� c C s2 t �� rttdd�dkrtjd gS tjgS t� S )z�Return the command used to execute current hg
This is different from hgexecutable() because on Windows we want
to avoid things opening new shell windows like batch files, so we
get either the python call or current executable.
r� Nr� r� ) r r� r r� r r� r r� � _gethgcmdr r r r �hgcmd7 s
r� c
s� t � � � fdd�}d}ttdd�}|durt�||�}z9t| �}|� sF|� v s+t|�s>|� s>W |dur<t�tj|� dS dS t�d� |� r#|W |durTt�tj|� S S |durat�tj|� w w )a_ Execute the argument list in a detached process.
condfn is a callable which is called repeatedly and should return
True once the child process is known to have started successfully.
At this point, the child process PID is returned. If the child
process fails to start or finishes before condfn() evaluates to
True, return -1.
c s � � t�� � d S r) )�addr� rU )�signum�frame��
terminatedr r �handlerY s z rundetached.<locals>.handlerN�SIGCHLDra g�������?)�setr �signal�
spawndetached�testpidr� �time�sleep)�args�condfnr� � prevhandlerr� �pidr r� r � rundetachedG s, �
��
��r� c # s� � t �t j�g� g �� ��fdd�}z+z t � t j|� W n ty( � d= Y nw dV W � r8t � t j� d � �r<t�dS � rIt � t j� d � �rMt�w )a Inhibit SIGINT handling on a region of code.
Note that if this is called in a non-main thread, it turns into a no-op.
Args:
warn: A callable which takes no arguments, and returns True if the
previous signal handling should be restored.
c s, �� rt � t j� d � � d= ��d� d S )Nr T)r� �SIGINT�append)r� ��oldsiginthandler�
shouldbail�warnr r �disabledsiginthandlerz s z.uninterruptible.<locals>.disabledsiginthandlerr N)r� � getsignalr� �
ValueError�KeyboardInterrupt)r r r r r �uninterruptiblel s( �
���r � FTc
C s� zAd}|durt �� }|�|� |�� |�d� tjt �t| �|t |�dt
|||d�} |dur5|| j � W |dur@|� � dS dS |durK|� � w w )z1Spawn a command without waiting for it to finish.Nr T)rj rv rk �
creationflagsru rl rz )
r �unnamedtempfiler! r6 �seekrm rn r� ro rw �_creationflagsrU rT )
�scriptrv rj rl rz � ensurestart� record_wait� stdin_bytesru rs r r r � runbgcommand� s2
�
� �
�r c
s� t jrt� |r | }nt| t�r| g} d�dd� | D ��}|du r(d| }d} nd} d}d}
z]z7|du r8tj}
nt �� }
|
� |� |
�
� |
� d� |du rQtj}|du rXtj}tj |d|d|
||| d �� W n t
yv |duru|d
� � w W |dur�|
dur�t|
t�r�J �|
�� n|dur�|
dur�t|
t�r�J �|
�� w w w |s�tj� fd d �d
�} d| _| �� dS � j} |dur�|| � dS dS )� Spawn a command without waiting for it to finish.
When `record_wait` is not None, the spawned process will not be fully
detached and the `record_wait` argument will be called with a the
`Subprocess.wait` function for the spawned process. This is mostly
useful for developers that need to make sure the spawned process
finished before a certain point. (eg: writing test)� c s s � | ]}t |�V qd S r) )�
shellquote)r� �xr r r r� � s � z"runbgcommandpy3.<locals>.<genexpr>Ns ( %s ) &TFr )rj rv rk ru rl rz �start_new_session� c s � j S r) )rU r �rs r r �<lambda> s z!runbgcommandpy3.<locals>.<lambda>��target)r r� r� r< �bytes�joinrm �DEVNULLr r! r6 r rn r� �intrT � threading�Thread�daemon�startrU )
re rv rj rl rz r r r r r ru �tr� r r r �runbgcommandpy3� sp
�
����
� �r& c
s t jrt� |du rYt�� � � rY|s'� fdd�}tj|d�} d| _| �� dS t� � d�\}
} t�
| �r:t� | �} nt� | � } | dkrW| dkrKt
j} t| d| t�| �f ��dS d} d}
z�zR|du rgt�� |du rrttjd �}
nt �� }
|
�|� |
�� |
�d� |du r�ttjd
�}|du r�ttjd
�}tj| ||d|
||d �}|dur�||j� d} W n' ty� } z|j
d@ } | dkr�d} W Y d}~nd}~w t y� d} Y nw W |
dur�|
� � |du r�t� | � dS dS |
dur�|
� � |du �rt� | � w w ) r Nc s t �� d� d S rD )r� �waitpidr �r� r r �_do_wait: s z!runbgcommandpy2.<locals>._do_waitr Tr r s error running %r: %s� r� w)rj rv rk ru rl rz ) r r� r� r� �forkr! r" r# r$ r' � WIFEXITED� WEXITSTATUS�WTERMSIGr �EINVALr� �strerror�setsidr r� r r! r6 r rm rn rU �EnvironmentErrorr� rT �_exit)re rv rj rl rz r r r r) r% �_pid�statusr� ru rs �exr r( r �runbgcommandpy2 s�
��
�
� ����
�r8 )r` ra r) )Nra )NNN)FNNTNN)k�
__future__r �
contextlibr r= r� r� rm r� r! r� �i18nr r r r r� r r r
�utilsr � importmodr
r� r r� r r � RawIOBaser �objectr' r> �registerrA rB rC �IOBaserK r; ru �bufferrl rz � winstdoutr� r� r� �gethgcmdr� �getuser�getpid�
hidewindow�readpipe� setbinary�setsignalhandlerr �
shellsplitr� �sshargsr� � setprocnamer �
unblocksignal�isposixrp rO rP rh rb rc rx r{ ry r r� r� r� r� r� r� r� r� r� r� r� r� r� � strfromlocalro �identityrw r� r� r� r� r� r� �contextmanagerr �DETACHED_PROCESS�CREATE_NEW_PROCESS_GROUPr
r r& r8 r r r r �<module> s
�
�
�
&�
&%
!���,
�W
�r