chore: missing typing for AC (midea-lan#85)

Necroneco · Jun 5, 2024 · 75fbd8e · 75fbd8e
1 parent faf7a38
commit 75fbd8e
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/midealocal/devices/ac/__init__.py b/midealocal/devices/ac/__init__.py
@@ -267,7 +267,9 @@ def make_message_uniq_set(self) -> MessageSubProtocolSet | MessageGeneralSet:
 
     def set_attribute(self, attr: str, value: Any) -> None:
         # if nat a sensor
-        message: MessageToggleDisplay | MessageNewProtocolSet | None = None
+        message: (
+            MessageToggleDisplay | MessageNewProtocolSet | MessageGeneralSet | None
+        ) = None
         if attr not in [
             DeviceAttributes.indoor_temperature,
             DeviceAttributes.outdoor_temperature,
@@ -334,7 +336,7 @@ def set_attribute(self, attr: str, value: Any) -> None:
                     )
                     setattr(message, str(self._fresh_air_version), fresh_air)
             elif attr in self._attributes.keys():
-                message = self.make_message_uniq_set()
+                message = self.make_message_set()
                 if attr in [
                     DeviceAttributes.boost_mode,
                     DeviceAttributes.sleep_mode,
@@ -354,15 +356,17 @@ def set_attribute(self, attr: str, value: Any) -> None:
             self.build_send(message)
 
     def set_target_temperature(self, target_temperature: float, mode: int) -> None:
-        message: MessageGeneralSet = self.make_message_uniq_set()
+        message: MessageSubProtocolSet | MessageGeneralSet = (
+            self.make_message_uniq_set()
+        )
         message.target_temperature = target_temperature
         if mode is not None:
             message.power = True
             message.mode = mode
         self.build_send(message)
 
     def set_swing(self, swing_vertical: bool, swing_horizontal: bool) -> None:
-        message: MessageGeneralSet = self.make_message_uniq_set()
+        message: MessageGeneralSet = self.make_message_set()
         message.swing_vertical = swing_vertical
         message.swing_horizontal = swing_horizontal
         self.build_send(message)