From faae7ac9025a583991d69bc18261323652d6887e Mon Sep 17 00:00:00 2001
From: Daofeng Wu <dolphin.w.e@gmail.com>
Date: Tue, 4 Feb 2025 10:52:31 +0900
Subject: [PATCH] feat(playwright): add closed flag

---
 npiai/core/browser/_playwright.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/npiai/core/browser/_playwright.py b/npiai/core/browser/_playwright.py
index ebd5862..13c41af 100644
--- a/npiai/core/browser/_playwright.py
+++ b/npiai/core/browser/_playwright.py
@@ -40,6 +40,7 @@ def _prepare_browser_utils():
 class PlaywrightContext:
     headless: bool
     ready: bool
+    closed: bool
     playwright: Playwright | None
     browser: Browser | None
     context: BrowserContext | None
@@ -63,6 +64,7 @@ def __init__(
         """
         self.headless = headless
         self.ready = False
+        self.closed = False
         self.playwright = None
         self.browser = None
         self.context = None
@@ -86,6 +88,7 @@ async def start(self):
         await self.restore_state(self.storage_state)
 
         self.ready = True
+        self.closed = False
 
     async def get_state(self) -> StorageState:
         return await self.context.storage_state()
@@ -207,3 +210,5 @@ async def stop(self):
         await self.context.close()
         await self.browser.close()
         await self.playwright.stop()
+        self.ready = False
+        self.closed = True