← Back
Editing: unix_events.cpython-37.pyc
B � f؟ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z ddl mZ ddl mZ ddl mZ ddl mZ dd l mZ dd l mZ dd l mZ dd l mZ dd lmZ dZe jdkr�ed��dd� ZG dd� dej �Z G dd� dej �Z G dd� dej ej!�Z"G dd� dej#�Z$G dd � d �Z%G d d � d e%�Z&G d d � d e&�Z'G d!d"� d"e&�Z(G d#d$� d$ej)�Z*e Z+e*Z,dS )%z2Selector event loop for Unix with signal handling.� N� )� base_events)�base_subprocess)� constants)� coroutines)�events)�futures)�selector_events)�tasks)� transports)�logger)�SelectorEventLoop�AbstractChildWatcher�SafeChildWatcher�FastChildWatcher�DefaultEventLoopPolicyZwin32z+Signals are not really supported on Windowsc C s dS )zDummy signal handler.N� )�signum�framer r �8/opt/alt/python37/lib64/python3.7/asyncio/unix_events.py�_sighandler_noop' s r c s� e Zd ZdZd)� fdd� Z� fdd�Zdd� Zd d � Zd d � Zd d� Z dd� Z d*dd�Z d+dd�Z d,dd�Z dd� Zd-ddddd�dd�Zd.dd ddd d �d d �Zd!d"� Zd#d$� Zd%d&� Zd'd(� Z� ZS )/�_UnixSelectorEventLoopzdUnix event loop. Adds signal handling and UNIX Domain Socket support to SelectorEventLoop. Nc s t � �|� i | _d S )N)�super�__init__�_signal_handlers)�self�selector)� __class__r r r 2 s z _UnixSelectorEventLoop.__init__c s^ t � �� t�� s2xFt| j�D ]}| �|� q W n(| jrZtjd| �d�t | d� | j� � d S )NzClosing the loop z@ on interpreter shutdown stage, skipping signal handlers removal)�source) r �close�sys� is_finalizing�listr �remove_signal_handler�warnings�warn�ResourceWarning�clear)r �sig)r r r r 6 s z _UnixSelectorEventLoop.closec C s" x |D ]}|sq| � |� qW d S )N)�_handle_signal)r �datar r r r �_process_self_dataD s z)_UnixSelectorEventLoop._process_self_datac G sL t �|�st �|�r td��| �|� | �� yt�| j� � � W n2 t t fk rt } zt t |���W dd}~X Y nX t�||| d�}|| j|< y t�|t� t�|d� W n� t k �rF } zz| j|= | j�syt�d� W n4 t t fk �r } zt�d|� W dd}~X Y nX |jtjk�r4t d|� d���n� W dd}~X Y nX dS )z�Add a handler for a signal. UNIX only. Raise ValueError if the signal number is invalid or uncatchable. Raise RuntimeError if there is a problem setting up the handler. z3coroutines cannot be used with add_signal_handler()NF���z set_wakeup_fd(-1) failed: %szsig z cannot be caught)r Z iscoroutineZiscoroutinefunction� TypeError� _check_signalZ _check_closed�signal� set_wakeup_fdZ_csock�fileno� ValueError�OSError� RuntimeError�strr ZHandler r � siginterruptr �info�errno�EINVAL)r r( �callback�args�exc�handleZnexcr r r �add_signal_handlerK s0 z)_UnixSelectorEventLoop.add_signal_handlerc C s8 | j �|�}|dkrdS |jr*| �|� n | �|� dS )z2Internal helper that is the actual signal handler.N)r �getZ _cancelledr# Z_add_callback_signalsafe)r r( r= r r r r) x s z%_UnixSelectorEventLoop._handle_signalc C s� | � |� y | j|= W n tk r* dS X |tjkr>tj}ntj}yt�||� W nB tk r� } z$|jtj kr�t d|� d���n� W dd}~X Y nX | js�yt� d� W n2 t tfk r� } zt �d|� W dd}~X Y nX dS )zwRemove a handler for a signal. UNIX only. Return True if a signal handler was removed, False if not. Fzsig z cannot be caughtNr, z set_wakeup_fd(-1) failed: %sT)r. r �KeyErrorr/ �SIGINT�default_int_handler�SIG_DFLr3 r8 r9 r4 r0 r2 r r7 )r r( Zhandlerr< r r r r# � s( z,_UnixSelectorEventLoop.remove_signal_handlerc C sL t |t�std|����d| kr.tjk sHn td|� dtj� d���dS )z�Internal helper to validate a signal. Raise ValueError if the signal number is invalid or uncatchable. Raise RuntimeError if there is a problem setting up the handler. zsig must be an int, not r zsig z out of range(1, �)N)� isinstance�intr- r/ �NSIGr2 )r r( r r r r. � s z$_UnixSelectorEventLoop._check_signalc C s t | ||||�S )N)�_UnixReadPipeTransport)r �pipe�protocol�waiter�extrar r r �_make_read_pipe_transport� s z0_UnixSelectorEventLoop._make_read_pipe_transportc C s t | ||||�S )N)�_UnixWritePipeTransport)r rI rJ rK rL r r r �_make_write_pipe_transport� s z1_UnixSelectorEventLoop._make_write_pipe_transportc � s� t �� ��} | �� } t| |||||||f| |d�| ��} | �| �� | j| � y| I d H W n, tk r� | �� | � � I d H � Y nX W d Q R X | S )N)rK rL ) r �get_child_watcher� create_future�_UnixSubprocessTransport�add_child_handlerZget_pid�_child_watcher_callback� Exceptionr Z_wait) r rJ r; �shell�stdin�stdout�stderr�bufsizerL �kwargs�watcherrK �transpr r r �_make_subprocess_transport� s z1_UnixSelectorEventLoop._make_subprocess_transportc C s | � |j|� d S )N)Zcall_soon_threadsafeZ_process_exited)r �pid� returncoder] r r r rT � s z._UnixSelectorEventLoop._child_watcher_callback)�ssl�sock�server_hostname�ssl_handshake_timeoutc � s |d kst |t�st�|r,|d krLtd��n |d k r<td��|d k rLtd��|d k r�|d k rdtd��t�|�}t�tjtjd�}y |� d� | � ||�I d H W q� |� � � Y q�X n@|d kr�td��|j tjks�|j tjkr�td|����|� d� | j|||||d �I d H \}}||fS ) Nz/you have to pass server_hostname when using sslz+server_hostname is only meaningful with sslz1ssl_handshake_timeout is only meaningful with sslz3path and sock can not be specified at the same timer Fz no path and sock were specifiedz.A UNIX Domain Stream Socket was expected, got )rd )rE r5 �AssertionErrorr2 �os�fspath�socket�AF_UNIX� SOCK_STREAM� setblockingZ sock_connectr �family�typeZ _create_connection_transport) r �protocol_factory�pathra rb rc rd � transportrJ r r r �create_unix_connection� sB z-_UnixSelectorEventLoop.create_unix_connection�d T)rb �backlogra rd � start_servingc � s� t |t�rtd��|d k r&|s&td��|d k �rH|d k r@td��t�|�}t�tjtj�}|d dkr�y t � t� |�j �r�t� |� W nB t k r� Y n0 tk r� } zt�d||� W d d }~X Y nX y|�|� W nl tk �r0 } z8|�� | jtjk�r d|�d�} ttj| �d �n� W d d } ~ X Y n |�� � Y nX n<|d k�rZtd ��|jtjk�sv|jtjk�r�td |����|�d � t�| |g||||�} |�r�| �� tj d| d �I d H | S ) Nz*ssl argument must be an SSLContext or Nonez1ssl_handshake_timeout is only meaningful with sslz3path and sock can not be specified at the same timer )r � z2Unable to check or remove stale UNIX socket %r: %rzAddress z is already in usez-path was not specified, and no sock specifiedz.A UNIX Domain Stream Socket was expected, got F)�loop) rE �boolr- r2 rf rg rh ri rj �stat�S_ISSOCK�st_mode�remove�FileNotFoundErrorr3 r �errorZbindr r8 Z EADDRINUSErl rm rk r ZServerZ_start_servingr Zsleep) r rn ro rb rs ra rd rt �errr< �msgZserverr r r �create_unix_server� sZ z)_UnixSelectorEventLoop.create_unix_serverc � s� y t j W n, tk r6 } zt�d��W d d }~X Y nX y |�� }W n2 ttjfk rv } zt�d��W d d }~X Y nX yt �|�j }W n, t k r� } zt�d��W d d }~X Y nX |r�|n|} | s�dS | � � } | � | d ||||| d� | I d H S )Nz os.sendfile() is not availableznot a regular filer ) rf �sendfile�AttributeErrorr �SendfileNotAvailableErrorr1 �io�UnsupportedOperation�fstat�st_sizer3 rQ �_sock_sendfile_native_impl) r rb �file�offset�countr< r1 r~ Zfsize� blocksize�futr r r �_sock_sendfile_native> s( z,_UnixSelectorEventLoop._sock_sendfile_nativec C s |� � } |d k r| �|� |�� r4| �|||� d S |rd|| }|dkrd| �|||� |�|� d S yt�| |||�} W �n* ttfk r� |d kr�| � ||� | � | | j || ||||||� Y �nH t k �rj } z�|d k �r| j t jk�rt| �tk �rtdt j�} | | _| } |dk�rBt�d�} | �|||� |�| � n| �|||� |�| � W d d } ~ X Y n� tk �r� } z | �|||� |�| � W d d } ~ X Y njX | dk�r�| �|||� |�|� nD|| 7 }|| 7 }|d k�r�| � ||� | � | | j || ||||||� d S )Nr zsocket is not connectedzos.sendfile call failed)r1 � remove_writer� cancelled� _sock_sendfile_update_fileposZ set_resultrf r� �BlockingIOError�InterruptedError� _sock_add_cancellation_callbackZ add_writerr� r3 r8 ZENOTCONNrm �ConnectionError� __cause__r r� Z set_exceptionrU )r r� Z registered_fdrb r1 r� r� r� � total_sent�fdZsentr< �new_excr~ r r r r� U s` z1_UnixSelectorEventLoop._sock_sendfile_native_implc C s |dkrt �||t j� d S )Nr )rf �lseek�SEEK_SET)r r1 r� r� r r r r� � s z4_UnixSelectorEventLoop._sock_sendfile_update_fileposc s � �fdd�}|� |� d S )Nc s& | � � r"��� }|dkr"� �|� d S )Nr, )r� r1 r� )r� r� )r rb r r �cb� s zB_UnixSelectorEventLoop._sock_add_cancellation_callback.<locals>.cb)Zadd_done_callback)r r� rb r� r )r rb r r� � s z6_UnixSelectorEventLoop._sock_add_cancellation_callback)N)NN)NN)N)N)N)�__name__� __module__� __qualname__�__doc__r r r+ r> r) r# r. rM rO r^ rT rq r� r� r� r� r� � __classcell__r r )r r r , s4 - *?Dr c s� e Zd ZdZd � fdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z d dd�Zdd� Zd d � Z� ZS ) rH i Nc s� t � �|� || jd< || _|| _|�� | _|| _d| _d| _ t � | j�j }t �|�s�t �|�s�t �|�s�d | _d | _d | _td��t �| jd� | j�| jj| � | j�| jj| j| j� |d k r�| j�tj|d � d S )NrI Fz)Pipe transport is for pipes/sockets only.)r r �_extra�_loop�_piper1 �_fileno� _protocol�_closing�_pausedrf r� rz rx �S_ISFIFOry �S_ISCHRr2 � set_blocking� call_soon�connection_made� _add_reader� _read_readyr � _set_result_unless_cancelled)r rv rI rJ rK rL �mode)r r r r � s. z _UnixReadPipeTransport.__init__c C s� | j jg}| jd kr |�d� n| jr0|�d� |�d| j� �� t| jdd �}| jd k r�|d k r�t� || jt j �}|r�|�d� q�|�d� n | jd k r�|�d� n |�d� d� d � |��S ) N�closed�closingzfd=� _selector�polling�idle�openz<{}>� )r r� r� �appendr� r� �getattrr� r �_test_selector_event� selectorsZ EVENT_READ�format�join)r r7 r r� r r r �__repr__� s" z _UnixReadPipeTransport.__repr__c C s� yt �| j| j�}W nD ttfk r, Y n� tk rX } z| �|d� W d d }~X Y n^X |rl| j� |� nJ| j � � r�t � d| � d| _| j �| j� | j �| jj� | j �| jd � d S )Nz"Fatal read error on pipe transportz%r was closed by peerT)rf �readr� �max_sizer� r� r3 � _fatal_errorr� Z data_receivedr� � get_debugr r7 r� �_remove_readerr� Z eof_received�_call_connection_lost)r r* r< r r r r� � s z"_UnixReadPipeTransport._read_readyc C s> | j s | jrd S d| _| j�| j� | j�� r:t�d| � d S )NTz%r pauses reading)r� r� r� r� r� r� r �debug)r r r r � pause_reading� s z$_UnixReadPipeTransport.pause_readingc C sB | j s | jsd S d| _| j�| j| j� | j�� r>t�d| � d S )NFz%r resumes reading) r� r� r� r� r� r� r� r r� )r r r r �resume_reading� s z%_UnixReadPipeTransport.resume_readingc C s || _ d S )N)r� )r rJ r r r � set_protocol� s z#_UnixReadPipeTransport.set_protocolc C s | j S )N)r� )r r r r � get_protocol s z#_UnixReadPipeTransport.get_protocolc C s | j S )N)r� )r r r r � is_closing s z!_UnixReadPipeTransport.is_closingc C s | j s| �d � d S )N)r� �_close)r r r r r s z _UnixReadPipeTransport.closec C s. | j d k r*tjd| ��t| d� | j �� d S )Nzunclosed transport )r )r� r$ r% r& r )r r r r �__del__ s z _UnixReadPipeTransport.__del__� Fatal error on pipe transportc C sZ t |t�r4|jtjkr4| j�� rLtjd| |dd� n| j�||| | j d�� | � |� d S )Nz%r: %sT)�exc_info)�message� exceptionrp rJ ) rE r3 r8 ZEIOr� r� r r� �call_exception_handlerr� r� )r r< r� r r r r� s z#_UnixReadPipeTransport._fatal_errorc C s( d| _ | j�| j� | j�| j|� d S )NT)r� r� r� r� r� r� )r r< r r r r� s z _UnixReadPipeTransport._closec C s4 z| j �|� W d | j�� d | _d | _ d | _X d S )N)r� �connection_lostr� r r� )r r< r r r r� % s z,_UnixReadPipeTransport._call_connection_lost)NN)r� )r� r� r� r� r r� r� r� r� r� r� r� r r� r� r� r� r� r r )r r rH � s rH c s� e Zd Zd%� fdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zd d � Zd&d d �Zd'd!d"�Zd#d$� Z� ZS )(rN Nc s� t � �||� || jd< || _|�� | _|| _t� | _d| _ d| _ t � | j�j }t�|�}t�|�}t�|�} |s�|s�| s�d | _d | _d | _td��t �| jd� | j�| jj| � | s�|r�tj�d�s�| j�| jj| j| j� |d k r�| j�t j |d � d S )NrI r Fz?Pipe transport is only for pipes, sockets and character devicesZaix) r r r� r� r1 r� r� � bytearray�_buffer� _conn_lostr� rf r� rz rx r� r� ry r2 r� r� r� r� r �platform� startswithr� r� r r� ) r rv rI rJ rK rL r� Zis_charZis_fifoZ is_socket)r r r r 2 s2 z _UnixWritePipeTransport.__init__c C s� | j jg}| jd kr |�d� n| jr0|�d� |�d| j� �� t| jdd �}| jd k r�|d k r�t� || jt j �}|r�|�d� n |�d� | � � }|�d|� �� n | jd k r�|�d� n |�d� d � d �|��S ) Nr� r� zfd=r� r� r� zbufsize=r� z<{}>r� )r r� r� r� r� r� r� r� r r� r� Z EVENT_WRITE�get_write_buffer_sizer� r� )r r7 r r� rZ r r r r� W s& z _UnixWritePipeTransport.__repr__c C s t | j�S )N)�lenr� )r r r r r� o s z-_UnixWritePipeTransport.get_write_buffer_sizec C s6 | j �� rt�d| � | jr*| �t� � n| �� d S )Nz%r was closed by peer)r� r� r r7 r� r� �BrokenPipeError)r r r r r� r s z#_UnixWritePipeTransport._read_readyc C s2 t |tttf�s tt|���t |t�r.t|�}|s6d S | jsB| jrj| jtj krXt � d� | jd7 _d S | j �syt �| j|�}W nV ttfk r� d}Y n< tk r� } z | jd7 _| �|d� d S d }~X Y nX |t|�kr�d S |dk�rt|�|d � }| j�| j| j� | j |7 _ | �� d S )Nz=pipe closed by peer or os.write(pipe, data) raised exception.r r z#Fatal write error on pipe transport)rE �bytesr� � memoryviewre �reprr� r� r Z!LOG_THRESHOLD_FOR_CONNLOST_WRITESr �warningr� rf �writer� r� r� rU r� r� r� Z _add_writer� _write_readyZ_maybe_pause_protocol)r r* �nr< r r r r� { s4 z _UnixWritePipeTransport.writec C s� | j std��yt�| j| j �}W nj ttfk r: Y n� tk r� } z6| j �� | j d7 _ | j � | j� | � |d� W d d }~X Y nfX |t | j �kr�| j �� | j � | j� | �� | jr�| j �| j� | �d � d S |dkr�| j d |�= d S )NzData should not be emptyr z#Fatal write error on pipe transportr )r� re rf r� r� r� r� rU r' r� r� �_remove_writerr� r� Z_maybe_resume_protocolr� r� r� )r r� r< r r r r� � s( z$_UnixWritePipeTransport._write_readyc C s dS )NTr )r r r r � can_write_eof� s z%_UnixWritePipeTransport.can_write_eofc C sB | j r d S | jst�d| _ | js>| j�| j� | j�| jd � d S )NT) r� r� re r� r� r� r� r� r� )r r r r � write_eof� s z!_UnixWritePipeTransport.write_eofc C s || _ d S )N)r� )r rJ r r r r� � s z$_UnixWritePipeTransport.set_protocolc C s | j S )N)r� )r r r r r� � s z$_UnixWritePipeTransport.get_protocolc C s | j S )N)r� )r r r r r� � s z"_UnixWritePipeTransport.is_closingc C s | j d k r| js| �� d S )N)r� r� r� )r r r r r � s z _UnixWritePipeTransport.closec C s. | j d k r*tjd| ��t| d� | j �� d S )Nzunclosed transport )r )r� r$ r% r& r )r r r r r� � s z _UnixWritePipeTransport.__del__c C s | � d � d S )N)r� )r r r r �abort� s z _UnixWritePipeTransport.abort� Fatal error on pipe transportc C sN t |t�r(| j�� r@tjd| |dd� n| j�||| | jd�� | �|� d S )Nz%r: %sT)r� )r� r� rp rJ ) rE r3 r� r� r r� r� r� r� )r r< r� r r r r� � s z$_UnixWritePipeTransport._fatal_errorc C sF d| _ | jr| j�| j� | j�� | j�| j� | j�| j|� d S )NT) r� r� r� r� r� r' r� r� r� )r r< r r r r� � s z _UnixWritePipeTransport._closec C s4 z| j �|� W d | j�� d | _d | _ d | _X d S )N)r� r� r� r r� )r r< r r r r� � s z-_UnixWritePipeTransport._call_connection_lost)NN)r� )N)r� r� r� r r� r� r� r� r� r� r� r� r� r� r r� r� r� r� r� r� r r )r r rN / s" % ! rN c @ s e Zd Zdd� ZdS )rR c K s� d }|t jkrt�� \}}zPt j|f||||d|d�|��| _|d k rh|�� t|�� d|d�| j_ d }W d |d k r�|�� |�� X d S )NF)rV rW rX rY Zuniversal_newlinesrZ �wb)� buffering) � subprocess�PIPErh Z socketpair�Popen�_procr r� �detachrW ) r r; rV rW rX rY rZ r[ Zstdin_wr r r �_start� s z _UnixSubprocessTransport._startN)r� r� r� r� r r r r rR � s rR c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )r aH Abstract base class for monitoring child processes. Objects derived from this class monitor a collection of subprocesses and report their termination or interruption by a signal. New callbacks are registered with .add_child_handler(). Starting a new process must be done within a 'with' block to allow the watcher to suspend its activity until the new process if fully registered (this is needed to prevent a race condition in some implementations). Example: with watcher: proc = subprocess.Popen("sleep 1") watcher.add_child_handler(proc.pid, callback) Notes: Implementations of this class must be thread-safe. Since child watcher objects may catch the SIGCHLD signal and call waitpid(-1), there should be only one active object per process. c G s t � �dS )a Register a new child handler. Arrange for callback(pid, returncode, *args) to be called when process 'pid' terminates. Specifying another callback for the same process replaces the previous handler. Note: callback() must be thread-safe. N)�NotImplementedError)r r_ r: r; r r r rS ) s z&AbstractChildWatcher.add_child_handlerc C s t � �dS )z�Removes the handler for process 'pid'. The function returns True if the handler was successfully removed, False if there was nothing to remove.N)r� )r r_ r r r �remove_child_handler4 s z)AbstractChildWatcher.remove_child_handlerc C s t � �dS )z�Attach the watcher to an event loop. If the watcher was previously attached to an event loop, then it is first detached before attaching to the new loop. Note: loop may be None. N)r� )r rv r r r � attach_loop< s z AbstractChildWatcher.attach_loopc C s t � �dS )zlClose the watcher. This must be called to make sure that any underlying resource is freed. N)r� )r r r r r F s zAbstractChildWatcher.closec C s t � �dS )zdEnter the watcher's context and allow starting new processes This function must return selfN)r� )r r r r � __enter__M s z AbstractChildWatcher.__enter__c C s t � �dS )zExit the watcher's contextN)r� )r �a�b�cr r r �__exit__S s z AbstractChildWatcher.__exit__N) r� r� r� r� rS r� r� r r� r� r r r r r s r c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dS )�BaseChildWatcherc C s d | _ i | _d S )N)r� � _callbacks)r r r r r Z s zBaseChildWatcher.__init__c C s | � d � d S )N)r� )r r r r r ^ s zBaseChildWatcher.closec C s t � �d S )N)r� )r � expected_pidr r r � _do_waitpida s z BaseChildWatcher._do_waitpidc C s t � �d S )N)r� )r r r r �_do_waitpid_alld s z BaseChildWatcher._do_waitpid_allc C s~ |d kst |tj�st�| jd k r<|d kr<| jr<t�dt� | jd k rT| j� t j � || _|d k rz|� t j | j � | �� d S )NzCA loop is being detached from a child watcher with pending handlers)rE r ZAbstractEventLoopre r� r� r$ r% �RuntimeWarningr# r/ �SIGCHLDr> � _sig_chldr� )r rv r r r r� g s z BaseChildWatcher.attach_loopc C sF y | � � W n4 tk r@ } z| j�d|d�� W d d }~X Y nX d S )Nz$Unknown exception in SIGCHLD handler)r� r� )r� rU r� r� )r r< r r r r� { s zBaseChildWatcher._sig_chldc C s2 t �|�rt �|� S t �|�r*t �|�S |S d S )N)rf � WIFSIGNALED�WTERMSIG� WIFEXITED� WEXITSTATUS)r �statusr r r �_compute_returncode� s z$BaseChildWatcher._compute_returncodeN) r� r� r� r r r� r� r� r� r r r r r r� X s r� c sP e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z � Z S )r ad 'Safe' child watcher implementation. This implementation avoids disrupting other code spawning processes by polling explicitly each process in the SIGCHLD handler instead of calling os.waitpid(-1). This is a safe solution but it has a significant overhead when handling a big number of children (O(n) each time SIGCHLD is raised) c s | j �� t� �� d S )N)r� r' r r )r )r r r r � s zSafeChildWatcher.closec C s | S )Nr )r r r r r� � s zSafeChildWatcher.__enter__c C s d S )Nr )r r� r� r� r r r r� � s zSafeChildWatcher.__exit__c G s. | j d krtd��||f| j|< | �|� d S )NzICannot add child handler, the child watcher does not have a loop attached)r� r4 r� r� )r r_ r: r; r r r rS � s z"SafeChildWatcher.add_child_handlerc C s& y | j |= dS tk r dS X d S )NTF)r� r@ )r r_ r r r r� � s z%SafeChildWatcher.remove_child_handlerc C s" x t | j�D ]}| �|� q W d S )N)r" r� r� )r r_ r r r r� � s z SafeChildWatcher._do_waitpid_allc C s� |dks t �yt�|tj�\}}W n( tk rJ |}d}t�d|� Y n0X |dkrXd S | �|�}| j� � rzt� d||� y| j � |�\}}W n. t k r� | j� � r�tjd|dd� Y nX |||f|�� d S )Nr � z8Unknown child process pid %d, will report returncode 255z$process %s exited with returncode %sz'Child watcher got an unexpected pid: %rT)r� )re rf �waitpid�WNOHANG�ChildProcessErrorr r� r r� r� r� r� �popr@ )r r� r_ r r` r: r; r r r r� � s, z SafeChildWatcher._do_waitpid) r� r� r� r� r r� r� rS r� r� r� r� r r )r r r � s r c sT e Zd ZdZ� fdd�Z� fdd�Zdd� Zdd � Zd d � Zd d � Z dd� Z � Z S )r aW 'Fast' child watcher implementation. This implementation reaps every terminated processes by calling os.waitpid(-1) directly, possibly breaking other code spawning processes and waiting for their termination. There is no noticeable overhead when handling a big number of children (O(1) each time a child terminates). c s$ t � �� t�� | _i | _d| _d S )Nr )r r � threadingZLock�_lock�_zombies�_forks)r )r r r r � s zFastChildWatcher.__init__c s" | j �� | j�� t� �� d S )N)r� r' r r r )r )r r r r � s zFastChildWatcher.closec C s$ | j � | jd7 _| S Q R X d S )Nr )r r )r r r r r� � s zFastChildWatcher.__enter__c C sT | j �8 | jd8 _| js"| js&d S t| j�}| j�� W d Q R X t�d|� d S )Nr z5Caught subprocesses termination from unknown pids: %s)r r r r5 r' r r� )r r� r� r� Zcollateral_victimsr r r r� � s zFastChildWatcher.__exit__c G sz | j std��| jd kr td��| j�: y| j�|�}W n" tk rZ ||f| j|< d S X W d Q R X |||f|�� d S )Nz Must use the context managerzICannot add child handler, the child watcher does not have a loop attached) r re r� r4 r r r r@ r� )r r_ r: r; r` r r r rS s z"FastChildWatcher.add_child_handlerc C s& y | j |= dS tk r dS X d S )NTF)r� r@ )r r_ r r r r� s z%FastChildWatcher.remove_child_handlerc C s� x�yt �dt j�\}}W n tk r, d S X |dkr:d S | �|�}| j�v y| j�|�\}}W nB tk r� | j r�|| j |< | j � � r�t �d||� wd }Y nX | j � � r�t �d||� W d Q R X |d kr�t �d||� q|||f|�� qW d S )Nr, r z,unknown process %s exited with returncode %sz$process %s exited with returncode %sz8Caught subprocess termination from unknown pid: %d -> %d)rf r r r r r r� r r@ r r r� r� r r� r� )r r_ r r` r: r; r r r r� ' s6 z FastChildWatcher._do_waitpid_all) r� r� r� r� r r r� r� rS r� r� r� r r )r r r � s r c sH e Zd ZdZeZ� fdd�Zdd� Z� fdd�Zdd � Z d d � Z � Z S ) �_UnixDefaultEventLoopPolicyz:UNIX event loop policy with a watcher for child processes.c s t � �� d | _d S )N)r r �_watcher)r )r r r r V s z$_UnixDefaultEventLoopPolicy.__init__c C sH t j�8 | jd kr:t� | _tt�� tj�r:| j�| j j � W d Q R X d S )N) r r r r rE r �current_thread� _MainThreadr� �_localr� )r r r r � _init_watcherZ s z)_UnixDefaultEventLoopPolicy._init_watcherc s6 t � �|� | jdk r2tt�� tj�r2| j�|� dS )z�Set the event loop. As a side effect, if a child watcher was set before, then calling .set_event_loop() from the main thread will call .attach_loop(loop) on the child watcher. N)r �set_event_loopr rE r r r r� )r rv )r r r r b s z*_UnixDefaultEventLoopPolicy.set_event_loopc C s | j dkr| �� | j S )zzGet the watcher for child processes. If not yet set, a SafeChildWatcher object is automatically created. N)r r )r r r r rP p s z-_UnixDefaultEventLoopPolicy.get_child_watcherc C s4 |dkst |t�st�| jdk r*| j�� || _dS )z$Set the watcher for child processes.N)rE r re r r )r r\ r r r �set_child_watcherz s z-_UnixDefaultEventLoopPolicy.set_child_watcher) r� r� r� r� r Z _loop_factoryr r r rP r r� r r )r r r R s r )-r� r8 r� rf r� r/ rh rx r� r r r$ � r r r r r r r r r �logr �__all__r� � ImportErrorr ZBaseSelectorEventLoopr Z ReadTransportrH Z_FlowControlMixinZWriteTransportrN ZBaseSubprocessTransportrR r r� r r ZBaseDefaultEventLoopPolicyr r r r r r r �<module> sR | JF=On3
Save File
Cancel