Update app/core/service/playwright/playwright_context.py

app/core/service/playwright/playwright_context.py

@@ -39,22 +39,24 @@ class AsyncPlaywrightContext:
             self.default_context = await self.browser.new_context()
         return self
 
-    async def new_context_page(
-        self,
-        screenshot_model: GetContentModel,
-        browser: Browser,
-        page_model: PageModel) -> Page:
-        if not screenshot_model.new_browser:
-            return await self.default_context.new_page()
-
-        new_context = await browser.new_context(
-            color_scheme=page_model.color_scheme,
-            java_script_enabled=page_model.java_script_enabled,
-            no_viewport=page_model.no_viewport,
-            proxy=page_model.proxy.model_dump() if page_model.proxy else None,
-            viewport=page_model.viewport.model_dump() if page_model.viewport else None,
-        )
-        return await new_context.new_page()
+   async def new_context_page(
+       self,
+       screenshot_model,
+       browser, page_model) -> Page:
+    params = {
+        'color_scheme': page_model.color_scheme,
+        'java_script_enabled': page_model.java_script_enabled,
+        'no_viewport': page_model.no_viewport,
+        'proxy': page_model.proxy.model_dump() if page_model.proxy else None,
+        'viewport': page_model.viewport.model_dump() if page_model.viewport else None,
+    }
+
+    if not screenshot_model.new_browser:
+        return await self.browser.new_page(**params)
+
+    new_context = await browser.new_context(**params)
+    return await new_context.new_page()
+
 
     async def screenshot(
         self,
@@ -124,4 +126,8 @@ class AsyncPlaywrightContext:
         exc: BaseException | None,
         tb: TracebackType | None,
     ) -> None:
-        print('speed up test exit')
+        print('speed test')
+        # if self.browser:
+        #     await self.browser.close()
+        # if self.playwright:
+        #     await self.playwright.stop()