Prevent thread crashing exception

andchiind · Jan 10, 2024 · 3b314c7 · 3b314c7
1 parent 465342e
commit 3b314c7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/isar_exr/api/energy_robotics_api.py b/src/isar_exr/api/energy_robotics_api.py
@@ -776,12 +776,14 @@ def get_current_site_stage(self, site_id: str) -> str:
             response_dict: dict[str, Any] = self.client.query(
                 dsl_gql(current_site_stage_query), params
             )
+
+            if response_dict["currentSiteStage"] is not None:
+                return response_dict["currentSiteStage"]["id"]
         except Exception as e:
             message: str = "Could not get current site stage"
             self.logger.error(message)
             raise RobotAPIException(
                 error_description=message,
             )
-        if response_dict["currentSiteStage"] is not None:
-            return response_dict["currentSiteStage"]["id"]
+
         return None