jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-08-02 18:18:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix tradfri switch typing

Browse Source
This commit is contained in:
Martin Hjelmare 2022-06-23 14:09:24 +02:00
parent 8774d4ae75
commit ab903f7fea
1 changed files with 7 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
homeassistant/components/tradfri/switch.py
View File

@ -2,7 +2,7 @@
from __future__ import annotations from __future__ import annotations
from typing import Any, cast from typing import Any
from pytradfri.api.aiocoap_api import APIRequestProtocol from pytradfri.api.aiocoap_api import APIRequestProtocol
@ -56,28 +56,24 @@ class TradfriSwitch(TradfriBaseEntity, SwitchEntity):
gateway_id=gateway_id, gateway_id=gateway_id,
) )
self._device_control = self._device.socket_control device_control = self._device.socket_control
self._device_data = self._device_control.sockets[0] assert device_control # socket_control is ensured when creating the entity
self._device_control = device_control
self._device_data = device_control.sockets[0]
def _refresh(self) -> None: def _refresh(self) -> None:
"""Refresh the device.""" """Refresh the device."""
self._device_data = self.coordinator.data.socket_control.sockets[0] self._device_data = self._device_control.sockets[0]
@property @property
def is_on(self) -> bool: def is_on(self) -> bool:
"""Return true if switch is on.""" """Return true if switch is on."""
if not self._device_data: return self._device_data.state
return False
return cast(bool, self._device_data.state)
async def async_turn_off(self, **kwargs: Any) -> None: async def async_turn_off(self, **kwargs: Any) -> None:
"""Instruct the switch to turn off.""" """Instruct the switch to turn off."""
if not self._device_control:
return
await self._api(self._device_control.set_state(False)) await self._api(self._device_control.set_state(False))
async def async_turn_on(self, **kwargs: Any) -> None: async def async_turn_on(self, **kwargs: Any) -> None:
"""Instruct the switch to turn on.""" """Instruct the switch to turn on."""
if not self._device_control:
return
await self._api(self._device_control.set_state(True)) await self._api(self._device_control.set_state(True))