U
|
Z ±d ã @ s d dl Z d dlZd dlZd dlmZ d dlZd dlZddlmZ dZ dd Zdd Zd
|
d Zed Zd e_edZde_ejG dd dZdS )é N)Úwrapsé )Úis_main_threadz@TRIO_KI_PROTECTION_ENABLEDc C s8 | d k r4t | jkr | jt S | jjdkr,dS | j} q dS )NÚ__del__T)Ú LOCALS_KEY_KI_PROTECTION_ENABLEDÚf_localsÚf_codeÚco_nameÚf_back)Úframe© r úEd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_core/_ki.pyÚki_protection_enabledV s
|
|
r c C s t t ¡ S )a Check whether the calling code has :exc:`KeyboardInterrupt` protection
|
enabled.
|
|
It's surprisingly easy to think that one's :exc:`KeyboardInterrupt`
|
protection is enabled when it isn't, or vice-versa. This function tells
|
you what Trio thinks of the matter, which makes it useful for ``assert``\s
|
and unit tests.
|
|
Returns:
|
bool: True if protection is enabled, and False otherwise.
|
|
)r ÚsysÚ _getframer r r r Úcurrently_ki_protected` s r c s fdd}|S )Nc s t ¡r$t fdd}|S t ¡rHt fdd}|S t ¡rlt fdd}|S t fdd}|S d S )Nc s | |} |j jt< |S ©N)Úcr_framer r )ÚargsÚkwargsÚcoro©ÚenabledÚfnr r Úwrapperw s
|
z<_ki_protection_decorator.<locals>.decorator.<locals>.wrapperc s | |} |j jt< |S r )Úgi_framer r )r r Úgenr r r r s
|
|
c s | |} |j jt< |S r )Úag_framer r )r r Zagenr r r r s
|
c s t t< | |S r )Úlocalsr )r r r r r r s
|
)ÚinspectÚiscoroutinefunctionr ÚisgeneratorfunctionÚasync_generatorÚisasyncgenfunction)r r ©r )r r Ú decoratorq s
|
|
|
z+_ki_protection_decorator.<locals>.decoratorr )r r% r r$ r Ú_ki_protection_decoratorp s 2r& TÚenable_ki_protectionFÚdisable_ki_protectionc @ s( e Zd ZejddZdd Zdd ZdS )Ú KIManagerN)Údefaultc sP | j d kstt r&t tj¡tjkr*d S fdd}|| _ t tj|¡ d S )Nc s. | t jkstt|}|s r& ntd S r )ÚsignalÚSIGINTÚAssertionErrorr ÚKeyboardInterrupt)Úsignumr Zprotection_enabled©Ú
|
deliver_cbÚ*restrict_keyboard_interrupt_to_checkpointsr r Úhandler¹ s
|
z"KIManager.install.<locals>.handler)r3 r- r r+ Ú getsignalr, Údefault_int_handler)Úselfr1 r2 r3 r r0 r Úinstall± s ÿþzKIManager.installc C s6 | j d k r2t tj¡| j kr,t tjtj¡ d | _ d S r )r3 r+ r4 r, r5 )r6 r r r ÚcloseÄ s
|
zKIManager.close)Ú__name__Ú
|
__module__Ú __qualname__ÚattrÚibr3 r7 r8 r r r r r) s r) )r r+ r Ú functoolsr r< r" Z_utilr ÚAnyÚTypeVarÚCallableÚFr r r r& r' r9 r( Úsr) r r r r Ú<module> s H
|
6
|
