core/homeassistant/components/somfy/switch.py

49 lines
1.5 KiB
Python

"""Support for Somfy Camera Shutter."""
from pymfy.api.devices.camera_protect import CameraProtect
from pymfy.api.devices.category import Category
from homeassistant.components.switch import SwitchEntity
from .const import COORDINATOR, DOMAIN
from .entity import SomfyEntity
async def async_setup_entry(hass, config_entry, async_add_entities):
"""Set up the Somfy switch platform."""
domain_data = hass.data[DOMAIN]
coordinator = domain_data[COORDINATOR]
switches = [
SomfyCameraShutter(coordinator, device_id)
for device_id, device in coordinator.data.items()
if Category.CAMERA.value in device.categories
]
async_add_entities(switches)
class SomfyCameraShutter(SomfyEntity, SwitchEntity):
"""Representation of a Somfy Camera Shutter device."""
def __init__(self, coordinator, device_id):
"""Initialize the Somfy device."""
super().__init__(coordinator, device_id)
self._create_device()
def _create_device(self):
"""Update the device with the latest data."""
self.shutter = CameraProtect(self.device, self.coordinator.client)
def turn_on(self, **kwargs) -> None:
"""Turn the entity on."""
self.shutter.open_shutter()
def turn_off(self, **kwargs):
"""Turn the entity off."""
self.shutter.close_shutter()
@property
def is_on(self) -> bool:
"""Return True if entity is on."""
return self.shutter.get_shutter_position() == "opened"