core/homeassistant/components/deluge/switch.py

54 lines
1.8 KiB
Python
Raw Normal View History

"""Support for setting the Deluge BitTorrent client in Pause."""
from __future__ import annotations
2022-03-23 04:01:24 +00:00
from typing import Any
from homeassistant.components.switch import SwitchEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform
from homeassistant.core import HomeAssistant
2022-03-23 04:01:24 +00:00
from homeassistant.helpers import entity_platform
2022-03-23 04:01:24 +00:00
from . import DelugeEntity
from .const import DOMAIN
2022-03-23 04:01:24 +00:00
from .coordinator import DelugeDataUpdateCoordinator
2022-03-23 04:01:24 +00:00
async def async_setup_entry(
hass: HomeAssistant,
entry: ConfigEntry,
async_add_entities: entity_platform.AddEntitiesCallback,
) -> None:
"""Set up the Deluge switch."""
async_add_entities([DelugeSwitch(hass.data[DOMAIN][entry.entry_id])])
2022-03-23 04:01:24 +00:00
class DelugeSwitch(DelugeEntity, SwitchEntity):
"""Representation of a Deluge switch."""
2022-03-23 04:01:24 +00:00
def __init__(self, coordinator: DelugeDataUpdateCoordinator) -> None:
"""Initialize the Deluge switch."""
2022-03-23 04:01:24 +00:00
super().__init__(coordinator)
self._attr_unique_id = f"{coordinator.config_entry.entry_id}_enabled"
2022-03-23 04:01:24 +00:00
def turn_on(self, **kwargs: Any) -> None:
"""Turn the device on."""
2022-03-23 04:01:24 +00:00
torrent_ids = self.coordinator.api.call("core.get_session_state")
self.coordinator.api.call("core.resume_torrent", torrent_ids)
2022-03-23 04:01:24 +00:00
def turn_off(self, **kwargs: Any) -> None:
"""Turn the device off."""
2022-03-23 04:01:24 +00:00
torrent_ids = self.coordinator.api.call("core.get_session_state")
self.coordinator.api.call("core.pause_torrent", torrent_ids)
2022-03-23 04:01:24 +00:00
@property
def is_on(self) -> bool:
"""Return state of the switch."""
if self.coordinator.data:
data: dict = self.coordinator.data[Platform.SWITCH]
for torrent in data.values():
item = torrent.popitem()
if not item[1]:
return True
return False