← Back
Editing: semaphore_tracker.cpython-36.pyc
3 \ � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZddlmZ ddlmZ dddgZ G dd � d e �Z e � Z e j Z e jZe jZe jZd d � ZdS ) � N� )�spawn)�util�ensure_running�register� unregisterc @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�SemaphoreTrackerc C s t j� | _d | _d | _d S )N)� threadingZLock�_lock�_fd�_pid)�self� r �9/usr/lib64/python3.6/multiprocessing/semaphore_tracker.py�__init__ s zSemaphoreTracker.__init__c C s | j � | jS )N)r r )r r r r �getfd" s zSemaphoreTracker.getfdc C s | j �� | jdk rRtj| jtj�\}}|s0dS tj| j� d| _d| _tjd� g }y|j t j j � � W n t k r~ Y nX d}tj� \}}zlyB|j |� tj� }|gtj� }|d|| g7 }tj|||�}W n tj|� � Y nX || _|| _W dtj|� X W dQ R X dS )z�Make sure that semaphore tracker process is running. This can be run from any process. Usually a child process will use the semaphore created by its parent.NzWsemaphore_tracker: process died unexpectedly, relaunching. Some semaphores might leak.z;from multiprocessing.semaphore_tracker import main;main(%d)z-c)r r �os�waitpid�WNOHANG�closer �warnings�warn�append�sys�stderr�fileno� Exception�piper Zget_executabler Z _args_from_interpreter_flagsZspawnv_passfds) r �pidZstatusZ fds_to_pass�cmd�r�wZexe�argsr r r r & s8 z SemaphoreTracker.ensure_runningc C s | j d|� dS )z2Register name of semaphore with semaphore tracker.ZREGISTERN)�_send)r �namer r r r Q s zSemaphoreTracker.registerc C s | j d|� dS )z4Unregister name of semaphore with semaphore tracker.Z UNREGISTERN)r# )r r$ r r r r U s zSemaphoreTracker.unregisterc C sP | j � dj||�jd�}t|�dkr.td��tj| j|�}|t|�ksLt�d S )Nz{0}:{1} �asciii z name too long) r �format�encode�len� ValueErrorr �writer �AssertionError)r r r$ �msg�nbytesr r r r# Y s zSemaphoreTracker._sendN) �__name__� __module__� __qualname__r r r r r r# r r r r r s +r c - C s� t j t jt j� t j t jt j� x6tjtjfD ]&}y |j� W q. tk rR Y q.X q.W t � }z�t | d���}x�|D ]�}yJ|j � j d�\}}|dkr�|j |� n |dkr�|j|� n td| ��W qr tk r� ytjtj� � W n Y nX Y qrX qrW W dQ R X W d|�r<ytjdt|� � W n tk �r: Y nX xf|D ]^}zT|jd�}ytj|� W n6 tk �r� } ztjd ||f � W Y dd}~X nX W dX �qBW X dS ) zRun semaphore tracker.�rb� :s REGISTERs UNREGISTERzunrecognized command %rNzRsemaphore_tracker: There appear to be %d leaked semaphores to clean up at shutdownr% zsemaphore_tracker: %r: %s)�signal�SIGINT�SIG_IGN�SIGTERMr �stdin�stdoutr r �set�open�strip�split�add�remove� RuntimeError� excepthook�exc_infor r r( �decode�_multiprocessingZ sem_unlink)�fd�f�cache�liner r$ �er r r �maink sJ (rI )r r3 r r r rC � r r �__all__�objectr Z_semaphore_trackerr r r r rI r r r r �<module> s I
Save File
Cancel