bug #397: mypy

Author: FabienDanieau

src/reachy_mini/daemon/app/main.py

@@ -161,9 +161,12 @@ def run_app(args: Args) -> None:
         from reachy_mini.media.audio_control_utils import init_respeaker_usb
 
         respeaker = init_respeaker_usb()
-        respeaker.write("REBOOT", [1])
-        respeaker.close()
-        logging.debug("Respeaker rebooted.")
+        if respeaker is None:
+            logging.error("Respeaker device not found. Cannot apply audio fix.")
+        else:
+            respeaker.write("REBOOT", [1])
+            respeaker.close()
+            logging.debug("Respeaker rebooted.")
 
     health_check_event = asyncio.Event()
     app = create_app(args, health_check_event)

src/reachy_mini/media/audio_base.py

@@ -23,7 +23,7 @@ def __init__(self, log_level: str = "INFO") -> None:
         """Initialize the audio device."""
         self.logger = logging.getLogger(__name__)
         self.logger.setLevel(log_level)
-        self._respeaker: ReSpeaker = init_respeaker_usb()
+        self._respeaker: Optional[ReSpeaker] = init_respeaker_usb()
 
     def __del__(self) -> None:
         """Destructor to ensure resources are released."""

src/reachy_mini/media/audio_sounddevice.py

@@ -30,10 +30,10 @@ def __init__(
         self.stream = None
         self._output_stream = None
         self._buffer: List[npt.NDArray[np.float32]] = []
-        self._output_device_id = self.get_device_id(
+        self._output_device_id = self._get_device_id(
             ["Reachy Mini Audio", "respeaker"], device_io_type="output"
         )
-        self._input_device_id = self.get_device_id(
+        self._input_device_id = self._get_device_id(
             ["Reachy Mini Audio", "respeaker"], device_io_type="input"
         )
 
@@ -185,12 +185,17 @@ def _clean_up_thread() -> None:
                 daemon=True,
             ).start()
 
-    def get_device_id(
+    def _get_device_id(
         self, names_contains: List[str], device_io_type: str = "output"
     ) -> int:
-        """Return the output device id whose name contains the given string (case-insensitive).
+        """Return the output device id whose name contains the given strings (case-insensitive).
+
+        Args:
+            names_contains (List[str]): List of strings that should be contained in the device name.
+            device_io_type (str): 'input' or 'output' to specify device type.
 
         If not found, return the default output device id.
+
         """
         devices = sd.query_devices()