first jiwoos commit

browser_launch.py

@@ -0,0 +1,58 @@
+from __future__ import annotations
+
+import shutil
+import time
+from pathlib import Path
+
+from playwright.sync_api import Error, Playwright
+
+
+IGNORE_PATTERNS = shutil.ignore_patterns(
+    "Singleton*",
+    "LOCK",
+    "lockfile",
+    "Crashpad*",
+    "BrowserMetrics*",
+)
+
+
+def launch_browser_context(
+    playwright: Playwright,
+    user_data_dir: str,
+    channel: str,
+    headless: bool,
+):
+    launch_kwargs = {
+        "channel": channel,
+        "headless": headless,
+        "args": ["--start-maximized"],
+        "no_viewport": True,
+    }
+    source_dir = Path(user_data_dir)
+    try:
+        return playwright.chromium.launch_persistent_context(
+            user_data_dir=str(source_dir),
+            **launch_kwargs,
+        )
+    except Error as exc:
+        message = str(exc)
+        if "Target page, context or browser has been closed" not in message:
+            raise
+
+        fallback_dir = source_dir.parent / f"{source_dir.name}-runtime-{int(time.time())}"
+        if fallback_dir.exists():
+            shutil.rmtree(fallback_dir, ignore_errors=True)
+        fallback_dir.mkdir(parents=True, exist_ok=True)
+
+        if source_dir.exists():
+            shutil.copytree(
+                source_dir,
+                fallback_dir,
+                dirs_exist_ok=True,
+                ignore=IGNORE_PATTERNS,
+            )
+
+        return playwright.chromium.launch_persistent_context(
+            user_data_dir=str(fallback_dir),
+            **launch_kwargs,
+        )