| """Browser session persistence manager.""" |
|
|
| import json |
| from pathlib import Path |
|
|
| from loguru import logger |
|
|
|
|
| class SessionManager: |
| """Manages browser session state (cookies, localStorage) persistence.""" |
|
|
| def __init__(self, profile_path: Path): |
| self.profile_path = Path(profile_path) |
| self.state_file = self.profile_path / "state.json" |
| self.profile_path.mkdir(parents=True, exist_ok=True) |
|
|
| def has_saved_state(self) -> bool: |
| """Check if a saved browser state exists.""" |
| return self.state_file.exists() |
|
|
| def get_state_path(self) -> str: |
| """Get the path to the state file for Playwright.""" |
| return str(self.state_file) |
|
|
| async def save_state(self, context) -> None: |
| """Save browser context state (cookies, localStorage). |
| |
| Args: |
| context: Playwright browser context. |
| """ |
| try: |
| await context.storage_state(path=self.get_state_path()) |
| logger.debug(f"Browser state saved to {self.state_file}") |
| except Exception as e: |
| logger.error(f"Failed to save browser state: {e}") |
|
|
| def load_state_for_context(self) -> dict | None: |
| """Load saved state for use in context creation. |
| |
| Returns: |
| State dict or None if no saved state. |
| """ |
| if not self.has_saved_state(): |
| return None |
| try: |
| with open(self.state_file, "r") as f: |
| return json.load(f) |
| except Exception as e: |
| logger.warning(f"Failed to load browser state: {e}") |
| return None |
|
|
