✨ Add filename override possibility

reworkd · Jun 5, 2024 · 720d7d5 · 720d7d5
1 parent e565019
commit 720d7d5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/harambe/core.py b/harambe/core.py
@@ -193,6 +193,7 @@ async def capture_url(
     async def capture_download(
         self,
         clickable: ElementHandle,
+        override_filename: str | None = None,
     ) -> DownloadMeta:
         """
         Capture the download of a click event. This will click the element, download the resulting file
@@ -212,7 +213,7 @@ async def capture_download(
         res = await self._notify_observers(
             "on_download",
             download.url,
-            download.suggested_filename,
+            override_filename if override_filename else download.suggested_filename,
             content,
             check_duplication=False,
         )

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "harambe-sdk"
-version = "0.13.1"
+version = "0.13.2"
 description = "Data extraction SDK for Playwright 🐒🍌"
 authors = ["awtkns <[email protected]>"]
 readme = "README.md"