Possibility of using local Playwright (#43)

Author: kiinami

docs/local.md

@@ -0,0 +1,31 @@
+# Local Playwright
+
+In case the Playwright serverless functions are down or you prefer not to use them, you can run the Playwright server locally and request against that.
+
+1. Install this package with the dependencies needed for Playwright:
+
+```bash
+pip install fast-flights[local]
+```
+
+2. Install the Playwright browser:
+
+```bash
+python -m playwright install chromium # or `python -m playwright install` if you want to install all browsers
+```
+
+3. Now you can use the `fetch_mode="local"` parameter in `get_flights`:
+
+```python
+get_flights(
+    ...,
+    fetch_mode="local"  # common/fallback/force-fallback/local
+)
+
+# ...or:
+
+get_fights_from_filter(
+    filter,
+    mode="local"  # common/fallback/force-fallback/local
+)
+```

fast_flights/core.py

@@ -20,7 +20,7 @@ def get_flights_from_filter(
     filter: TFSData,
     currency: str = "",
     *,
-    mode: Literal["common", "fallback", "force-fallback"] = "common",
+    mode: Literal["common", "fallback", "force-fallback", "local"] = "common",
 ) -> Result:
     data = filter.as_b64()
 
@@ -40,6 +40,11 @@ def get_flights_from_filter(
             else:
                 raise e
 
+    elif mode == "local":
+        from .local_playwright import local_playwright_fetch
+
+        res = local_playwright_fetch(params)
+
     else:
         res = fallback_playwright_fetch(params)
 
@@ -57,7 +62,7 @@ def get_flights(
     trip: Literal["round-trip", "one-way", "multi-city"],
     passengers: Passengers,
     seat: Literal["economy", "premium-economy", "business", "first"],
-    fetch_mode: Literal["common", "fallback", "force-fallback"] = "common",
+    fetch_mode: Literal["common", "fallback", "force-fallback", "local"] = "common",
     max_stops: Optional[int] = None,
 ) -> Result:
     return get_flights_from_filter(

fast_flights/local_playwright.py

@@ -0,0 +1,29 @@
+from typing import Any
+import asyncio
+from playwright.async_api import async_playwright
+
+async def fetch_with_playwright(url: str) -> str:
+    async with async_playwright() as p:
+        browser = await p.chromium.launch()
+        page = await browser.new_page()
+        await page.goto(url)
+        if page.url.startswith("https://consent.google.com"):
+            await page.click('text="Accept all"')
+        locator = page.locator('.eQ35Ce')
+        await locator.wait_for()
+        body = await page.evaluate(
+            "() => document.querySelector('[role=\"main\"]').innerHTML"
+        )
+        await browser.close()
+    return body
+
+def local_playwright_fetch(params: dict) -> Any:
+    url = "https://www.google.com/travel/flights?" + "&".join(f"{k}={v}" for k, v in params.items())
+    body = asyncio.run(fetch_with_playwright(url))
+
+    class DummyResponse:
+        status_code = 200
+        text = body
+        text_markdown = body
+
+    return DummyResponse

pyproject.toml

@@ -24,6 +24,11 @@ dependencies = [
   "selectolax"
 ]
 
+[project.optional-dependencies]
+local = [
+    "playwright"
+]
+
 [project.urls]
 "Source" = "https://github.com/AWeirdDev/flights"
 "Issues" = "https://github.com/AWeirdDev/flights/issues"