b4bac0f7a0
* Remove unnecessary exception re-wraps * Preserve exception chains on re-raise We slap "from cause" to almost all possible cases here. In some cases it could conceivably be better to do "from None" if we really want to hide the cause. However those should be in the minority, and "from cause" should be an improvement over the corresponding raise without a "from" in all cases anyway. The only case where we raise from None here is in plex, where the exception for an original invalid SSL cert is not the root cause for failure to validate a newly fetched one. Follow local convention on exception variable names if there is a consistent one, otherwise `err` to match with majority of codebase. * Fix mistaken re-wrap in homematicip_cloud/hap.py Missed the difference between HmipConnectionError and HmipcConnectionError. * Do not hide original error on plex new cert validation error Original is not the cause for the new one, but showing old in the traceback is useful nevertheless.
114 lines
3.3 KiB
Python
114 lines
3.3 KiB
Python
"""Support for setting the Deluge BitTorrent client in Pause."""
|
|
import logging
|
|
|
|
from deluge_client import DelugeRPCClient, FailedToReconnectException
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.switch import PLATFORM_SCHEMA
|
|
from homeassistant.const import (
|
|
CONF_HOST,
|
|
CONF_NAME,
|
|
CONF_PASSWORD,
|
|
CONF_PORT,
|
|
CONF_USERNAME,
|
|
STATE_OFF,
|
|
STATE_ON,
|
|
)
|
|
from homeassistant.exceptions import PlatformNotReady
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.helpers.entity import ToggleEntity
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
DEFAULT_NAME = "Deluge Switch"
|
|
DEFAULT_PORT = 58846
|
|
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
|
|
{
|
|
vol.Required(CONF_HOST): cv.string,
|
|
vol.Required(CONF_PASSWORD): cv.string,
|
|
vol.Optional(CONF_PORT, default=DEFAULT_PORT): cv.port,
|
|
vol.Required(CONF_USERNAME): cv.string,
|
|
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
|
|
}
|
|
)
|
|
|
|
|
|
def setup_platform(hass, config, add_entities, discovery_info=None):
|
|
"""Set up the Deluge switch."""
|
|
|
|
name = config[CONF_NAME]
|
|
host = config[CONF_HOST]
|
|
username = config[CONF_USERNAME]
|
|
password = config[CONF_PASSWORD]
|
|
port = config[CONF_PORT]
|
|
|
|
deluge_api = DelugeRPCClient(host, port, username, password)
|
|
try:
|
|
deluge_api.connect()
|
|
except ConnectionRefusedError as err:
|
|
_LOGGER.error("Connection to Deluge Daemon failed")
|
|
raise PlatformNotReady from err
|
|
|
|
add_entities([DelugeSwitch(deluge_api, name)])
|
|
|
|
|
|
class DelugeSwitch(ToggleEntity):
|
|
"""Representation of a Deluge switch."""
|
|
|
|
def __init__(self, deluge_client, name):
|
|
"""Initialize the Deluge switch."""
|
|
self._name = name
|
|
self.deluge_client = deluge_client
|
|
self._state = STATE_OFF
|
|
self._available = False
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the switch."""
|
|
return self._name
|
|
|
|
@property
|
|
def state(self):
|
|
"""Return the state of the device."""
|
|
return self._state
|
|
|
|
@property
|
|
def is_on(self):
|
|
"""Return true if device is on."""
|
|
return self._state == STATE_ON
|
|
|
|
@property
|
|
def available(self):
|
|
"""Return true if device is available."""
|
|
return self._available
|
|
|
|
def turn_on(self, **kwargs):
|
|
"""Turn the device on."""
|
|
torrent_ids = self.deluge_client.call("core.get_session_state")
|
|
self.deluge_client.call("core.resume_torrent", torrent_ids)
|
|
|
|
def turn_off(self, **kwargs):
|
|
"""Turn the device off."""
|
|
torrent_ids = self.deluge_client.call("core.get_session_state")
|
|
self.deluge_client.call("core.pause_torrent", torrent_ids)
|
|
|
|
def update(self):
|
|
"""Get the latest data from deluge and updates the state."""
|
|
|
|
try:
|
|
torrent_list = self.deluge_client.call(
|
|
"core.get_torrents_status", {}, ["paused"]
|
|
)
|
|
self._available = True
|
|
except FailedToReconnectException:
|
|
_LOGGER.error("Connection to Deluge Daemon Lost")
|
|
self._available = False
|
|
return
|
|
for torrent in torrent_list.values():
|
|
item = torrent.popitem()
|
|
if not item[1]:
|
|
self._state = STATE_ON
|
|
return
|
|
|
|
self._state = STATE_OFF
|