formatting fix

harambe/__init__.py

@@ -2,4 +2,11 @@
 from .types import AsyncScraperType, ScrapeResult
 from .utils import PlaywrightUtils
 
-__all__ = ["ScrapeResult", "SDK", "PlaywrightUtils", "AsyncScraperType", "AsyncScraper", "PAGE_PDF_FILENAME"]
+__all__ = [
+    "ScrapeResult",
+    "SDK",
+    "PlaywrightUtils",
+    "AsyncScraperType",
+    "AsyncScraper",
+    "PAGE_PDF_FILENAME",
+]

harambe/core.py

@@ -188,11 +188,16 @@ async def capture_pdf(
         Capture the current page as a pdf and then apply some download handling logic
         from the observer to transform to a usable URL
         """
-        await self.page.wait_for_timeout(1000)  # Allow for some extra time for the page to load
+        await self.page.wait_for_timeout(
+            1000
+        )  # Allow for some extra time for the page to load
         pdf_content = await self.page.pdf()
         file_name = PAGE_PDF_FILENAME
         res = await asyncio.gather(
-            *[o.on_download(self.page.url, file_name, pdf_content) for o in self._observers]
+            *[
+                o.on_download(self.page.url, file_name, pdf_content)
+                for o in self._observers
+            ]
         )
         return res[0]
 

harambe/observer.py

@@ -17,7 +17,9 @@ async def on_queue_url(self, url: URL, context: Dict[str, Any]) -> None:
         raise NotImplementedError()
 
     @abstractmethod
-    async def on_download(self, download_url: str, filename: str, content: bytes) -> "DownloadMeta":
+    async def on_download(
+        self, download_url: str, filename: str, content: bytes
+    ) -> "DownloadMeta":
         raise NotImplementedError()
 
 
@@ -29,7 +31,9 @@ async def on_save_data(self, data: Dict[str, Any]):
     async def on_queue_url(self, url: URL, context: Dict[str, Any]) -> None:
         print(f"Enqueuing: {url} with context {context}")
 
-    async def on_download(self, download_url: str, filename: str, content: bytes) -> "DownloadMeta":
+    async def on_download(
+        self, download_url: str, filename: str, content: bytes
+    ) -> "DownloadMeta":
         print(f"Downloading file: {filename}")  # TODO: use logger
         return {
             "url": f"{download_url}/{quote(filename)}",
@@ -47,7 +51,9 @@ async def on_save_data(self, data: Dict[str, Any]):
     async def on_queue_url(self, url: URL, context: Dict[str, Any]) -> None:
         self._tracker.save_data({"url": url, "context": context})
 
-    async def on_download(self, download_url: str, filename: str, content: bytes) -> "DownloadMeta":
+    async def on_download(
+        self, download_url: str, filename: str, content: bytes
+    ) -> "DownloadMeta":
         data = {
             "url": f"{download_url}/{quote(filename)}",
             "filename": filename,
@@ -68,7 +74,9 @@ async def on_save_data(self, data: Dict[str, Any]):
     async def on_queue_url(self, url: URL, context: Dict[str, Any]) -> None:
         self._urls.append((url, context))
 
-    async def on_download(self, download_url: str, filename: str, content: bytes) -> "DownloadMeta":
+    async def on_download(
+        self, download_url: str, filename: str, content: bytes
+    ) -> "DownloadMeta":
         data = {
             "url": f"{download_url}/{quote(filename)}",
             "filename": filename,