Read stream for downloads

playwright/_impl/_artifact.py

@@ -14,7 +14,7 @@
 
 import pathlib
 from pathlib import Path
-from typing import Dict, Optional, Union, cast
+from typing import Dict, Iterable, Optional, Union, cast
 
 from playwright._impl._connection import ChannelOwner, from_channel
 from playwright._impl._helper import Error, make_dirs_for_file, patch_error_message
@@ -41,6 +41,11 @@ async def save_as(self, path: Union[str, Path]) -> None:
         make_dirs_for_file(path)
         await stream.save_as(path)
 
+    async def read_stream(self) -> Iterable[bytes]:
+        stream = cast(Stream, from_channel(await self._channel.send("stream")))
+        async for chunk in stream.read_stream():
+            yield chunk
+
     async def failure(self) -> Optional[str]:
         return patch_error_message(await self._channel.send("failure"))
 

playwright/_impl/_download.py

@@ -14,7 +14,7 @@
 
 import pathlib
 from pathlib import Path
-from typing import TYPE_CHECKING, Optional, Union
+from typing import TYPE_CHECKING, Iterable, Optional, Union
 
 from playwright._impl._artifact import Artifact
 
@@ -60,5 +60,9 @@ async def path(self) -> pathlib.Path:
     async def save_as(self, path: Union[str, Path]) -> None:
         await self._artifact.save_as(path)
 
+    async def read_stream(self) -> Iterable[bytes]:
+        async for chunk in self._artifact.read_stream():
+            yield chunk
+
     async def cancel(self) -> None:
         return await self._artifact.cancel()

playwright/_impl/_stream.py

@@ -14,7 +14,7 @@
 
 import base64
 from pathlib import Path
-from typing import Dict, Union
+from typing import Dict, Iterable, Union
 
 from playwright._impl._connection import ChannelOwner
 
@@ -36,6 +36,13 @@ async def save_as(self, path: Union[str, Path]) -> None:
             )
         await self._loop.run_in_executor(None, lambda: file.close())
 
+    async def read_stream(self) -> Iterable[bytes]:
+        while True:
+            binary = await self._channel.send("read", {"size": 1024 * 1024})
+            if not binary:
+                break
+            yield base64.b64decode(binary)
+
     async def read_all(self) -> bytes:
         binary = b""
         while True:

playwright/async_api/_generated.py

@@ -15,7 +15,7 @@
 
 import pathlib
 import typing
-from typing import Literal
+from typing import Iterable, Literal
 
 from playwright._impl._accessibility import Accessibility as AccessibilityImpl
 from playwright._impl._api_structures import (
@@ -6852,6 +6852,15 @@ async def save_as(self, path: typing.Union[str, pathlib.Path]) -> None:
 
         return mapping.from_maybe_impl(await self._impl_obj.save_as(path=path))
 
+    async def read_stream(self) -> Iterable[bytes]:
+        """Download.read_stream
+
+        Yields a readable stream chunks for a successful download, or throws for a failed/canceled download.
+        """
+
+        async for chunk in mapping.from_maybe_impl(self._impl_obj.read_stream()):
+            yield chunk
+
     async def cancel(self) -> None:
         """Download.cancel
 

playwright/sync_api/_generated.py

@@ -15,7 +15,7 @@
 
 import pathlib
 import typing
-from typing import Literal
+from typing import Iterable, Literal
 
 from playwright._impl._accessibility import Accessibility as AccessibilityImpl
 from playwright._impl._api_structures import (
@@ -6962,6 +6962,15 @@ def save_as(self, path: typing.Union[str, pathlib.Path]) -> None:
 
         return mapping.from_maybe_impl(self._sync(self._impl_obj.save_as(path=path)))
 
+    def read_stream(self) -> Iterable[bytes]:
+        """Download.read_stream
+
+        Yields a readable stream chunks for a successful download, or throws for a failed/canceled download.
+        """
+
+        for chunk in mapping.from_maybe_impl(self._sync(self._impl_obj.read_stream())):
+            yield chunk
+
     def cancel(self) -> None:
         """Download.cancel
 

tests/async/test_download.py

@@ -202,6 +202,19 @@ async def test_should_error_when_saving_after_deletion(
     await page.close()
 
 
+async def test_stream_reading(
+    browser: Browser, server: Server
+) -> None:
+    page = await browser.new_page(accept_downloads=True)
+    await page.set_content(f'<a href="{server.PREFIX}/download">download</a>')
+    async with page.expect_download() as download_info:
+        await page.click("a")
+    download = await download_info.value
+    data = b"".join([chunk async for chunk in download.read_stream()])
+    assert data == b"Hello world"
+    await page.close()
+
+
 async def test_should_report_non_navigation_downloads(
     browser: Browser, server: Server
 ) -> None: