retry failed camera snapshot (#97)

maciej-or · Jul 25, 2023 · c649eb1 · c649eb1
1 parent 8de9131
commit c649eb1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/custom_components/hikvision_next/camera.py b/custom_components/hikvision_next/camera.py
@@ -53,4 +53,8 @@ async def stream_source(self) -> str | None:
 
     async def async_camera_image(self, width: int | None = None, height: int | None = None) -> bytes | None:
         """Return a still image response from the camera."""
-        return await self.isapi.get_camera_image(self.stream_info, width, height)
+        data = await self.isapi.get_camera_image(self.stream_info, width, height)
+        if data.startswith(b'<?xml '):
+            # retry if got XML error response
+            data = await self.isapi.get_camera_image(self.stream_info, width, height)
+        return data