Rework for mypy

Author: CoMPaTech

airos/base.py

@@ -383,15 +383,16 @@ async def login(self) -> None:
         else:
             return
 
-    async def status(self, underived: bool = False) -> AirOSDataModel | dict[str, Any]:
+    async def raw_status(self) -> dict[str, Any]:
+        """Retrieve raw status from the device."""
+        return await self._request_json("GET", self._status_cgi_url, authenticated=True)
+
+    async def status(self) -> AirOSDataModel:
         """Retrieve status from the device."""
         response = await self._request_json(
             "GET", self._status_cgi_url, authenticated=True
         )
 
-        if underived:
-            return response
-
         try:
             adjusted_json = self.derived_data(response)
             return self.data_model.from_dict(adjusted_json)

airos/helpers.py

@@ -37,7 +37,7 @@ async def async_get_firmware_data(
 
     try:
         await detect_device.login()
-        device_data = await detect_device.status(underived=True)
+        device_data = await detect_device.raw_status()
     except (
         AirOSConnectionSetupError,
         AirOSDeviceConnectionError,
@@ -51,8 +51,6 @@ async def async_get_firmware_data(
         _LOGGER.exception("Key data missing from device at %s", host)
         raise
 
-    assert isinstance(device_data, dict)
-
     return {
         "fw_major": AirOS8.get_fw_major(device_data.get("host", {}).get("fwversion")),
         "hostname": device_data.get("host", {}).get("hostname"),