B
YI~dc � @ sb d dl Z d dlZd dlmZ dZG dd� de�Ze�� Zdd� Z dd � Z
da d
d � Z dd d
�Z
dS )� N)�wrapsTc @ s e Zd Zdd� Zdd� ZdS )�_InterruptHandlerc C sN d| _ || _t|t�rD|tjkr(tj}n |tjkr.default_handlerzYexpected SIGINT signal handler to be signal.SIG_IGN, signal.SIG_DFL, or a callable object)
�called�original_handler�
isinstance�int�signal�SIG_DFL�default_int_handler�SIG_IGN� TypeErrorr )�selfr r r r �__init__
s
z_InterruptHandler.__init__c C sV t �t j�}|| k r | �||� | jr2| �||� d| _xt�� D ] }|�� qBW d S )NT)r � getsignal�SIGINTr r �_results�keys�stop)r Zsignum�frameZinstalled_handler�resultr r r �__call__ s z_InterruptHandler.__call__N)�__name__�
__module__� __qualname__r r r r r r r s r c C s dt | < d S )N� )r )r r r r �registerResult* s r c C s t t�| d ��S )N)�boolr �pop)r r r r � removeResult- s r! c C s. t d kr*t�tj�} t| �a t�tjt � d S )N)�_interrupt_handlerr r r r )r r r r �installHandler1 s r# c s< � d k r t � �� fdd��}|S td k r8t�tjtj� d S )Nc
s2 t �t j�}t� z
� | |�S t � t j|� X d S )N)r r r �
removeHandler)�args�kwargs�initial)�methodr r �inner; s
z removeHandler..inner)r r" r r r )r( r) r )r( r r$ 9 s
r$ )N)r �weakref� functoolsr Z
__unittest�objectr �WeakKeyDictionaryr r r! r" r# r$ r r r r � s