fix(playwright): add context manager methods

npi-ai · Feb 5, 2025 · 68a09ca · 68a09ca
1 parent f249833
commit 68a09ca
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/npiai/core/browser/_playwright.py b/npiai/core/browser/_playwright.py
@@ -212,3 +212,10 @@ async def stop(self):
         await self.playwright.stop()
         self.ready = False
         self.closed = True
+
+    async def __aenter__(self):
+        await self.start()
+        return self
+
+    async def __aexit__(self, exc_type, exc, tb):
+        await self.stop()