1 files changed, 101 insertions, 0 deletions
diff --git a/venv/lib/python3.11/site-packages/litestar/datastructures/upload_file.py b/venv/lib/python3.11/site-packages/litestar/datastructures/upload_file.py
new file mode 100644
index 0000000..09ad2d3
--- /dev/null
+++ b/venv/lib/python3.11/site-packages/litestar/datastructures/upload_file.py
@@ -0,0 +1,101 @@
+from __future__ import annotations
+
+from tempfile import SpooledTemporaryFile
+
+from litestar.concurrency import sync_to_thread
+from litestar.constants import ONE_MEGABYTE
+
+__all__ = ("UploadFile",)
+
+
+class UploadFile:
+    """Representation of a file upload"""
+
+    __slots__ = ("filename", "file", "content_type", "headers")
+
+    def __init__(
+        self,
+        content_type: str,
+        filename: str,
+        file_data: bytes | None = None,
+        headers: dict[str, str] | None = None,
+        max_spool_size: int = ONE_MEGABYTE,
+    ) -> None:
+        """Upload file in-memory container.
+
+        Args:
+            content_type: Content type for the file.
+            filename: The filename.
+            file_data: File data.
+            headers: Any attached headers.
+            max_spool_size: The size above which the temporary file will be rolled to disk.
+        """
+        self.filename = filename
+        self.content_type = content_type
+        self.file = SpooledTemporaryFile(max_size=max_spool_size)
+        self.headers = headers or {}
+
+        if file_data:
+            self.file.write(file_data)
+            self.file.seek(0)
+
+    @property
+    def rolled_to_disk(self) -> bool:
+        """Determine whether the spooled file exceeded the rolled-to-disk threshold and is no longer in memory.
+
+        Returns:
+            A boolean flag
+        """
+        return getattr(self.file, "_rolled", False)
+
+    async def write(self, data: bytes) -> int:
+        """Proxy for data writing.
+
+        Args:
+            data: Byte string to write.
+
+        Returns:
+            None
+        """
+        if self.rolled_to_disk:
+            return await sync_to_thread(self.file.write, data)
+        return self.file.write(data)
+
+    async def read(self, size: int = -1) -> bytes:
+        """Proxy for data reading.
+
+        Args:
+            size: position from which to read.
+
+        Returns:
+            Byte string.
+        """
+        if self.rolled_to_disk:
+            return await sync_to_thread(self.file.read, size)
+        return self.file.read(size)
+
+    async def seek(self, offset: int) -> int:
+        """Async proxy for file seek.
+
+        Args:
+            offset: start position..
+
+        Returns:
+            None.
+        """
+        if self.rolled_to_disk:
+            return await sync_to_thread(self.file.seek, offset)
+        return self.file.seek(offset)
+
+    async def close(self) -> None:
+        """Async proxy for file close.
+
+        Returns:
+            None.
+        """
+        if self.rolled_to_disk:
+            return await sync_to_thread(self.file.close)
+        return self.file.close()
+
+    def __repr__(self) -> str:
+        return f"{self.filename} - {self.content_type}"