Skip to content

Commit d7344d9

Browse files
committed
validate frame_format
1 parent aa81138 commit d7344d9

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

src/arduino/app_peripherals/camera/websocket_camera.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
from arduino.app_utils import Logger
2020

2121
from .camera import BaseCamera
22-
from .errors import CameraOpenError
22+
from .errors import CameraConfigError, CameraOpenError
2323

2424
logger = Logger("WebSocketCamera")
2525

@@ -71,6 +71,8 @@ def __init__(
7171
self.protocol = "ws"
7272
self.port = port
7373
self.timeout = timeout
74+
if frame_format not in ["binary", "base64", "json"]:
75+
raise CameraConfigError(f"Invalid frame format: {frame_format}")
7476
self.frame_format = frame_format
7577
self.logger = logger
7678

@@ -270,7 +272,9 @@ def _parse_message(self, message: str | bytes) -> np.ndarray | None:
270272
frame = cv2.imdecode(nparr, cv2.IMREAD_UNCHANGED)
271273
return frame
272274

273-
return None
275+
else:
276+
logger.error(f"Unknown video format: {self.frame_format}")
277+
return None
274278

275279
except Exception as e:
276280
logger.warning(f"Error parsing message: {e}")

0 commit comments

Comments
 (0)