assert bitrate value

mrlt8 · Sep 11, 2023 · e913bae · e913bae
1 parent 8b4def2
commit e913bae
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/app/wyzecam/tutk/tutk_protocol.py b/app/wyzecam/tutk/tutk_protocol.py
@@ -553,6 +553,7 @@ def __init__(
         :param bitrate: the bit rate, in KB/s to target in the h264/h265 encoder.
         """
         super().__init__(10052)
+        assert 0 <= bitrate <= 255, "bitrate value must be 1-255"
         self.frame_size = frame_size + 1
         self.bitrate = bitrate
         self.fps = fps
@@ -583,7 +584,7 @@ def __init__(self, value: int = 0):
         self.bitrate = value
 
     def encode(self) -> bytes:
-        return encode(self.code, bytes([self.bitrate, 0, 0, 0, 0]))
+        return encode(self.code, bytes([self.bitrate, 0, 0, 0, 0, 0]))
 
 
 class K10052HorizontalFlip(TutkWyzeProtocolMessage):