PlaywrightCapture 1.28.0__py3-none-any.whl → 1.28.2__py3-none-any.whl

playwrightcapture/capture.py

@@ -17,7 +17,7 @@ from dataclasses import dataclass
 from io import BytesIO
 from logging import LoggerAdapter, Logger
 from tempfile import NamedTemporaryFile
-from typing import Any, TypedDict, Literal, TYPE_CHECKING
+from typing import Any, Literal, TYPE_CHECKING
 from collections.abc import MutableMapping, Iterator
 from urllib.parse import urlparse, unquote, urljoin, urlsplit, urlunsplit
 from zipfile import ZipFile
@@ -46,11 +46,16 @@ 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)
+                                                  ProxySettings, StorageState)
     BROWSER = Literal['chromium', 'firefox', 'webkit']
 
 try:
@@ -66,6 +71,7 @@ class CaptureResponse(TypedDict, total=False):
     last_redirected_url: str
     har: dict[str, Any] | None
     cookies: list[Cookie] | None
+    storage: StorageState | None
     error: str | None
     error_name: str | None
     html: str | None
@@ -186,6 +192,7 @@ class Capture():
         self.should_retry: bool = False
         self.__network_not_idle: int = 2  # makes sure we do not wait for network idle the max amount of time the capture is allowed to take
         self._cookies: list[SetCookieParam] = []
+        self._storage: StorageState = {}
         self._http_credentials: HttpCredentials = {}
         self._geolocation: Geolocation = {}
         self._headers: Headers = {}
@@ -358,6 +365,18 @@ class Capture():
                 path = cookie.get("path")
                 self.logger.warning(f'The cookie must have a URL ({url}) or a domain ({domain}) and a path ({path})')
 
+    @property
+    def storage(self) -> StorageState:
+        return self._storage
+
+    @storage.setter
+    def storage(self, storage: dict[str, Any] | None) -> None:
+        if not storage:
+            return
+        if 'cookies' in storage and 'origins' in storage:
+            self._storage['cookies'] = storage['cookies']
+            self._storage['origins'] = storage['origins']
+
     @property
     def headers(self) -> Headers:
         return self._headers
@@ -458,6 +477,7 @@ class Capture():
             timezone_id=self.timezone_id if self.timezone_id else None,
             color_scheme=self.color_scheme if self.color_scheme else None,
             viewport=vp,
+            storage_state=self.storage if self.storage else None,
             # For debug only
             # record_video_dir='./videos/',
             **device_context_settings
@@ -1177,6 +1197,7 @@ class Capture():
             self.logger.debug('Finishing up capture.')
             if not capturing_sub:
                 try:
+                    to_return['storage'] = await self.context.storage_state(indexed_db=True)
                     to_return['cookies'] = await self.context.cookies()
                     self.logger.debug('Done with cookies.')
                 except Exception as e:

playwrightcapture/helpers.py

@@ -2,13 +2,19 @@
 
 from __future__ import annotations
 
+import sys
+
 from collections import defaultdict
-from typing import TypedDict
 
 from playwright.sync_api import sync_playwright
 
 from .exceptions import UnknownPlaywrightDeviceType
 
+if sys.version_info < (3, 12):
+    from typing_extensions import TypedDict
+else:
+    from typing import TypedDict
+
 
 class PlaywrightDevice(TypedDict):
 
@@ -22,9 +28,8 @@ class PlaywrightDevice(TypedDict):
 
 def get_devices(in_testsuite: bool=False) -> dict[str, dict[str, dict[str, PlaywrightDevice]]]:
     to_return: dict[str, dict[str, dict[str, PlaywrightDevice]]] = {'desktop': defaultdict(dict), 'mobile': defaultdict(dict)}
-    playwright = sync_playwright().start()
-    devices: dict[str, PlaywrightDevice] = playwright.devices
-    playwright.stop()
+    with sync_playwright() as playwright:
+        devices: dict[str, PlaywrightDevice] = playwright.devices
     for device_name, settings in devices.items():
         splitted_name = device_name.split(' ')
         if splitted_name[0] == 'Desktop':

{playwrightcapture-1.28.0.dist-info → playwrightcapture-1.28.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: PlaywrightCapture
-Version: 1.28.0
+Version: 1.28.2
 Summary: A simple library to capture websites using playwright
 License: BSD-3-Clause
 Author: Raphaël Vinot
@@ -20,15 +20,16 @@ Classifier: Topic :: Security
 Provides-Extra: recaptcha
 Requires-Dist: SpeechRecognition (>=3.14.1) ; extra == "recaptcha"
 Requires-Dist: aiohttp-socks (>=0.10.1)
-Requires-Dist: aiohttp[speedups] (>=3.11.12)
+Requires-Dist: aiohttp[speedups] (>=3.11.14)
 Requires-Dist: async-timeout (>=5.0.1) ; python_version < "3.11"
 Requires-Dist: beautifulsoup4[charset-normalizer,lxml] (>=4.13.3)
 Requires-Dist: dateparser (>=1.2.1)
-Requires-Dist: playwright (>=1.50.0)
+Requires-Dist: playwright (>=1.51.0)
 Requires-Dist: playwright-stealth (>=1.0.6)
 Requires-Dist: puremagic (>=1.28)
 Requires-Dist: pydub (>=0.25.1) ; extra == "recaptcha"
-Requires-Dist: setuptools (>=75.8.0)
+Requires-Dist: setuptools (>=77.0.1)
+Requires-Dist: typing-extensions (>=4.12.2,<5.0.0) ; python_version < "3.12"
 Requires-Dist: tzdata (>=2025.1)
 Requires-Dist: w3lib (>=2.3.1)
 Project-URL: Issues, https://github.com/Lookyloo/PlaywrightCapture/issues

playwrightcapture-1.28.2.dist-info/RECORD

@@ -0,0 +1,9 @@
+playwrightcapture/__init__.py,sha256=F90Y8wYS13tDjgsfjuFrCfmzQfdnH44G-ovuilJfLEE,511
+playwrightcapture/capture.py,sha256=UbZ8kA_JLoWdu2-vxS0-ScV41DEp-WyTk1bYRY5YANA,81515
+playwrightcapture/exceptions.py,sha256=LhGJQCGHzEu7Sx2Dfl28OFeDg1OmrwufFjAWXlxQnEA,366
+playwrightcapture/helpers.py,sha256=Xqs09zHhzAWnpBtQ0A9YAxg80P3Lj7aBj5M2WuEr0so,1843
+playwrightcapture/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+playwrightcapture-1.28.2.dist-info/LICENSE,sha256=uwFc39fTLacBUG-XTuxX6IQKTKhg4z14gWOLt3ex4Ho,1775
+playwrightcapture-1.28.2.dist-info/METADATA,sha256=a_Jpi4OyZcmYkV82L-cHdPTuC_woBOlSnj-VTorJZKI,3075
+playwrightcapture-1.28.2.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+playwrightcapture-1.28.2.dist-info/RECORD,,

playwrightcapture-1.28.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-playwrightcapture/__init__.py,sha256=F90Y8wYS13tDjgsfjuFrCfmzQfdnH44G-ovuilJfLEE,511
-playwrightcapture/capture.py,sha256=0jYIAx5EmdZKGfGaeGhGorc9wIAzDHhgBaBjMkuV5kI,80780
-playwrightcapture/exceptions.py,sha256=LhGJQCGHzEu7Sx2Dfl28OFeDg1OmrwufFjAWXlxQnEA,366
-playwrightcapture/helpers.py,sha256=SXQLEuxMs8-bcWykMiUVosHzzxBKuS-QC0gBV3OmKmo,1764
-playwrightcapture/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-playwrightcapture-1.28.0.dist-info/LICENSE,sha256=uwFc39fTLacBUG-XTuxX6IQKTKhg4z14gWOLt3ex4Ho,1775
-playwrightcapture-1.28.0.dist-info/METADATA,sha256=Gy161otOL3sz3p-kF_IMwcEmqbGl7_Xal2Z3GdvwzTw,2998
-playwrightcapture-1.28.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-playwrightcapture-1.28.0.dist-info/RECORD,,