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 cover typing

Browse Source
This commit is contained in:
Martin Hjelmare 2022-06-23 13:56:51 +02:00
parent 951ae92668
commit e7cc87be4c
1 changed files with 7 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
homeassistant/components/tradfri/cover.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,12 +56,14 @@ class TradfriCover(TradfriBaseEntity, CoverEntity):
gateway_id=gateway_id, gateway_id=gateway_id,
) )
self._device_control = self._device.blind_control device_control = self._device.blind_control
self._device_data = self._device_control.blinds[0] assert device_control # blind_control is ensured when creating the entity
self._device_control = device_control
self._device_data = device_control.blinds[0]
def _refresh(self) -> None: def _refresh(self) -> None:
"""Refresh the device.""" """Refresh the device."""
self._device_data = self.coordinator.data.blind_control.blinds[0] self._device_data = self._device_control.blinds[0]
@property @property
def extra_state_attributes(self) -> dict[str, str] | None: def extra_state_attributes(self) -> dict[str, str] | None:
@ -74,32 +76,22 @@ class TradfriCover(TradfriBaseEntity, CoverEntity):
None is unknown, 0 is closed, 100 is fully open. None is unknown, 0 is closed, 100 is fully open.
""" """
if not self._device_data: return 100 - self._device_data.current_cover_position
return None
return 100 - cast(int, self._device_data.current_cover_position)
async def async_set_cover_position(self, **kwargs: Any) -> None: async def async_set_cover_position(self, **kwargs: Any) -> None:
"""Move the cover to a specific position.""" """Move the cover to a specific position."""
if not self._device_control:
return
await self._api(self._device_control.set_state(100 - kwargs[ATTR_POSITION])) await self._api(self._device_control.set_state(100 - kwargs[ATTR_POSITION]))
async def async_open_cover(self, **kwargs: Any) -> None: async def async_open_cover(self, **kwargs: Any) -> None:
"""Open the cover.""" """Open the cover."""
if not self._device_control:
return
await self._api(self._device_control.set_state(0)) await self._api(self._device_control.set_state(0))
async def async_close_cover(self, **kwargs: Any) -> None: async def async_close_cover(self, **kwargs: Any) -> None:
"""Close cover.""" """Close cover."""
if not self._device_control:
return
await self._api(self._device_control.set_state(100)) await self._api(self._device_control.set_state(100))
async def async_stop_cover(self, **kwargs: Any) -> None: async def async_stop_cover(self, **kwargs: Any) -> None:
"""Close cover.""" """Close cover."""
if not self._device_control:
return
await self._api(self._device_control.trigger_blind()) await self._api(self._device_control.trigger_blind())
@property @property