Format: application/x-python-code
1 o
2
3 �h� � @ sN d Z ddlZddlZddlZddlZddlZddlZddlZG dd� d�ZdS )�A trio loop runner.� Nc @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
4 d � Zd S )
5 �
6 TrioRunnerr c C s d| _ d| _dS )�Initialize the runner.N)�_cell_cancel_scope� _trio_token��self� r �Kc:\Users\shubh\Videos\agent\nova\Lib\site-packages\ipykernel/trio_runner.py�__init__ s
7 zTrioRunner.__init__c C sT |j �| � |j �dd� dd� |j jjd d< d| _tj|jddd �}|�� d
8 S ) r Z autoawait�trioc S s t jddd�S )Nz5Autoawait isn't allowed in Trio background loop mode.� )�
9 stacklevel)�warnings�warn)�_r r r
10 �<lambda> s �z'TrioRunner.initialize.<locals>.<lambda>�lineFTZTornadoBackground)�target�daemon�nameN) �shellZset_trio_runnerZrun_line_magicZmagics_managerZmagicsZ _interrupted� threading�Thread�start)r ZkernelZio_loopZ bg_threadr r r
11 �
12 initialize s zTrioRunner.initializec C s | j r
13 | j �� dS d}t|��)zInteruppt the runner.z)Kernel interrupted but no cell is runningN)r �cancel� Exception)r Zsignum�frame�msgr r r
14 � interrupt s zTrioRunner.interruptc sB t � t j�j�}dd� � � �fdd�}t�|� t � t j|� dS )z
15 Run the loop.c S s* d� t�t| �| | j��} t�d| � d S )N�
16 z2An exception occurred in a global nursery task.
17 %s)�join� traceback�format_exception�type�
18 __traceback__�logging�error)�excr r r
19 �log_nursery_exc+ s z'TrioRunner.run.<locals>.log_nursery_excc � sj �t j�� �_t �� 4 I dH �} � | _| t_t �� I dH W d �I dH dS 1 I dH s.w Y dS )zRun the main loop.N) r ZlowlevelZcurrent_trio_tokenr Z open_nurseryZ_add_exc�builtinsZGLOBAL_NURSERYZ
20 sleep_forever)Znursery�r* r r r
21 � trio_main/ s � .�z!TrioRunner.run.<locals>.trio_mainN)�signal�SIGINTr r �run)r Zold_sigr- r r, r
22 r0 ' s
23
24
25 zTrioRunner.runc s � fdd�}t jj||� jd�S )zHandle a function call.c � sF �t �� � _� j�
26 | I dH W d � S 1 sw Y d� _dS )zA thread runner context.N)r Z CancelScoper )�coror r r
27 �loc? s �
28 �z TrioRunner.__call__.<locals>.loc)Z
29 trio_token)r Z from_threadr0 r )r Zasync_fnr2 r r r
30 �__call__< s zTrioRunner.__call__N) �__name__�
31 __module__� __qualname__�__doc__r r r r0 r3 r r r r
32 r s r ) r7 r+ r' r. r r# r r r r r r r
33 �<module> s