split up application

Dockers/puppeteer-api/app/utils/browser_utils.py

@@ -0,0 +1,43 @@
+from pyppeteer import launch
+from app.config import CUSTOM_USER_AGENT
+
+async def wait_for_network_idle(page):
+    """Wait until no network requests are in flight"""
+    await page.waitForNetworkIdle(idleTime=500, timeout=30000)
+
+async def safe_browser_operation(url, operation_func):
+    """Safely perform browser operations with proper cleanup"""
+    browser = None
+    try:
+        browser = await launch(
+            headless=True,
+            executablePath='/usr/bin/google-chrome',
+            args=['--no-sandbox', '--disable-setuid-sandbox'],
+            handleSIGINT=False,
+            handleSIGTERM=False,
+            handleSIGHUP=False
+        )
+
+        # Create new page with timeout
+        page = await browser.newPage()
+        page.setDefaultNavigationTimeout(30000)
+
+        # Set custom user agent
+        await page.setUserAgent(CUSTOM_USER_AGENT)
+
+        # Call the operation function that uses the page
+        result = await operation_func(page)
+
+        # Explicitly close the page
+        await page.close()
+
+        return result
+
+    finally:
+        # Ensure browser is closed properly
+        if browser:
+            try:
+                await browser.close()
+            except Exception as e:
+                print(f"Error closing browser: {e}")
+                # We don't re-raise here to avoid masking the original error