CasperSecurity
o
`�wic � @ 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�r"|tjkrtj}n|tjkr dd� }ntd��|| _ d S )NFc S s d S �N� )�
unused_signum� unused_framer r �'/usr/lib/python3.10/unittest/signals.py�default_handler s z3_InterruptHandler.__init__.<locals>.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| �||� | jr| �||� d| _t�� D ]}|�� q d S )NT)r � getsignal�SIGINTr r
�_results�keys�stop)r �signum�frame�installed_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 s2 t d u rt�tj�} t| �a t�tjt � d S d S r )�_interrupt_handlerr r r r )r r r r �installHandler1 s
�r) c s@ � d urt � �� fdd��}|S td ur t�tjtj� d S d S )Nc
sB t �t j�}t� z� | i |��W t � t j|� S t � t j|� w r )r r r �
removeHandler)�args�kwargs�initial��methodr r �inner; s
z removeHandler.<locals>.inner)r r( r r r )r/ r0 r r. r r* 9 s �r* r )r �weakref� functoolsr �
__unittest�objectr �WeakKeyDictionaryr r$ r' r( r) r* r r r r �<module> s