Format: application/x-python-code
1 o
2
3 ;'h�L � @ s� d dl mZ d dlZd dlZd dlZd dlmZ d dlmZm Z d dl
4 m Z d dl m
5 Z
6 mZmZmZmZ ddlmZ dd lmZ dd
7 lmZ e
8 rTd d lmZmZmZm Z G d d
9 � d
10 ee �Z G dd� dee �Z G dd� dee �Z G dd� dee �Z e d0d1d d ��Z!e d0d2d!d ��Z! d0d3d$d �Z!e d4d5d&d'��Z"e d4d6d)d'��Z" d4d7d+d'�Z"d8d,d-�Z#d9d.d/�Z$dS ):� )� annotationsN)�Iterable)�BytesIO�
11 TextIOWrapper)�
12 TracebackType)�
13 TYPE_CHECKING�Any�AnyStr�Generic�overload� )� to_thread)� AsyncFile)�checkpoint_if_cancelled)�OpenBinaryMode� OpenTextMode�ReadableBuffer�WriteableBufferc @ s� e Zd ZU dZded< e d)dd�d*dd��Ze d+dd�d,dd��Z d-dd�d.d d�Zd/d d �Zd0d'd(�ZdS )1�
14 TemporaryFilea�
15 An asynchronous temporary file that is automatically created and cleaned up.
16
17 This class provides an asynchronous context manager interface to a temporary file.
18 The file is created using Python's standard `tempfile.TemporaryFile` function in a
19 background thread, and is wrapped as an asynchronous file using `AsyncFile`.
20
21 :param mode: The mode in which the file is opened. Defaults to "w+b".
22 :param buffering: The buffering policy (-1 means the default buffering).
23 :param encoding: The encoding used to decode or encode the file. Only applicable in
24 text mode.
25 :param newline: Controls how universal newlines mode works (only applicable in text
26 mode).
27 :param suffix: The suffix for the temporary file name.
28 :param prefix: The prefix for the temporary file name.
29 :param dir: The directory in which the temporary file is created.
30 :param errors: The error handling scheme used for encoding/decoding errors.
31 �AsyncFile[AnyStr]� _async_file.��errors�self�TemporaryFile[bytes]�moder � buffering�int�encoding�
32 str | None�newline�suffix�prefix�dirr c C � d S �N� � r r r r r r! r" r# r r& r& �KC:\Users\shubh\Videos\agent\nova\Lib\site-packages\anyio/_core/_tempfile.py�__init__/ � zTemporaryFile.__init__�TemporaryFile[str]r c C r$ r% r& r' r& r& r( r) < r* �w+b�����N� OpenTextMode | OpenBinaryMode�return�Nonec C s4 || _ || _|| _|| _|| _|| _|| _|| _d S r% �r r r r r! r" r# r r' r& r& r( r) J s
33 c � �* �t �� fdd��I d H }t|�� _� jS )Nc
34 s* t j� j� j� j� j� j� j� j� j d�S )Nr )
35 �tempfiler r r r r r! r" r# r r& �r r& r( �<lambda>a s �z*TemporaryFile.__aenter__.<locals>.<lambda>�r
36 �run_syncr r �r �fpr& r4 r( �
37 __aenter___ s �
38
39 �
40 zTemporaryFile.__aenter__�exc_type�type[BaseException] | None� exc_value�BaseException | None� traceback�TracebackType | Nonec � � �| j �� I d H d S r% �r �aclose�r r; r= r? r& r& r( � __aexit__o � �zTemporaryFile.__aexit__�.......)r r r r r r r r r r r! r r" r r# r r r )......)r r+ r r r r r r r r r! r r" r r# r r r )r, r- NNNNN)r r. r r r r r r r! r r" r r# r r r r/ r0 �r/ r �r; r< r= r> r? r@ r/ r0 � �__name__�
41 __module__� __qualname__�__doc__�__annotations__r r) r: rE r& r& r& r( r sH
42 �
43 � �
44 ��
45 �
46 r c @ s� e Zd ZU dZded< e d-ddd�d.dd��Ze d/ddd�d0dd��Z d1d d d�d2d"d�Zd3d#d$�Zd4d+d,�Zd S )5�NamedTemporaryFilea|
47 An asynchronous named temporary file that is automatically created and cleaned up.
48
49 This class provides an asynchronous context manager for a temporary file with a
50 visible name in the file system. It uses Python's standard
51 :func:`~tempfile.NamedTemporaryFile` function and wraps the file object with
52 :class:`AsyncFile` for asynchronous operations.
53
54 :param mode: The mode in which the file is opened. Defaults to "w+b".
55 :param buffering: The buffering policy (-1 means the default buffering).
56 :param encoding: The encoding used to decode or encode the file. Only applicable in
57 text mode.
58 :param newline: Controls how universal newlines mode works (only applicable in text
59 mode).
60 :param suffix: The suffix for the temporary file name.
61 :param prefix: The prefix for the temporary file name.
62 :param dir: The directory in which the temporary file is created.
63 :param delete: Whether to delete the file when it is closed.
64 :param errors: The error handling scheme used for encoding/decoding errors.
65 :param delete_on_close: (Python 3.12+) Whether to delete the file on close.
66 r r .)r �delete_on_closer �NamedTemporaryFile[bytes]r r r r r r r r! r" r# �delete�boolr rQ c C r$ r% r& � r r r r r r! r" r# rS r rQ r& r& r( r) � � zNamedTemporaryFile.__init__�NamedTemporaryFile[str]r c C r$ r% r& rU r& r& r( r) � rV r, r- NT� OpenBinaryMode | OpenTextModer/ r0 c
67 C s6 ||||||||| d� | _ tjdkr|
68 | j d< d S d S )N) r r r r r! r" r# rS r �� � rQ )�_params�sys� version_inforU r& r& r( r) � s �
69 �c � r2 )Nc � t jdi � j��S �Nr& )r3 rP r\ r& r4 r& r( r5 � � z/NamedTemporaryFile.__aenter__.<locals>.<lambda>r6 r8 r& r4 r( r: � s �
70
71 �
72 z NamedTemporaryFile.__aenter__r; r< r= r> r? r@ c � rA r% rB rD r& r& r( rE � rF z NamedTemporaryFile.__aexit__�........)r rR r r r r r r r r r! r r" r r# r rS rT r r rQ rT rG )r rW r r r r r r r r r! r r" r r# r rS rT r r rQ rT )r, r- NNNNNT)r rX r r r r r r r! r r" r r# r rS rT r r rQ rT r/ r0 rH rI rJ r& r& r& r( rP x sT
73 � �� �� �
74 rP c s� e Zd ZU dZdZded< e dTdd�dUdd��Ze dTdd�dVdd��Z dWd d�dX� fd!d�ZdY� fd"d#� ZdYd$d%�Z dYd&d'�Z
75 e dZd(d)��Z d[d\� fd,d-�
76 Z
77 d[d]� fd/d0�
78 Zd^� fd1d2� Zd_� fd4d5� Zd`� fd8d9� Zd`� fd:d;� Zejfda� fd?d@�
79 Zdb� fdAdB� Zdcdd� fdCdD�
80 ZededFdG��ZedfdIdG��Zdg� fdKdG� ZedhdNdO��ZedidQdO��Zdj� fdSdO� Z� ZS )k�SpooledTemporaryFilea�
81 An asynchronous spooled temporary file that starts in memory and is spooled to disk.
82
83 This class provides an asynchronous interface to a spooled temporary file, much like
84 Python's standard :class:`~tempfile.SpooledTemporaryFile`. It supports asynchronous
85 write operations and provides a method to force a rollover to disk.
86
87 :param max_size: Maximum size in bytes before the file is rolled over to disk.
88 :param mode: The mode in which the file is opened. Defaults to "w+b".
89 :param buffering: The buffering policy (-1 means the default buffering).
90 :param encoding: The encoding used to decode or encode the file (text mode only).
91 :param newline: Controls how universal newlines mode works (text mode only).
92 :param suffix: The suffix for the temporary file name.
93 :param prefix: The prefix for the temporary file name.
94 :param dir: The directory in which the temporary file is created.
95 :param errors: The error handling scheme used for encoding/decoding errors.
96 FrT �_rolled.r r �SpooledTemporaryFile[bytes]�max_sizer r r r r r r r! r" r# r c
97 C r$ r% r& �
98 r rf r r r r r! r" r# r r& r& r( r) � �
99 z SpooledTemporaryFile.__init__�SpooledTemporaryFile[str]r c
100 C r$ r% r& rg r& r& r( r) � rh r r, r- NrX r/ r0 c
101 sX |||||||| d�| _ || _d|v r t� �t� � d S t� �tt� || |dd�� d S )Nr1 �bT)r r r �
102 write_through)�_tempfile_params� _max_size�superr) r r rg �� __class__r& r( r) s* �
103 ��c � s* �| j s | j�� d S t� �� I d H d S r% )rd �_fp�closern rC r4 ro r& r( rC 3 s
104 �
105 zSpooledTemporaryFile.aclosec � s. �| j s | j�� | jk rd S | �� I d H d S r% )rd rq �tellrm �rolloverr4 r& r& r( �_check: s �zSpooledTemporaryFile._checkc � s\ �� j rd S d� _ � j}|�d� t�� fdd��I d H � _� �|�� �I d H |�� d S )NTr c r_ r` )r3 r rl r& r4 r& r( r5 H ra z/SpooledTemporaryFile.rollover.<locals>.<lambda>)rd rq �seekr
106 r7 �write�readrr )r �bufferr& r4 r( rt @ s �
107
108 � z SpooledTemporaryFile.rolloverc C s | j jS r% )rq �closedr4 r& r& r( rz M s zSpooledTemporaryFile.closed�sizer c � �2 �| j st� I d H | j�|�S t� �|�I d H S r% )rd r rq rx rn �r r{ ro r& r( rx Q �
109 � zSpooledTemporaryFile.read�bytesc � r| r% )rd r rq �read1rn r} ro r& r( r� X r~ zSpooledTemporaryFile.read1c � �. �| j st� I d H | j�� S t� �� I d H S r% )rd r rq �readlinern r4 ro r& r( r� _ �
110 �
111 z SpooledTemporaryFile.readline� list[AnyStr]c � r� r% )rd r rq � readlinesrn r4 ro r& r( r� f r� z SpooledTemporaryFile.readlinesrj r c � s2 �| j st� I d H | j�|� t� �|�I d H S r% )rd r rq �readintorn �r rj ro r& r( r� m r~ z SpooledTemporaryFile.readintoc � s2 �| j st� I d H | j�|� t� �|�I d H S r% )rd r rq r� rn � readinto1r� ro r& r( r� t r~ z SpooledTemporaryFile.readinto1�offset�whence�
112 int | Nonec � s6 �| j st� I d H | j�||�S t� �||�I d H S r% )rd r rq rv rn )r r� r� ro r& r( rv { s
113 � zSpooledTemporaryFile.seekc � r� r% )rd r rq rs rn r4 ro r& r( rs � r� zSpooledTemporaryFile.tellc � r| r% )rd r rq �truncatern r} ro r& r( r� � r~ z SpooledTemporaryFile.truncater c � � �d S r% r& r� r& r& r( rw � � �zSpooledTemporaryFile.write�strc � r� r% r& r� r& r& r( rw � r� �ReadableBuffer | strc � �D �| j st� I dH | j�|�}| �� I dH |S t� �|�I dH S )a�
114 Asynchronously write data to the spooled temporary file.
115
116 If the file has not yet been rolled over, the data is written synchronously,
117 and a rollover is triggered if the size exceeds the maximum size.
118
119 :param s: The data to write.
120 :return: The number of bytes written.
121 :raises RuntimeError: If the underlying file is not initialized.
122
123 N)rd r rq rw ru rn )r rj �resultro r& r( rw � s � �lines�Iterable[ReadableBuffer]c � r� r% r& �r r� r& r& r( �
124 writelines� � �z SpooledTemporaryFile.writelines�
125 Iterable[str]c � r� r% r& r� r& r& r( r� � r� �(Iterable[str] | Iterable[ReadableBuffer]c � r� )aw
126 Asynchronously write a list of lines to the spooled temporary file.
127
128 If the file has not yet been rolled over, the lines are written synchronously,
129 and a rollover is triggered if the size exceeds the maximum size.
130
131 :param lines: An iterable of lines to write.
132 :raises RuntimeError: If the underlying file is not initialized.
133
134 N)rd r rq r� ru rn )r r� r� ro r& r( r� � s � rb )r re rf r r r r r r r r r r! r r" r r# r r r )r ri rf r r r r r r r r r r! r r" r r# r r r )r r, r- NNNNN)rf r r rX r r r r r r r! r r" r r# r r r r/ r0 �r/ r0 )r/ rT )r- )r{ r r/ r )r re r{ r r/ r )r/ r )r/ r� )r re rj r r/ r )r� r r� r� r/ r )r/ r r% )r{ r� r/ r )r re rj r r/ r )r ri rj r� r/ r )rj r� r/ r )r re r� r� r/ r0 )r ri r� r� r/ r0 )r� r� r/ r0 )rK rL rM rN rd rO r r) rC ru rt �propertyrz rx r� r� r� r� r� �os�SEEK_SETrv rs r� rw r� �
135 __classcell__r& r& ro r( rc � s|
136 � �
137 � �� �%
138
139
140 rc c @ sH e Zd ZdZ d ddd�d dd�Zd dd�Zd!dd�Zd"d d �ZdS )#�TemporaryDirectorya�
141 An asynchronous temporary directory that is created and cleaned up automatically.
142
143 This class provides an asynchronous context manager for creating a temporary
144 directory. It wraps Python's standard :class:`~tempfile.TemporaryDirectory` to
145 perform directory creation and cleanup operations in a background thread.
146
147 :param suffix: Suffix to be added to the temporary directory name.
148 :param prefix: Prefix to be added to the temporary directory name.
149 :param dir: The parent directory where the temporary directory is created.
150 :param ignore_cleanup_errors: Whether to ignore errors during cleanup
151 (Python 3.10+).
152 :param delete: Whether to delete the directory upon closing (Python 3.12+).
153 NFT)�ignore_cleanup_errorsrS r! �
154 AnyStr | Noner" r# r� rT rS r/ r0 c C s( || _ || _|| _|| _|| _d | _d S r% )r! r" r# r� rS �_tempdir)r r! r" r# r� rS r& r& r( r) � s
155 zTemporaryDirectory.__init__r� c � sj �| j | j| jd�� tjdkr| j� d< tjdkr | j� d< t�� fdd��I d H | _ t�| j j
156 �I d H S )N�r! r" r# )rZ �
157 r� rY rS c s t jdi � ��S r` )r3 r� r& ��paramsr& r( r5 � s z/TemporaryDirectory.__aenter__.<locals>.<lambda>) r! r" r# r] r^ r� rS r
158 r7 r� � __enter__r4 r& r� r( r: � s ��
159
160
161
162
163 �z TemporaryDirectory.__aenter__r; r< r= r> r? r@ c � s. �| j d urt�| j j|||�I d H d S d S r% )r� r
164 r7 �__exit__rD r& r& r( rE � s �
165 ��z TemporaryDirectory.__aexit__c � s( �| j d urt�| j j�I d H d S d S r% )r� r
166 r7 �cleanupr4 r& r& r( r� s �
167 �zTemporaryDirectory.cleanup�NNN) r! r� r" r� r# r� r� rT rS rT r/ r0 �r/ r� rI r� )rK rL rM rN r) r: rE r� r& r& r& r( r� � s ��
168
169 r� Fr! r r" r# �textrT r/ �tuple[int, str]c � r� r% r& �r! r" r# r� r& r& r( �mkstemp � �r� � bytes | None�tuple[int, bytes]c � r� r% r& r� r& r& r( r� r� r� �tuple[int, str | bytes]c � s �t �tj| |||�I dH S )a�
170 Asynchronously create a temporary file and return an OS-level handle and the file
171 name.
172
173 This function wraps `tempfile.mkstemp` and executes it in a background thread.
174
175 :param suffix: Suffix to be added to the file name.
176 :param prefix: Prefix to be added to the file name.
177 :param dir: Directory in which the temporary file is created.
178 :param text: Whether the file is opened in text mode.
179 :return: A tuple containing the file descriptor and the file name.
180
181 N)r
182 r7 r3 r� r� r& r& r( r� s �r� c � r� r% r& r� r& r& r( �mkdtemp0 � �r� r c � r� r% r& r� r& r& r( r� 8 r� � str | bytesc � s �t �tj| ||�I dH S )a�
183 Asynchronously create a temporary directory and return its path.
184
185 This function wraps `tempfile.mkdtemp` and executes it in a background thread.
186
187 :param suffix: Suffix to be added to the directory name.
188 :param prefix: Prefix to be added to the directory name.
189 :param dir: Parent directory where the temporary directory is created.
190 :return: The path of the created temporary directory.
191
192 N)r
193 r7 r3 r� r� r& r& r( r� @ s �c � � �t �tj�I dH S )z�
194 Asynchronously return the name of the directory used for temporary files.
195
196 This function wraps `tempfile.gettempdir` and executes it in a background thread.
197
198 :return: The path of the temporary directory as a string.
199
200 N)r
201 r7 r3 �
202 gettempdirr& r& r& r( r� S � � r� c � r� )z�
203 Asynchronously return the name of the directory used for temporary files in bytes.
204
205 This function wraps `tempfile.gettempdirb` and executes it in a background thread.
206
207 :return: The path of the temporary directory as bytes.
208
209 N)r
210 r7 r3 � gettempdirbr& r& r& r( r� _ r� r� )NNNF)
211 r! r r" r r# r r� rT r/ r� )
212 r! r� r" r� r# r� r� rT r/ r� )
213 r! r� r" r� r# r� r� rT r/ r� r� )r! r r" r r# r r/ r� )r! r� r" r� r# r� r/ r )r! r� r" r� r# r� r/ r� r� )r/ r )%�
214 __future__r r� r] r3 �collections.abcr �ior r �typesr �typingr r r r
215 r � r
216 Z
217 _core._fileior Zlowlevelr Z _typeshedr r r r r rP rc r� r� r� r� r� r& r& r& r( �<module> sf _d kB�� ��� �
218