Refactor UI dump upload handling to start uploads in the background on server startup

Author: mdshakib007

node_cli.py

@@ -38,7 +38,7 @@
 from Framework.install_handler.android.java import update_java_path
 from settings import ZEUZ_NODE_PRIVATE_RSA_KEYS_DIR
 from Framework.install_handler.long_poll_handler import InstallHandler
-from server.mobile import upload_android_ui_dump, upload_ios_ui_dump
+from server.mobile import start_ui_dump_uploads
 from Framework.install_handler.android.android_sdk import update_android_sdk_path
 
 def adjust_python_path():
@@ -1347,8 +1347,7 @@ async def main():
     update_android_sdk_path()
     update_outdated_modules()
     asyncio.create_task(start_server())
-    asyncio.create_task(upload_android_ui_dump())
-    asyncio.create_task(upload_ios_ui_dump())
+    start_ui_dump_uploads()
     asyncio.create_task(delete_old_automationlog_folders())
     await destroy_session()
 

server/main.py

@@ -8,7 +8,7 @@
 from server.connect import router as connect_router
 from server.evaluator import router as evaluator_router
 from server.node_operator import router as operator_router
-from server.mobile import router as mobile_router, upload_android_ui_dump
+from server.mobile import router as mobile_router, start_ui_dump_uploads
 from server.mac import router as mac_router
 from server.linux import router as linux_router
 from server.installers import router as installers_router
@@ -45,6 +45,10 @@ def main() -> FastAPI:
     v1router.include_router(linux_router)
     v1router.include_router(installers_router)
     app = FastAPI()
+
+    @app.on_event("startup")
+    async def _start_background_uploads():
+        start_ui_dump_uploads()
     app.include_router(v1router)
 
     origins = [

server/mobile.py

@@ -29,6 +29,18 @@
 
 router = APIRouter(prefix="/mobile", tags=["mobile"])
 
+_UPLOAD_TASKS_STARTED = False
+
+
+def start_ui_dump_uploads() -> None:
+    """Start background UI dump uploads once per process."""
+    global _UPLOAD_TASKS_STARTED
+    if _UPLOAD_TASKS_STARTED:
+        return
+    _UPLOAD_TASKS_STARTED = True
+    asyncio.create_task(upload_android_ui_dump())
+    asyncio.create_task(upload_ios_ui_dump())
+
 
 def is_wda_running(port: int) -> bool:
     """Check if WebDriverAgent is running on given port."""
@@ -778,4 +790,4 @@ def is_ios_app_installed(sim_udid: str, bundle_id: str):
 
         return {"installed": False}
     except Exception as e:
-        return {"installed": False, "error": str(e)}
+        return {"installed": False, "error": str(e)}