PlaywrightCapture 1.31.6__py3-none-any.whl → 1.32.0__py3-none-any.whl

playwrightcapture/__init__.py

@@ -1,4 +1,6 @@
 from .capture import Capture  # noqa
+from .capture import CaptureResponse  # noqa
+from .capture import SetCookieParam, Cookie  # noqa
 from .helpers import get_devices  # noqa
 from .exceptions import (PlaywrightCaptureException, UnknownPlaywrightDeviceType,  # noqa
                          UnknownPlaywrightBrowser, UnknownPlaywrightDevice,
@@ -6,6 +8,8 @@ from .exceptions import (PlaywrightCaptureException, UnknownPlaywrightDeviceType
 
 __all__ = [
     'Capture',
+    'CaptureResponse',
+    'SetCookieParam', 'Cookie',
     'get_devices',
     'PlaywrightCaptureException',
     'UnknownPlaywrightDeviceType',

playwrightcapture/capture.py

@@ -46,18 +46,6 @@ if sys.version_info < (3, 11):
 else:
     from asyncio import timeout
 
-if sys.version_info < (3, 12):
-    from typing_extensions import TypedDict
-else:
-    from typing import TypedDict
-
-if TYPE_CHECKING:
-    from playwright._impl._api_structures import (SetCookieParam, Geolocation,
-                                                  HttpCredentials, Headers,
-                                                  ViewportSize, Cookie,
-                                                  ProxySettings, StorageState)
-    BROWSER = Literal['chromium', 'firefox', 'webkit']
-
 try:
     if sys.version_info < (3, 10):
         from pydub import AudioSegment  # type: ignore[attr-defined]
@@ -68,6 +56,54 @@ try:
 except ImportError:
     CAN_SOLVE_CAPTCHA = False
 
+# ####
+# This bit is required as long as we need to support python < 3.12, stop removing it, you idiot
+# That's the runtime failure: https://github.com/ail-project/lacus/actions/runs/16447900822/job/46484807036
+# And the MyPy failure: https://github.com/ail-project/LacusCore/actions/runs/16447753492/job/46484287947
+if sys.version_info < (3, 12):
+    from typing_extensions import TypedDict
+else:
+    from typing import TypedDict
+
+# These two classes are copies of te ones in Playwright, we need them until we can discard python 3.11
+# They come from this file:
+#   https://github.com/microsoft/playwright-python/blob/main/playwright/_impl/_api_structures.py
+
+
+class SetCookieParam(TypedDict, total=False):
+    name: str
+    value: str
+    url: str
+    domain: str
+    path: str
+    expires: float
+    httpOnly: bool
+    secure: bool
+    sameSite: Literal["Lax", "None", "Strict"]
+    partitionKey: str
+
+
+class Cookie(TypedDict, total=False):
+    name: str
+    value: str
+    domain: str
+    path: str
+    expires: float
+    httpOnly: bool
+    secure: bool
+    sameSite: Literal["Lax", "None", "Strict"]
+    partitionKey: str
+
+# ####################################
+
+
+if TYPE_CHECKING:
+    from playwright._impl._api_structures import (Geolocation,
+                                                  HttpCredentials, Headers,
+                                                  ViewportSize,
+                                                  ProxySettings, StorageState)
+    BROWSER = Literal['chromium', 'firefox', 'webkit']
+
 
 class CaptureResponse(TypedDict, total=False):
 
@@ -198,8 +234,6 @@ class Capture():
 
     async def __aenter__(self) -> Capture:
         '''Launch the browser'''
-        self._temp_harfile = NamedTemporaryFile(delete=False)
-
         self.playwright = await async_playwright().start()
 
         if self.device_name:
@@ -219,11 +253,12 @@ class Capture():
         # Set of URLs that were captured in that context
         self._already_captured: set[str] = set()
 
+        # Create the temporary file to store the HAR content.
+        self._temp_harfile = NamedTemporaryFile(delete=False, prefix="playwright_capture_har", suffix=".json")
+
         return self
 
     async def __aexit__(self, exc_type: Any, exc: Any, tb: Any) -> bool:
-        if hasattr(self, '_temp_harfile'):
-            os.unlink(self._temp_harfile.name)
 
         try:
             await self.browser.close(reason="Closing browser at the end of the capture.")
@@ -235,6 +270,13 @@ class Capture():
         except Exception as e:
             # this should't happen, but just in case it does...
             self.logger.info(f'Unable to stop playwright: {e}')
+
+        if hasattr(self, '_temp_harfile'):
+            try:
+                os.unlink(self._temp_harfile.name)
+            except Exception as e:
+                self.logger.warning(f'Unable to remove temp HAR file {self._temp_harfile.name}: {e}')
+
         return True
 
     @property
@@ -515,7 +557,9 @@ class Capture():
 
         if self.cookies:
             try:
-                await self.context.add_cookies(self.cookies)
+                # NOTE: Ignore type until we can use python 3.12 + only
+                # playwrightcapture.capture.SetCookieParam == playwright._impl._api_structures.SetCookieParam
+                await self.context.add_cookies(self.cookies)  # type: ignore[arg-type]
             except Exception:
                 self.logger.exception(f'Unable to set cookies: {self.cookies}')
 
@@ -1237,7 +1281,9 @@ class Capture():
 
                 try:
                     async with timeout(15):
-                        to_return['cookies'] = await self.context.cookies()
+                        # NOTE: Ignore type until we can use python 3.12 + only
+                        # playwrightcapture.capture.SetCookieParam == playwright._impl._api_structures.SetCookieParam
+                        to_return['cookies'] = await self.context.cookies()  # type: ignore[typeddict-item]
                 except (TimeoutError, asyncio.TimeoutError):
                     self.logger.warning("Unable to get cookies (timeout).")
                     errors.append("Unable to get the cookies (timeout).")

{playwrightcapture-1.31.6.dist-info → playwrightcapture-1.32.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: PlaywrightCapture
-Version: 1.31.6
+Version: 1.32.0
 Summary: A simple library to capture websites using playwright
 License: BSD-3-Clause
 Author: Raphaël Vinot
@@ -20,9 +20,9 @@ Classifier: Topic :: Security
 Provides-Extra: recaptcha
 Requires-Dist: SpeechRecognition (>=3.14.3) ; extra == "recaptcha"
 Requires-Dist: aiohttp-socks (>=0.10.1)
-Requires-Dist: aiohttp[speedups] (>=3.12.14)
+Requires-Dist: aiohttp[speedups] (>=3.12.15)
 Requires-Dist: async-timeout (>=5.0.1) ; python_version < "3.11"
-Requires-Dist: beautifulsoup4[charset-normalizer,lxml] (>=4.13.4)
+Requires-Dist: beautifulsoup4[charset-normalizer,lxml] (>=4.13.5)
 Requires-Dist: dateparser (>=1.2.2)
 Requires-Dist: dnspython (>=2.7.0,<3.0.0)
 Requires-Dist: playwright (>=1.54.0)

playwrightcapture-1.32.0.dist-info/RECORD

@@ -0,0 +1,10 @@
+playwrightcapture/__init__.py,sha256=NAL0-ymloDBm_ghp1PsefIwRMQmEFKPhn83WVUD7t_0,663
+playwrightcapture/capture.py,sha256=yPGhrhFrwKS7mlTeMw_acYIjlDHZnAR0psOdXKoi4Lw,88922
+playwrightcapture/exceptions.py,sha256=LhGJQCGHzEu7Sx2Dfl28OFeDg1OmrwufFjAWXlxQnEA,366
+playwrightcapture/helpers.py,sha256=Xqs09zHhzAWnpBtQ0A9YAxg80P3Lj7aBj5M2WuEr0so,1843
+playwrightcapture/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+playwrightcapture/socks5dnslookup.py,sha256=ZpOf8tgsRQZi-WDcn9JbbG1bKz9DSfK_jz1l53UI1Ho,4058
+playwrightcapture-1.32.0.dist-info/LICENSE,sha256=uwFc39fTLacBUG-XTuxX6IQKTKhg4z14gWOLt3ex4Ho,1775
+playwrightcapture-1.32.0.dist-info/METADATA,sha256=2xaXTHpeA5DICsbA2k7ygsPrjOGZzLlSiYc6tl6h33E,3285
+playwrightcapture-1.32.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+playwrightcapture-1.32.0.dist-info/RECORD,,

playwrightcapture-1.31.6.dist-info/RECORD

@@ -1,10 +0,0 @@
-playwrightcapture/__init__.py,sha256=F90Y8wYS13tDjgsfjuFrCfmzQfdnH44G-ovuilJfLEE,511
-playwrightcapture/capture.py,sha256=0coNzognowvVhFK87kNTl6d3cacsEEA0Thx8Nz0ZYxY,87183
-playwrightcapture/exceptions.py,sha256=LhGJQCGHzEu7Sx2Dfl28OFeDg1OmrwufFjAWXlxQnEA,366
-playwrightcapture/helpers.py,sha256=Xqs09zHhzAWnpBtQ0A9YAxg80P3Lj7aBj5M2WuEr0so,1843
-playwrightcapture/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-playwrightcapture/socks5dnslookup.py,sha256=ZpOf8tgsRQZi-WDcn9JbbG1bKz9DSfK_jz1l53UI1Ho,4058
-playwrightcapture-1.31.6.dist-info/LICENSE,sha256=uwFc39fTLacBUG-XTuxX6IQKTKhg4z14gWOLt3ex4Ho,1775
-playwrightcapture-1.31.6.dist-info/METADATA,sha256=MuiFWvqR48aKPlOu6mHhohOdygHa869cS-8qasGlUV0,3285
-playwrightcapture-1.31.6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-playwrightcapture-1.31.6.dist-info/RECORD,,