From 9f8372d8445024813acc5b185241f2d2a440ba41 Mon Sep 17 00:00:00 2001 From: ziesorx Date: Fri, 26 Sep 2025 15:00:24 +0700 Subject: [PATCH] fix: change save image logic --- core/communication/websocket.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/core/communication/websocket.py b/core/communication/websocket.py index 7394280..4e40d2a 100644 --- a/core/communication/websocket.py +++ b/core/communication/websocket.py @@ -549,10 +549,6 @@ class WebSocketHandler: # Update tracking integrations with session ID shared_stream_manager.set_session_id(display_identifier, session_id) - # Save snapshot image after getting sessionId - if session_id: - await self._save_snapshot(display_identifier, session_id) - async def _handle_set_progression_stage(self, message: SetProgressionStageMessage) -> None: """Handle setProgressionStage message.""" display_identifier = message.payload.displayIdentifier @@ -568,6 +564,10 @@ class WebSocketHandler: if session_id: shared_stream_manager.set_progression_stage(session_id, stage) + # Save snapshot image when progression stage is car_fueling + if stage == 'car_fueling' and session_id: + await self._save_snapshot(display_identifier, session_id) + # If stage indicates session is cleared/finished, clear from tracking if stage in ['finished', 'cleared', 'idle']: # Get session ID for this display and clear it