mirror of
https://github.com/home-assistant/core
synced 2024-09-09 12:51:22 +02:00
![cgtobi](/assets/img/avatar_default.png)
* Remove dependencies and requirements * Revert "Remove dependencies and requirements" This reverts commitfe7171b4cd
. * Remove dependencies and requirements * Revert "Remove dependencies and requirements" This reverts commit391355ee2c
. * Remove dependencies and requirements * Fix flake8 complaints * Fix more flake8 complaints * Revert non-component removals
184 lines
5.5 KiB
Python
184 lines
5.5 KiB
Python
"""Support for Iperf3 network measurement tool."""
|
|
import logging
|
|
from datetime import timedelta
|
|
|
|
import voluptuous as vol
|
|
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
|
|
from homeassistant.const import CONF_MONITORED_CONDITIONS, CONF_PORT, \
|
|
CONF_HOST, CONF_PROTOCOL, CONF_HOSTS, CONF_SCAN_INTERVAL
|
|
from homeassistant.helpers.discovery import async_load_platform
|
|
from homeassistant.helpers.dispatcher import dispatcher_send
|
|
from homeassistant.helpers.event import async_track_time_interval
|
|
|
|
DOMAIN = 'iperf3'
|
|
DATA_UPDATED = '{}_data_updated'.format(DOMAIN)
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
CONF_DURATION = 'duration'
|
|
CONF_PARALLEL = 'parallel'
|
|
CONF_MANUAL = 'manual'
|
|
|
|
DEFAULT_DURATION = 10
|
|
DEFAULT_PORT = 5201
|
|
DEFAULT_PARALLEL = 1
|
|
DEFAULT_PROTOCOL = 'tcp'
|
|
DEFAULT_INTERVAL = timedelta(minutes=60)
|
|
|
|
ATTR_DOWNLOAD = 'download'
|
|
ATTR_UPLOAD = 'upload'
|
|
ATTR_VERSION = 'Version'
|
|
ATTR_HOST = 'host'
|
|
|
|
UNIT_OF_MEASUREMENT = 'Mbit/s'
|
|
|
|
SENSOR_TYPES = {
|
|
ATTR_DOWNLOAD: [ATTR_DOWNLOAD.capitalize(), UNIT_OF_MEASUREMENT],
|
|
ATTR_UPLOAD: [ATTR_UPLOAD.capitalize(), UNIT_OF_MEASUREMENT],
|
|
}
|
|
|
|
PROTOCOLS = ['tcp', 'udp']
|
|
|
|
HOST_CONFIG_SCHEMA = vol.Schema({
|
|
vol.Required(CONF_HOST): cv.string,
|
|
vol.Optional(CONF_PORT, default=DEFAULT_PORT): cv.port,
|
|
vol.Optional(CONF_DURATION, default=DEFAULT_DURATION): vol.Range(5, 10),
|
|
vol.Optional(CONF_PARALLEL, default=DEFAULT_PARALLEL): vol.Range(1, 20),
|
|
vol.Optional(CONF_PROTOCOL, default=DEFAULT_PROTOCOL): vol.In(PROTOCOLS),
|
|
})
|
|
|
|
CONFIG_SCHEMA = vol.Schema({
|
|
DOMAIN: vol.Schema({
|
|
vol.Required(CONF_HOSTS): vol.All(
|
|
cv.ensure_list, [HOST_CONFIG_SCHEMA]
|
|
),
|
|
vol.Optional(CONF_MONITORED_CONDITIONS, default=list(SENSOR_TYPES)):
|
|
vol.All(cv.ensure_list, [vol.In(list(SENSOR_TYPES))]),
|
|
vol.Optional(CONF_SCAN_INTERVAL, default=DEFAULT_INTERVAL): vol.All(
|
|
cv.time_period, cv.positive_timedelta
|
|
),
|
|
vol.Optional(CONF_MANUAL, default=False): cv.boolean,
|
|
})
|
|
}, extra=vol.ALLOW_EXTRA)
|
|
|
|
SERVICE_SCHEMA = vol.Schema({
|
|
vol.Optional(ATTR_HOST, default=None): cv.string,
|
|
})
|
|
|
|
|
|
async def async_setup(hass, config):
|
|
"""Set up the iperf3 component."""
|
|
import iperf3
|
|
|
|
hass.data[DOMAIN] = {}
|
|
|
|
conf = config[DOMAIN]
|
|
for host in conf[CONF_HOSTS]:
|
|
host_name = host[CONF_HOST]
|
|
|
|
client = iperf3.Client()
|
|
client.duration = host[CONF_DURATION]
|
|
client.server_hostname = host_name
|
|
client.port = host[CONF_PORT]
|
|
client.num_streams = host[CONF_PARALLEL]
|
|
client.protocol = host[CONF_PROTOCOL]
|
|
client.verbose = False
|
|
|
|
data = hass.data[DOMAIN][host_name] = Iperf3Data(hass, client)
|
|
|
|
if not conf[CONF_MANUAL]:
|
|
async_track_time_interval(
|
|
hass, data.update, conf[CONF_SCAN_INTERVAL]
|
|
)
|
|
|
|
def update(call):
|
|
"""Service call to manually update the data."""
|
|
called_host = call.data[ATTR_HOST]
|
|
if called_host in hass.data[DOMAIN]:
|
|
hass.data[DOMAIN][called_host].update()
|
|
else:
|
|
for iperf3_host in hass.data[DOMAIN].values():
|
|
iperf3_host.update()
|
|
|
|
hass.services.async_register(
|
|
DOMAIN, 'speedtest', update, schema=SERVICE_SCHEMA
|
|
)
|
|
|
|
hass.async_create_task(
|
|
async_load_platform(
|
|
hass,
|
|
SENSOR_DOMAIN,
|
|
DOMAIN,
|
|
conf[CONF_MONITORED_CONDITIONS],
|
|
config
|
|
)
|
|
)
|
|
|
|
return True
|
|
|
|
|
|
class Iperf3Data:
|
|
"""Get the latest data from iperf3."""
|
|
|
|
def __init__(self, hass, client):
|
|
"""Initialize the data object."""
|
|
self._hass = hass
|
|
self._client = client
|
|
self.data = {
|
|
ATTR_DOWNLOAD: None,
|
|
ATTR_UPLOAD: None,
|
|
ATTR_VERSION: None
|
|
}
|
|
|
|
@property
|
|
def protocol(self):
|
|
"""Return the protocol used for this connection."""
|
|
return self._client.protocol
|
|
|
|
@property
|
|
def host(self):
|
|
"""Return the host connected to."""
|
|
return self._client.server_hostname
|
|
|
|
@property
|
|
def port(self):
|
|
"""Return the port on the host connected to."""
|
|
return self._client.port
|
|
|
|
def update(self, now=None):
|
|
"""Get the latest data from iperf3."""
|
|
if self.protocol == 'udp':
|
|
# UDP only have 1 way attribute
|
|
result = self._run_test(ATTR_DOWNLOAD)
|
|
self.data[ATTR_DOWNLOAD] = self.data[ATTR_UPLOAD] = getattr(
|
|
result, 'Mbps', None)
|
|
self.data[ATTR_VERSION] = getattr(result, 'version', None)
|
|
else:
|
|
result = self._run_test(ATTR_DOWNLOAD)
|
|
self.data[ATTR_DOWNLOAD] = getattr(
|
|
result, 'received_Mbps', None)
|
|
self.data[ATTR_VERSION] = getattr(result, 'version', None)
|
|
self.data[ATTR_UPLOAD] = getattr(
|
|
self._run_test(ATTR_UPLOAD), 'sent_Mbps', None)
|
|
|
|
dispatcher_send(self._hass, DATA_UPDATED, self.host)
|
|
|
|
def _run_test(self, test_type):
|
|
"""Run and return the iperf3 data."""
|
|
self._client.reverse = test_type == ATTR_DOWNLOAD
|
|
try:
|
|
result = self._client.run()
|
|
except (AttributeError, OSError, ValueError) as error:
|
|
_LOGGER.error("Iperf3 error: %s", error)
|
|
return None
|
|
|
|
if result is not None and \
|
|
hasattr(result, 'error') and \
|
|
result.error is not None:
|
|
_LOGGER.error("Iperf3 error: %s", result.error)
|
|
return None
|
|
|
|
return result
|