Fix mypy issues

custom_components/sector/alarm_control_panel.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 import logging
+from typing import TYPE_CHECKING
 
 from homeassistant.components.alarm_control_panel import (
     AlarmControlPanelEntity,
@@ -79,6 +80,8 @@ def alarm_state(self) -> AlarmControlPanelState | None:
 
     async def async_alarm_arm_away(self, code: str | None = None) -> None:
         """Send arm away command."""
+        if TYPE_CHECKING:
+            assert code is not None
         if not self._is_valid_code(code):
             raise ServiceValidationError("Invalid code length")
         _LOGGER.debug("Arming away with code: %s", code)
@@ -87,6 +90,8 @@ async def async_alarm_arm_away(self, code: str | None = None) -> None:
 
     async def async_alarm_arm_home(self, code: str | None = None) -> None:
         """Send arm home command."""
+        if TYPE_CHECKING:
+            assert code is not None
         if not self._is_valid_code(code):
             raise ServiceValidationError("Invalid code length")
         _LOGGER.debug("Arming home with code: %s", code)
@@ -95,6 +100,8 @@ async def async_alarm_arm_home(self, code: str | None = None) -> None:
 
     async def async_alarm_disarm(self, code: str | None = None) -> None:
         """Send disarm command."""
+        if TYPE_CHECKING:
+            assert code is not None
         if not self._is_valid_code(code):
             raise ServiceValidationError("Invalid code length")
         _LOGGER.debug("Disarming with code: %s", code)

custom_components/sector/client.py

@@ -32,7 +32,7 @@ def __init__(self, hass: HomeAssistant, email, password, panel_id):
         self.password = password
         self.panel_id = panel_id
         self.access_token = None
-        self.headers = {}
+        self.headers:dict[str,str] = {}
         self.session = None
         self.data_endpoints = get_data_endpoints(self.panel_id)
         self.action_endpoints = get_action_endpoints()

custom_components/sector/config_flow.py

@@ -58,10 +58,10 @@ class SectorAlarmConfigFlow(ConfigFlow, domain=DOMAIN):
     VERSION = 4
 
     def __init__(self):
-        self.email: str | None = None
-        self.password: str | None = None
-        self.code_format: int | None = None
-        self.panel_ids: dict[str, str] = {}
+        self.email: str | None
+        self.password: str | None
+        self.code_format: int | None
+        self.panel_ids: dict[str, str]
 
     async def async_step_reauth(
         self, entry_data: Mapping[str, Any]

custom_components/sector/coordinator.py

@@ -68,7 +68,7 @@ async def _async_update_data(self) -> dict[str, Any]:
 
     def _process_devices(self, api_data) -> tuple[dict[str, Any], dict[str, Any]]:
         """Process device data from the API, including humidity, closed, and alarm sensors."""
-        devices = {}
+        devices:dict[str,Any] = {}
         panel_status = api_data.get("Panel Status", {})
 
         for category_name, category_data in api_data.items():

custom_components/sector/lock.py

@@ -1,7 +1,7 @@
 """Locks for Sector Alarm."""
 
 import logging
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from homeassistant.components.lock import LockEntity
 from homeassistant.const import ATTR_CODE
@@ -28,7 +28,7 @@ async def async_setup_entry(
 
     for serial_no, device_info in devices.items():
         if device_info.get("model") == "Smart Lock":
-            device_name = device_info.get("name")
+            device_name:str = device_info["name"]
             entities.append(
                 SectorAlarmLock(
                     coordinator, code_format, serial_no, device_name, "Smart Lock"
@@ -78,6 +78,8 @@ def is_locked(self) -> bool:
     async def async_lock(self, **kwargs) -> None:
         """Lock the device."""
         code: str | None = kwargs.get(ATTR_CODE)
+        if TYPE_CHECKING:
+            assert code is not None
         _LOGGER.debug("Lock requested for lock %s. Code: %s", self._serial_no, code)
         success = await self.coordinator.api.lock_door(self._serial_no, code=code)
         if success:
@@ -86,6 +88,8 @@ async def async_lock(self, **kwargs) -> None:
     async def async_unlock(self, **kwargs) -> None:
         """Unlock the device."""
         code: str | None = kwargs.get(ATTR_CODE)
+        if TYPE_CHECKING:
+            assert code is not None
         _LOGGER.debug("Unlock requested for lock %s. Code: %s", self._serial_no, code)
         success = await self.coordinator.api.unlock_door(self._serial_no, code=code)
         if success: