Format: application/x-python-code
1 o
2
3 ;'hI � @ s� U d dl mZ d dlZd dlmZ d dlmZmZmZm Z m
4 Z
5 d dl m Z ddl
6 mZ e d�Ze d �Zd#d d
7 �Zd#dd�Zd#dd�Zd$dd�Ze � Zded< i Zded< edd�G dd � d ��ZG d d � d ej�ZG d d � d ee �ZG d!d"� d"ee �Z dS )%� )� annotationsN)� dataclass)�Any�Generic�Literal�TypeVar�overload)�WeakKeyDictionary� )�get_async_backend�T�D�return�Nonec � � �t � �� I dH dS )z�
8 Check for cancellation and allow the scheduler to switch to another task.
9
10 Equivalent to (but more efficient than)::
11
12 await checkpoint_if_cancelled()
13 await cancel_shielded_checkpoint()
14
15
16 .. versionadded:: 3.0
17
18 N)r �
19 checkpoint� r r �DC:\Users\shubh\Videos\agent\nova\Lib\site-packages\anyio/lowlevel.pyr � �
20 r c � r )z�
21 Enter a checkpoint if the enclosing cancel scope has been cancelled.
22
23 This does not allow the scheduler to switch to a different task.
24
25 .. versionadded:: 3.0
26
27 N)r �checkpoint_if_cancelledr r r r r s � r c � r )z�
28 Allow the scheduler to switch to another task but without checking for cancellation.
29
30 Equivalent to (but potentially more efficient than)::
31
32 with CancelScope(shield=True):
33 await checkpoint()
34
35
36 .. versionadded:: 3.0
37
38 N)r �cancel_shielded_checkpointr r r r r * r r �objectc C s
39 t � �� S )ze
40 Return a backend specific token object that can be used to get back to the event
41 loop.
42
43 )r �
44 current_tokenr r r r r : s
45 r z&WeakKeyDictionary[Any, dict[str, Any]]� _run_varszdict[Any, _TokenWrapper]�_token_wrappersT)�frozenc @ s e Zd ZU dZded< dS )�
46 _TokenWrapper)�_token� __weakref__r r N)�__name__�
47 __module__� __qualname__� __slots__�__annotations__r r r r r G s
48 r c @ s e Zd Ze�� ZdS )� _NoValueSetN)r r r! �enum�auto� NO_VALUE_SETr r r r r$ M s r$ c @ s e Zd ZdZd dd�ZdS )
49 � RunvarToken��_var�_value� _redeemed�var� RunVar[T]�value�%T | Literal[_NoValueSet.NO_VALUE_SET]c C s || _ || _d| _d S )NFr) )�selfr- r/ r r r �__init__T s
50 zRunvarToken.__init__N)r- r. r/ r0 )r r r! r" r2 r r r r r( Q s r( c @ s� e Zd ZU dZdZejZded< e� Z ded< efd$d d �Z
51 e d%dd��Z e
52 d&dd��Ze
53 d'dd��Zefd(dd�Zd)dd �Zd*d d �Zd+d!d"�Zd#S ),�RunVarz[
54 Like a :class:`~contextvars.ContextVar`, except scoped to the running event loop.
55 ��_name�_defaultz!Literal[_NoValueSet.NO_VALUE_SET]r' zset[_TokenWrapper]r �name�str�defaultr0 c C s || _ || _d S �Nr4 )r1 r7 r9 r r r r2 e s
56 zRunVar.__init__r � dict[str, T]c C s4 t � }zt| W S ty i }t|< | Y S w r: )r r �KeyError)r1 �tokenZrun_varsr r r �
57 _current_varsk s
58 �zRunVar._current_varsr
59 �T | Dc C � d S r: r �r1 r9 r r r �gett � z
60 RunVar.getr c C r@ r: r �r1 r r r rB w rC �%D | Literal[_NoValueSet.NO_VALUE_SET]c C s\ z| j | j W S ty$ |tjur| Y S | jtjur"| j Y S Y nw td| j� d���)NzRun variable "z!" has no value and no default set)r> r5 r< r3 r' r6 � LookupErrorrA r r r rB z s
61
62 �� �r/ �RunvarToken[T]c C s* | j }t| |�| jtj��}||| j< |S r: )r> r( rB r5 r3 r' )r1 r/ Z current_varsr= r r r �set� s
63 z
64 RunVar.setr= r c C sf |j | ur td��|jrtd��|jtju r'z| j| j= W n ty& Y nw |j| j| j< d|_d S )Nz)This token does not belong to this RunVarz This token has already been usedT) r* �
65 ValueErrorr, r+ r$ r' r> r5 r< )r1 r= r r r �reset� s
66 �
67 z RunVar.resetc C s d| j �d�S )Nz
68 <RunVar name=�>)r5 rD r r r �__repr__� s zRunVar.__repr__N)r7 r8 r9 r0 )r r; )r9 r
69 r r? )r r )r9 rE r r? )r/ r r rG )r= rG r r )r r8 )r r r! �__doc__r" r$ r' r# rH r r2 �propertyr> r rB rJ rL r r r r r3 Z s$
70 � �
71
72 r3 )r r )r r ) �
73 __future__r r% � dataclassesr �typingr r r r r �weakrefr Z_core._eventloopr r r
74 r r r r r r# r r �Enumr$ r( r3 r r r r �<module> s&
75
76
77
78