build: enable mypy

.pre-commit-config.yaml

@@ -48,8 +48,8 @@ repos:
     rev: v2.3.0
     hooks:
       - id: codespell
-  # - repo: https://github.com/pre-commit/mirrors-mypy
-  #   rev: v1.10.0
-  #   hooks:
-  #     - id: mypy
-  #       additional_dependencies: []
+  - repo: https://github.com/pre-commit/mirrors-mypy
+    rev: v1.10.0
+    hooks:
+      - id: mypy
+        additional_dependencies: []

library_test.py

@@ -78,7 +78,9 @@ async def main() -> None:
         account=args.username,
         password=args.password,
     )
-    cloud_keys = await cloud.get_keys(first_device["device_id"])
+    cloud_keys = {}
+    if cloud:
+        cloud_keys = await cloud.get_keys(first_device["device_id"])
     print("-" * 20)
     print("Fist device Cloud info: ", cloud_keys)
 

midealocal/cloud.py

@@ -90,7 +90,7 @@ def __init__(
         self._api_url = api_url
         self._access_token = None
         self._uid = None
-        self._login_id = None
+        self._login_id = ""
 
     def _make_general_data(self):
         return {}
@@ -289,15 +289,11 @@ async def list_appliances(self, home_id) -> dict | None:
                             "model": appliance.get("productModel"),
                             "online": appliance.get("onlineStatus") == "1",
                         }
-                        if (
-                            device_info.get("sn8") is None
-                            or len(device_info.get("sn8")) == 0
-                        ):
+                        sn8 = device_info.get("sn8")
+                        if not sn8 or len(sn8) == 0:
                             device_info["sn8"] = "00000000"
-                        if (
-                            device_info.get("model") is None
-                            or len(device_info.get("model")) == 0
-                        ):
+                        model = device_info.get("model")
+                        if not model or len(model) == 0:
                             device_info["model"] = device_info["sn8"]
                         appliances[int(appliance["applianceCode"])] = device_info
             return appliances
@@ -312,9 +308,10 @@ async def get_device_info(self, device_id: int):
                 model_number = int(response.get("modelNumber", 0))
             except ValueError:
                 model_number = 0
+            model_type = response.get("type")
             device_info = {
                 "name": response.get("name"),
-                "type": int(response.get("type"), 16),
+                "type": int(model_type, 16) if model_type else 0,
                 "sn": (
                     self._security.aes_decrypt(response.get("sn"))
                     if response.get("sn")
@@ -326,9 +323,11 @@ async def get_device_info(self, device_id: int):
                 "model": response.get("productModel"),
                 "online": response.get("onlineStatus") == "1",
             }
-            if device_info.get("sn8") is None or len(device_info.get("sn8")) == 0:
+            sn8 = device_info.get("sn8")
+            if sn8 is None or len(sn8) == 0:
                 device_info["sn8"] = "00000000"
-            if device_info.get("model") is None or len(device_info.get("model")) == 0:
+            model = device_info.get("model")
+            if model is None or len(model) == 0:
                 device_info["model"] = device_info["sn8"]
             return device_info
         return None
@@ -487,8 +486,9 @@ async def list_appliances(self, home_id) -> dict | None:
                     "model": "",
                     "online": appliance.get("onlineStatus") == "1",
                 }
+                serial_num = device_info.get("sn")
                 device_info["sn8"] = (
-                    device_info.get("sn")[9:17] if len(device_info["sn"]) > 17 else ""
+                    serial_num[9:17] if (serial_num and len(serial_num) > 17) else ""
                 )
                 device_info["model"] = device_info.get("sn8")
                 appliances[int(appliance["id"])] = device_info
@@ -645,8 +645,9 @@ async def list_appliances(self, home_id) -> dict | None:
                     "model": "",
                     "online": appliance.get("onlineStatus") == "1",
                 }
+                serial_num = device_info.get("sn")
                 device_info["sn8"] = (
-                    device_info.get("sn")[9:17] if len(device_info["sn"]) > 17 else ""
+                    serial_num[9:17] if (serial_num and len(serial_num) > 17) else ""
                 )
                 device_info["model"] = device_info.get("sn8")
                 appliances[int(appliance["id"])] = device_info

midealocal/device.py

@@ -74,8 +74,8 @@ def __init__(
         self._model = model
         self._subtype = subtype
         self._protocol_version = 0
-        self._updates = []
-        self._unsupported_protocol = []
+        self._updates = [dict]
+        self._unsupported_protocol = [dict]
         self._is_run = False
         self._available = True
         self._appliance_query = True

midealocal/devices/a1/__init__.py

@@ -1,10 +1,13 @@
 import logging
 from .message import MessageQuery, MessageA1Response, MessageSet
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)

midealocal/devices/ac/__init__.py

@@ -12,10 +12,13 @@
     MessageSubProtocolSet,
 )
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)

midealocal/devices/b0/__init__.py

@@ -1,10 +1,13 @@
 import logging
 from .message import MessageQuery01, MessageB0Response
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)

midealocal/devices/b1/__init__.py

@@ -1,10 +1,13 @@
 import logging
 from .message import MessageQuery, MessageB1Response
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)

midealocal/devices/b3/__init__.py

@@ -1,10 +1,13 @@
 import logging
 from .message import MessageQuery, MessageB3Response
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)

midealocal/devices/b4/__init__.py

@@ -1,10 +1,13 @@
 import logging
 from .message import MessageQuery, MessageB4Response
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)

midealocal/devices/b6/__init__.py

@@ -2,10 +2,13 @@
 import json
 from .message import MessageQuery, MessageB6Response, MessageSet
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)

midealocal/devices/bf/__init__.py

@@ -1,10 +1,13 @@
 import logging
 from .message import MessageQuery, MessageBFResponse
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)

midealocal/devices/c2/__init__.py

@@ -2,10 +2,13 @@
 import json
 from .message import MessageQuery, MessageC2Response, MessageSet, MessagePower
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)

midealocal/devices/c3/__init__.py

@@ -7,10 +7,13 @@
     MessageSet,
 )
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)

midealocal/devices/ca/__init__.py

@@ -1,10 +1,13 @@
 import logging
 from .message import MessageQuery, MessageCAResponse
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)

midealocal/devices/cc/__init__.py

@@ -1,10 +1,13 @@
 import logging
 from .message import MessageQuery, MessageSet, MessageCCResponse
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)

midealocal/devices/cd/__init__.py

@@ -1,11 +1,16 @@
 import logging
 import json
+from typing import Any
 from .message import MessageQuery, MessageSet, MessageCDResponse
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)
@@ -64,7 +69,7 @@ def __init__(
                 DeviceAttributes.compressor_status: None,
             },
         )
-        self._fields = {}
+        self._fields = dict[Any, Any]
         self._temperature_step = None
         self._default_temperature_step = 1
         self.set_customize(customize)

midealocal/devices/ce/__init__.py

@@ -2,10 +2,13 @@
 import json
 from .message import MessageQuery, MessageCEResponse, MessageSet
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)

midealocal/devices/cf/__init__.py

@@ -1,10 +1,13 @@
 import logging
 from .message import MessageQuery, MessageCFResponse, MessageSet
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)

midealocal/devices/da/__init__.py

@@ -1,10 +1,13 @@
 import logging
 from .message import MessageQuery, MessagePower, MessageStart, MessageDAResponse
 
-try:
-    from enum import StrEnum
-except ImportError:
+import sys
+
+if sys.version_info < (3, 12):
     from ...backports.enum import StrEnum
+else:
+    from enum import StrEnum
+
 from ...device import MideaDevice
 
 _LOGGER = logging.getLogger(__name__)