From 22c295fbdceb441a75d269d0e1e8a6f46b4c0dea Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Thu, 2 Feb 2023 22:30:32 +0100 Subject: [PATCH] Improve typing in ihc util (#87069) --- homeassistant/components/ihc/util.py | 20 +++++++++++++++----- 1 file changed, 15 insertions(+), 5 deletions(-) diff --git a/homeassistant/components/ihc/util.py b/homeassistant/components/ihc/util.py index 8765243e76d..23d9b494e8c 100644 --- a/homeassistant/components/ihc/util.py +++ b/homeassistant/components/ihc/util.py @@ -2,10 +2,14 @@ import asyncio -from homeassistant.core import callback +from ihcsdk.ihccontroller import IHCController + +from homeassistant.core import HomeAssistant, callback -async def async_pulse(hass, ihc_controller, ihc_id: int): +async def async_pulse( + hass: HomeAssistant, ihc_controller: IHCController, ihc_id: int +) -> None: """Send a short on/off pulse to an IHC controller resource.""" await async_set_bool(hass, ihc_controller, ihc_id, True) await asyncio.sleep(0.1) @@ -13,7 +17,9 @@ async def async_pulse(hass, ihc_controller, ihc_id: int): @callback -def async_set_bool(hass, ihc_controller, ihc_id: int, value: bool): +def async_set_bool( + hass: HomeAssistant, ihc_controller: IHCController, ihc_id: int, value: bool +) -> asyncio.Future[bool]: """Set a bool value on an IHC controller resource.""" return hass.async_add_executor_job( ihc_controller.set_runtime_value_bool, ihc_id, value @@ -21,7 +27,9 @@ def async_set_bool(hass, ihc_controller, ihc_id: int, value: bool): @callback -def async_set_int(hass, ihc_controller, ihc_id: int, value: int): +def async_set_int( + hass: HomeAssistant, ihc_controller: IHCController, ihc_id: int, value: int +) -> asyncio.Future[bool]: """Set a int value on an IHC controller resource.""" return hass.async_add_executor_job( ihc_controller.set_runtime_value_int, ihc_id, value @@ -29,7 +37,9 @@ def async_set_int(hass, ihc_controller, ihc_id: int, value: int): @callback -def async_set_float(hass, ihc_controller, ihc_id: int, value: float): +def async_set_float( + hass: HomeAssistant, ihc_controller: IHCController, ihc_id: int, value: float +) -> asyncio.Future[bool]: """Set a float value on an IHC controller resource.""" return hass.async_add_executor_job( ihc_controller.set_runtime_value_float, ihc_id, value