117 lines
3.4 KiB
Python
117 lines
3.4 KiB
Python
"""Support for Speedtest.net internet speed testing sensor."""
|
|
import logging
|
|
|
|
from homeassistant.const import ATTR_ATTRIBUTION
|
|
from homeassistant.core import callback
|
|
from homeassistant.helpers.restore_state import RestoreEntity
|
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
|
|
|
from .const import (
|
|
ATTR_BYTES_RECEIVED,
|
|
ATTR_BYTES_SENT,
|
|
ATTR_SERVER_COUNTRY,
|
|
ATTR_SERVER_ID,
|
|
ATTR_SERVER_NAME,
|
|
ATTRIBUTION,
|
|
DEFAULT_NAME,
|
|
DOMAIN,
|
|
ICON,
|
|
SENSOR_TYPES,
|
|
)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
async def async_setup_entry(hass, config_entry, async_add_entities):
|
|
"""Set up the Speedtestdotnet sensors."""
|
|
|
|
speedtest_coordinator = hass.data[DOMAIN]
|
|
|
|
entities = []
|
|
for sensor_type in SENSOR_TYPES:
|
|
entities.append(SpeedtestSensor(speedtest_coordinator, sensor_type))
|
|
|
|
async_add_entities(entities)
|
|
|
|
|
|
class SpeedtestSensor(CoordinatorEntity, RestoreEntity):
|
|
"""Implementation of a speedtest.net sensor."""
|
|
|
|
def __init__(self, coordinator, sensor_type):
|
|
"""Initialize the sensor."""
|
|
super().__init__(coordinator)
|
|
self._name = SENSOR_TYPES[sensor_type][0]
|
|
self.type = sensor_type
|
|
self._unit_of_measurement = SENSOR_TYPES[self.type][1]
|
|
self._state = None
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the sensor."""
|
|
return f"{DEFAULT_NAME} {self._name}"
|
|
|
|
@property
|
|
def unique_id(self):
|
|
"""Return sensor unique_id."""
|
|
return self.type
|
|
|
|
@property
|
|
def state(self):
|
|
"""Return the state of the device."""
|
|
return self._state
|
|
|
|
@property
|
|
def unit_of_measurement(self):
|
|
"""Return the unit of measurement of this entity, if any."""
|
|
return self._unit_of_measurement
|
|
|
|
@property
|
|
def icon(self):
|
|
"""Return icon."""
|
|
return ICON
|
|
|
|
@property
|
|
def device_state_attributes(self):
|
|
"""Return the state attributes."""
|
|
if not self.coordinator.data:
|
|
return None
|
|
attributes = {
|
|
ATTR_ATTRIBUTION: ATTRIBUTION,
|
|
ATTR_SERVER_NAME: self.coordinator.data["server"]["name"],
|
|
ATTR_SERVER_COUNTRY: self.coordinator.data["server"]["country"],
|
|
ATTR_SERVER_ID: self.coordinator.data["server"]["id"],
|
|
}
|
|
if self.type == "download":
|
|
attributes[ATTR_BYTES_RECEIVED] = self.coordinator.data["bytes_received"]
|
|
|
|
if self.type == "upload":
|
|
attributes[ATTR_BYTES_SENT] = self.coordinator.data["bytes_sent"]
|
|
|
|
return attributes
|
|
|
|
async def async_added_to_hass(self):
|
|
"""Handle entity which will be added."""
|
|
await super().async_added_to_hass()
|
|
state = await self.async_get_last_state()
|
|
if state:
|
|
self._state = state.state
|
|
|
|
@callback
|
|
def update():
|
|
"""Update state."""
|
|
self._update_state()
|
|
self.async_write_ha_state()
|
|
|
|
self.async_on_remove(self.coordinator.async_add_listener(update))
|
|
self._update_state()
|
|
|
|
def _update_state(self):
|
|
"""Update sensors state."""
|
|
if self.coordinator.data:
|
|
if self.type == "ping":
|
|
self._state = self.coordinator.data["ping"]
|
|
elif self.type == "download":
|
|
self._state = round(self.coordinator.data["download"] / 10 ** 6, 2)
|
|
elif self.type == "upload":
|
|
self._state = round(self.coordinator.data["upload"] / 10 ** 6, 2)
|