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

Align Switcher cover platform with changes from light platform (#130094)

Browse Source 
Switcher small fix for cover
This commit is contained in:
YogevBokobza
2024-11-08 02:59:30 +02:00
committed by GitHub
parent dac6271e01
commit 0d19e85a0d
1 changed files with 12 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
homeassistant/components/switcher_kis/cover.py
View File

@@ -41,16 +41,20 @@ async def async_setup_entry(
def async_add_cover(coordinator: SwitcherDataUpdateCoordinator) -> None:
"""Add cover from Switcher device."""
entities: list[CoverEntity] = []
if coordinator.data.device_type.category in (
DeviceCategory.SHUTTER,
DeviceCategory.SINGLE_SHUTTER_DUAL_LIGHT,
DeviceCategory.DUAL_SHUTTER_SINGLE_LIGHT,
):
entities.append(SwitcherSingleCoverEntity(coordinator, 0))
if (
coordinator.data.device_type.category
== DeviceCategory.DUAL_SHUTTER_SINGLE_LIGHT
):
entities.extend(SwitcherDualCoverEntity(coordinator, i) for i in range(2))
number_of_covers = len(cast(SwitcherShutter, coordinator.data).position)
if number_of_covers == 1:
entities.append(SwitcherSingleCoverEntity(coordinator, 0))
else:
entities.extend(
SwitcherMultiCoverEntity(coordinator, i)
for i in range(number_of_covers)
)
async_add_entities(entities)
config_entry.async_on_unload(
@@ -152,8 +156,8 @@ class SwitcherSingleCoverEntity(SwitcherBaseCoverEntity):
self._update_data()
class SwitcherDualCoverEntity(SwitcherBaseCoverEntity):
"""Representation of a Switcher dual cover entity."""
class SwitcherMultiCoverEntity(SwitcherBaseCoverEntity):
"""Representation of a Switcher multiple cover entity."""
_attr_translation_key = "cover"