From 634b41af4733c44daf67febef0fe65d091187adb Mon Sep 17 00:00:00 2001 From: Katie Mulliken Date: Sun, 8 Dec 2024 18:37:02 -0500 Subject: [PATCH] fix: alarm_state replaces state and added typehints --- custom_components/wyzeapi/alarm_control_panel.py | 12 ++++++------ custom_components/wyzeapi/config_flow.py | 2 +- 2 files changed, 7 insertions(+), 7 deletions(-) diff --git a/custom_components/wyzeapi/alarm_control_panel.py b/custom_components/wyzeapi/alarm_control_panel.py index 4fc38996..5651eac4 100644 --- a/custom_components/wyzeapi/alarm_control_panel.py +++ b/custom_components/wyzeapi/alarm_control_panel.py @@ -65,9 +65,9 @@ def __init__(self, hms_service: HMSService): self._hms_service = hms_service self._state = AlarmControlPanelState.DISARMED self._server_out_of_sync = False - + @property - def state(self): + def alarm_state(self) -> str: return self._state # NotImplemented Methods @@ -151,13 +151,13 @@ async def async_update(self) -> None: if not self._server_out_of_sync: state = await self._hms_service.update(self._hms_service.hms_id) if state is HMSMode.DISARMED: - self._state = "disarmed" + self._state = AlarmControlPanelState.DISARMED elif state is HMSMode.AWAY: - self._state = "armed_away" + self._state = AlarmControlPanelState.ARMED_AWAY elif state is HMSMode.HOME: - self._state = "armed_home" + self._state = AlarmControlPanelState.ARMED_HOME elif state is HMSMode.CHANGING: - self._state = "disarmed" + self._state = AlarmControlPanelState.DISARMED else: _LOGGER.warning(f"Received {state} from server") diff --git a/custom_components/wyzeapi/config_flow.py b/custom_components/wyzeapi/config_flow.py index cb9b7c91..2af729cd 100644 --- a/custom_components/wyzeapi/config_flow.py +++ b/custom_components/wyzeapi/config_flow.py @@ -88,7 +88,7 @@ async def async_step_user( step_id="user", data_schema=STEP_USER_DATA_SCHEMA, errors=errors ) - async def async_step_2fa(self, user_input: Dict[str, Any] = None) -> Dict[str, Any]: + async def async_step_2fa(self, user_input: Optional[dict[str, Any]] = None) -> dict[str, Any]: if user_input is None: return self.async_show_form(step_id="2fa", data_schema=STEP_2FA_DATA_SCHEMA)