evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Address late feedback on Deluge config flow (#71497)

Browse source 
Address late feedback on Deluge
This commit is contained in:
Robert Hillis 2022-05-21 05:20:37 -04:00 • committed by GitHub
parent 4710ad07c4
commit 72dbca4f5b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 54 additions and 44 deletions
Download patch file Download diff file Expand all files Collapse all files

76
homeassistant/components/deluge/sensor.py
View file

@ -1,6 +1,10 @@
"""Support for monitoring the Deluge BitTorrent client API."""
from __future__ import annotations
from collections.abc import Callable
from dataclasses import dataclass
from typing import Any
from homeassistant.components.sensor import (
SensorEntity,
SensorEntityDescription,
@ -13,25 +17,52 @@ from homeassistant.helpers import entity_platform
from homeassistant.helpers.typing import StateType
from . import DelugeEntity
from .const import DOMAIN
from .const import CURRENT_STATUS, DATA_KEYS, DOMAIN, DOWNLOAD_SPEED, UPLOAD_SPEED
from .coordinator import DelugeDataUpdateCoordinator
SENSOR_TYPES: tuple[SensorEntityDescription, ...] = (
SensorEntityDescription(
key="current_status",
def get_state(data: dict[str, float], key: str) -> str | float:
"""Get current download/upload state."""
upload = data[DATA_KEYS[0]] - data[DATA_KEYS[2]]
download = data[DATA_KEYS[1]] - data[DATA_KEYS[3]]
if key == CURRENT_STATUS:
if upload > 0 and download > 0:
return "Up/Down"
if upload > 0 and download == 0:
return "Seeding"
if upload == 0 and download > 0:
return "Downloading"
return STATE_IDLE
kb_spd = float(upload if key == UPLOAD_SPEED else download) / 1024
return round(kb_spd, 2 if kb_spd < 0.1 else 1)
@dataclass
class DelugeSensorEntityDescription(SensorEntityDescription):
"""Class to describe a Deluge sensor."""
value: Callable[[dict[str, float]], Any] = lambda val: val
SENSOR_TYPES: tuple[DelugeSensorEntityDescription, ...] = (
DelugeSensorEntityDescription(
key=CURRENT_STATUS,
name="Status",
value=lambda data: get_state(data, CURRENT_STATUS),
),
SensorEntityDescription(
key="download_speed",
DelugeSensorEntityDescription(
key=DOWNLOAD_SPEED,
name="Down Speed",
native_unit_of_measurement=DATA_RATE_KILOBYTES_PER_SECOND,
state_class=SensorStateClass.MEASUREMENT,
value=lambda data: get_state(data, DOWNLOAD_SPEED),
),
SensorEntityDescription(
key="upload_speed",
DelugeSensorEntityDescription(
key=UPLOAD_SPEED,
name="Up Speed",
native_unit_of_measurement=DATA_RATE_KILOBYTES_PER_SECOND,
state_class=SensorStateClass.MEASUREMENT,
value=lambda data: get_state(data, UPLOAD_SPEED),
),
)
@ -51,10 +82,12 @@ async def async_setup_entry(
class DelugeSensor(DelugeEntity, SensorEntity):
"""Representation of a Deluge sensor."""
entity_description: DelugeSensorEntityDescription
def __init__(
self,
coordinator: DelugeDataUpdateCoordinator,
description: SensorEntityDescription,
description: DelugeSensorEntityDescription,
) -> None:
"""Initialize the sensor."""
super().__init__(coordinator)
@ -65,27 +98,4 @@ class DelugeSensor(DelugeEntity, SensorEntity):
@property
def native_value(self) -> StateType:
"""Return the state of the sensor."""
if self.coordinator.data:
data = self.coordinator.data[Platform.SENSOR]
upload = data[b"upload_rate"] - data[b"dht_upload_rate"]
download = data[b"download_rate"] - data[b"dht_download_rate"]
if self.entity_description.key == "current_status":
if data:
if upload > 0 and download > 0:
return "Up/Down"
if upload > 0 and download == 0:
return "Seeding"
if upload == 0 and download > 0:
return "Downloading"
return STATE_IDLE
if data:
if self.entity_description.key == "download_speed":
kb_spd = float(download)
kb_spd = kb_spd / 1024
return round(kb_spd, 2 if kb_spd < 0.1 else 1)
if self.entity_description.key == "upload_speed":
kb_spd = float(upload)
kb_spd = kb_spd / 1024
return round(kb_spd, 2 if kb_spd < 0.1 else 1)
return None
return self.entity_description.value(self.coordinator.data[Platform.SENSOR])