a
XC?hc � @ 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 sR t �t j�}|| ur | �||� | jr2| �||� d| _t�� D ] }|�� q@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 r )�boolr �popr r r r � removeResult- s r# c C s. t d u r*t�tj�} t| �a t�tjt � d S r )�_interrupt_handlerr r r r )r r r r �installHandler1 s r% c s< � d ur t � �� fdd��}|S td ur8t�tjtj� d S )Nc
sF t �t j�}t� z � | i |��W t � t j|� S t � t j|� 0 d S r )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