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

Add network throughput statistics to systemmonitor sensor (#21575)

Browse source 
* add network throughput

* lint
This commit is contained in:
Diogo Gomes 2019-03-02 10:38:15 +00:00 • committed by Fabian Affolter
parent ed2b9e5483
commit 1ad4779443
1 changed files with 26 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
homeassistant/components/sensor/systemmonitor.py
View file

@ -1,4 +1,5 @@
"""Support for monitoring the local system."""
from datetime import datetime
import logging
import os
import socket
@ -34,6 +35,10 @@ SENSOR_TYPES = {
'network_out': ['Network out', 'MiB', 'mdi:server-network', None],
'packets_in': ['Packets in', ' ', 'mdi:server-network', None],
'packets_out': ['Packets out', ' ', 'mdi:server-network', None],
'throughput_network_in': ['Network throughput in', 'MB/s',
'mdi:server-network', None],
'throughput_network_out': ['Network throughput out', 'MB/s',
'mdi:server-network', None],
'process': ['Process', ' ', 'mdi:memory', None],
'processor_use': ['Processor use', '%', 'mdi:memory', None],
'swap_free': ['Swap free', 'MiB', 'mdi:harddisk', None],
@ -54,6 +59,8 @@ IO_COUNTER = {
'network_in': 1,
'packets_out': 2,
'packets_in': 3,
'throughput_network_out': 0,
'throughput_network_in': 1,
}
IF_ADDRS_FAMILY = {
@ -84,6 +91,9 @@ class SystemMonitorSensor(Entity):
self.type = sensor_type
self._state = None
self._unit_of_measurement = SENSOR_TYPES[sensor_type][1]
if sensor_type in ['throughput_network_out', 'throughput_network_in']:
self._last_value = None
self._last_update_time = None
@property
def name(self):
@ -162,6 +172,22 @@ class SystemMonitorSensor(Entity):
self._state = counters[self.argument][IO_COUNTER[self.type]]
else:
self._state = None
elif self.type == 'throughput_network_out' or\
self.type == 'throughput_network_in':
counters = psutil.net_io_counters(pernic=True)
if self.argument in counters:
counter = counters[self.argument][IO_COUNTER[self.type]]
now = datetime.now()
if self._last_value and self._last_value < counter:
self._state = round(
(counter - self._last_value) / 1000**2 /
(now - self._last_update_time).seconds, 3)
else:
self._state = None
self._last_update_time = now
self._last_value = counter
else:
self._state = None
elif self.type == 'ipv4_address' or self.type == 'ipv6_address':
addresses = psutil.net_if_addrs()
if self.argument in addresses: