U
iÛfc ã @ 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¡}|| k r | ||¡ | 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 kr*t tj¡} t| ƒa t tjt ¡ d S r )Ú_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
s6 t t j¡}tƒ zˆ | |ŽW ¢S t t j|¡ X 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