add status function

Author: OasisAkari

akari_bot_webrender/functions/main.py

@@ -81,8 +81,8 @@ def __init__(self,
                  debug: bool = False,
                  remote_webrender_url: Optional[str] = None,
                  remote_only: bool = False,
-                 export_logs = False,
-                 logs_path = None,
+                 export_logs=False,
+                 logs_path=None,
                  name: str = None):
         """
         :param debug: If True, the browser will run on non-headless mode, the page will not be closed after the screenshot is taken.
@@ -93,11 +93,12 @@ def __init__(self,
             self.remote_webrender_url += "/"
         self.remote_only = remote_only
         self.export_logs = export_logs
-        if export_logs: 
+        if export_logs:
             if logs_path:
                 self.logs_path = Path(logs_path)
             else:
-                self.logs_path = (Path(__file__).parent.parent.parent / "logs").resolve()
+                self.logs_path = (
+                    Path(__file__).parent.parent.parent / "logs").resolve()
         if name:
             self.name = name
 
@@ -213,7 +214,8 @@ async def select_element_and_screenshot(self,
                                             page: Page,
                                             start_time: float,
                                             count_time=True,
-                                            output_type: Literal["png", "jpeg"] = "jpeg",
+                                            output_type: Literal["png",
+                                                                 "jpeg"] = "jpeg",
                                             output_quality: int = 90):
         el, selected_ = await self.select_element(elements, page)
         if not el:
@@ -340,3 +342,18 @@ async def source(self, options: SourceOptions):
                 return await _source.inner_text()
 
             return _source
+
+    @webrender_fallback
+    async def status(self):
+        pages_open = 0
+        for context in self.browser.browser.contexts:
+            pages_open += len(context.pages)
+        return {
+            "browser_initialized": self.browser.browser is not None,
+            "debug_mode": self.debug,
+            "remote_only": self.remote_only,
+            "export_logs": self.export_logs,
+            "logs_path": str(self.logs_path) if self.logs_path else None,
+            "name": self.name,
+            "pages_open": pages_open,
+        }

akari_bot_webrender/server/main.py

@@ -16,7 +16,7 @@
 
 webrender = WebRender(debug=config.get("debug", False),
                       export_logs=config.get("export_logs", False)
-                     )
+                      )
 
 
 @asynccontextmanager
@@ -74,6 +74,11 @@ async def source(options: SourceOptions):
     return ORJSONResponse(content=source_content)
 
 
+@app.get("/status/")
+async def status():
+    return ORJSONResponse(content=await webrender.status())
+
+
 @app.get("/favicon.ico")
 async def favicon():
     return FileResponse((Path(__file__).parent / "favicon.ico").resolve())

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "akari-bot-webrender"
-version = "0.4.8"
+version = "0.4.9"
 description = ""
 authors = [
     {name = "OasisAkari", email = "OasisAkari@gmail.com"},