77b6f8c9f2
When speedtest starts up, it would saturate the network interface and cause other integrations to randomly fail to setup. We now wait to do the first speed test until after the started event is fired.
124 lines
3.6 KiB
Python
124 lines
3.6 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 .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(RestoreEntity):
|
|
"""Implementation of a speedtest.net sensor."""
|
|
|
|
def __init__(self, coordinator, sensor_type):
|
|
"""Initialize the sensor."""
|
|
self._name = SENSOR_TYPES[sensor_type][0]
|
|
self.coordinator = coordinator
|
|
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 should_poll(self):
|
|
"""Return the polling requirement for this sensor."""
|
|
return False
|
|
|
|
@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)
|
|
|
|
async def async_update(self):
|
|
"""Request coordinator to update data."""
|
|
await self.coordinator.async_request_refresh()
|