Apply ruff formatting via pre-commit

Author: OdynBrouwer

custom_components/bestway/aws_iot/api.py

@@ -131,7 +131,9 @@ def normalize_aws_state(device_state: dict[str, Any]) -> dict[str, Any]:
             wave_normalized = 50  # Map V02 MEDIUM (40) → V01 Airjet MEDIUM (50)
         else:
             wave_normalized = wave_state  # 0 and 100 are same in both
-        _LOGGER.debug("🔵 normalize_aws_state: OUTPUT wave_normalized=%s", wave_normalized)
+        _LOGGER.debug(
+            "🔵 normalize_aws_state: OUTPUT wave_normalized=%s", wave_normalized
+        )
 
         # Build normalized dict, only including fields with actual values
         # This prevents None values from overwriting existing data during merges
@@ -851,7 +853,7 @@ async def airjet_v01_spa_set_bubbles(
         Try sending absolute values first (simplest approach).
         """
         _LOGGER.debug("🔵 airjet_v01_spa_set_bubbles: input level=%s", level)
-        
+
         # Map BubblesLevel enum to V02 wave_state values
         value_map = {
             BubblesLevel.OFF: 0,
@@ -891,7 +893,9 @@ async def airjet_v01_spa_set_bubbles(
                 pass
             # Now send MEDIUM
             await self.set_device_state(device_id, {"wave_state": 40})
-            _LOGGER.debug("🔵 Set bubbles to %s (wave_state=%d after toggle)", level.name, 40)
+            _LOGGER.debug(
+                "🔵 Set bubbles to %s (wave_state=%d after toggle)", level.name, 40
+            )
             return
 
         # Default: send the requested absolute value
@@ -905,7 +909,10 @@ async def hydrojet_spa_set_bubbles(
 
         V02 uses same toggle approach as Airjet V02.
         """
-        _LOGGER.debug("🔵 hydrojet_spa_set_bubbles: delegating to airjet_v01_spa_set_bubbles with level=%s", level)
+        _LOGGER.debug(
+            "🔵 hydrojet_spa_set_bubbles: delegating to airjet_v01_spa_set_bubbles with level=%s",
+            level,
+        )
         # V02 uses toggle approach (same as Airjet V02)
         await self.airjet_v01_spa_set_bubbles(device_id, level)
 

custom_components/bestway/select.py

@@ -126,10 +126,16 @@ def current_option(self) -> str | None:
         if device := self.coordinator.data.devices.get(self.device_id):
             wave_value = device.attrs.get("wave")
             _LOGGER.debug("🔵 current_option: device.attrs['wave']=%s", wave_value)
-            
-            bubbles_level = self.entity_description.get_fn(wave_value)
+
+            # Ensure we pass an int to the mapping function (mypy-friendly).
+            try:
+                wave_int: int = int(wave_value) if wave_value is not None else 0
+            except Exception:
+                wave_int = 0
+
+            bubbles_level = self.entity_description.get_fn(wave_int)
             _LOGGER.debug("🔵 current_option: mapped to BubblesLevel=%s", bubbles_level)
-            
+
             option = _BUBBLES_OPTIONS.get(bubbles_level)
             _LOGGER.debug("🔵 current_option: final option=%s", option)
             return option
@@ -138,14 +144,16 @@ def current_option(self) -> str | None:
     async def async_select_option(self, option: str) -> None:
         """Change the selected option."""
         _LOGGER.debug("🔵 async_select_option: user selected option=%s", option)
-        
+
         bubbles_level = BubblesLevel.OFF
         if option == _BUBBLES_OPTIONS[BubblesLevel.MEDIUM]:
             bubbles_level = BubblesLevel.MEDIUM
         elif option == _BUBBLES_OPTIONS[BubblesLevel.MAX]:
             bubbles_level = BubblesLevel.MAX
 
-        _LOGGER.debug("🔵 async_select_option: mapped to BubblesLevel=%s", bubbles_level)
+        _LOGGER.debug(
+            "🔵 async_select_option: mapped to BubblesLevel=%s", bubbles_level
+        )
         await self.entity_description.set_fn(
             self.coordinator.api, self.device_id, bubbles_level
         )