225 lines
8.4 KiB
Python
225 lines
8.4 KiB
Python
"""Support for monitoring the local system."""
|
|
import logging
|
|
import os
|
|
import socket
|
|
|
|
import psutil
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.sensor import PLATFORM_SCHEMA
|
|
from homeassistant.const import CONF_RESOURCES, CONF_TYPE, STATE_OFF, STATE_ON
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.helpers.entity import Entity
|
|
import homeassistant.util.dt as dt_util
|
|
|
|
# mypy: allow-untyped-defs, no-check-untyped-defs
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
CONF_ARG = "arg"
|
|
|
|
SENSOR_TYPES = {
|
|
"disk_free": ["Disk free", "GiB", "mdi:harddisk", None],
|
|
"disk_use": ["Disk use", "GiB", "mdi:harddisk", None],
|
|
"disk_use_percent": ["Disk use (percent)", "%", "mdi:harddisk", None],
|
|
"ipv4_address": ["IPv4 address", "", "mdi:server-network", None],
|
|
"ipv6_address": ["IPv6 address", "", "mdi:server-network", None],
|
|
"last_boot": ["Last boot", "", "mdi:clock", "timestamp"],
|
|
"load_15m": ["Load (15m)", " ", "mdi:memory", None],
|
|
"load_1m": ["Load (1m)", " ", "mdi:memory", None],
|
|
"load_5m": ["Load (5m)", " ", "mdi:memory", None],
|
|
"memory_free": ["Memory free", "MiB", "mdi:memory", None],
|
|
"memory_use": ["Memory use", "MiB", "mdi:memory", None],
|
|
"memory_use_percent": ["Memory use (percent)", "%", "mdi:memory", None],
|
|
"network_in": ["Network in", "MiB", "mdi:server-network", None],
|
|
"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],
|
|
"swap_use": ["Swap use", "MiB", "mdi:harddisk", None],
|
|
"swap_use_percent": ["Swap use (percent)", "%", "mdi:harddisk", None],
|
|
}
|
|
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
|
|
{
|
|
vol.Optional(CONF_RESOURCES, default={CONF_TYPE: "disk_use"}): vol.All(
|
|
cv.ensure_list,
|
|
[
|
|
vol.Schema(
|
|
{
|
|
vol.Required(CONF_TYPE): vol.In(SENSOR_TYPES),
|
|
vol.Optional(CONF_ARG): cv.string,
|
|
}
|
|
)
|
|
],
|
|
)
|
|
}
|
|
)
|
|
|
|
IO_COUNTER = {
|
|
"network_out": 0,
|
|
"network_in": 1,
|
|
"packets_out": 2,
|
|
"packets_in": 3,
|
|
"throughput_network_out": 0,
|
|
"throughput_network_in": 1,
|
|
}
|
|
|
|
IF_ADDRS_FAMILY = {"ipv4_address": socket.AF_INET, "ipv6_address": socket.AF_INET6}
|
|
|
|
|
|
def setup_platform(hass, config, add_entities, discovery_info=None):
|
|
"""Set up the system monitor sensors."""
|
|
dev = []
|
|
for resource in config[CONF_RESOURCES]:
|
|
if CONF_ARG not in resource:
|
|
resource[CONF_ARG] = ""
|
|
dev.append(SystemMonitorSensor(resource[CONF_TYPE], resource[CONF_ARG]))
|
|
|
|
add_entities(dev, True)
|
|
|
|
|
|
class SystemMonitorSensor(Entity):
|
|
"""Implementation of a system monitor sensor."""
|
|
|
|
def __init__(self, sensor_type, argument=""):
|
|
"""Initialize the sensor."""
|
|
self._name = "{} {}".format(SENSOR_TYPES[sensor_type][0], argument)
|
|
self.argument = argument
|
|
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):
|
|
"""Return the name of the sensor."""
|
|
return self._name.rstrip()
|
|
|
|
@property
|
|
def device_class(self):
|
|
"""Return the class of this sensor."""
|
|
return SENSOR_TYPES[self.type][3]
|
|
|
|
@property
|
|
def icon(self):
|
|
"""Icon to use in the frontend, if any."""
|
|
return SENSOR_TYPES[self.type][2]
|
|
|
|
@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
|
|
|
|
def update(self):
|
|
"""Get the latest system information."""
|
|
if self.type == "disk_use_percent":
|
|
self._state = psutil.disk_usage(self.argument).percent
|
|
elif self.type == "disk_use":
|
|
self._state = round(psutil.disk_usage(self.argument).used / 1024 ** 3, 1)
|
|
elif self.type == "disk_free":
|
|
self._state = round(psutil.disk_usage(self.argument).free / 1024 ** 3, 1)
|
|
elif self.type == "memory_use_percent":
|
|
self._state = psutil.virtual_memory().percent
|
|
elif self.type == "memory_use":
|
|
virtual_memory = psutil.virtual_memory()
|
|
self._state = round(
|
|
(virtual_memory.total - virtual_memory.available) / 1024 ** 2, 1
|
|
)
|
|
elif self.type == "memory_free":
|
|
self._state = round(psutil.virtual_memory().available / 1024 ** 2, 1)
|
|
elif self.type == "swap_use_percent":
|
|
self._state = psutil.swap_memory().percent
|
|
elif self.type == "swap_use":
|
|
self._state = round(psutil.swap_memory().used / 1024 ** 2, 1)
|
|
elif self.type == "swap_free":
|
|
self._state = round(psutil.swap_memory().free / 1024 ** 2, 1)
|
|
elif self.type == "processor_use":
|
|
self._state = round(psutil.cpu_percent(interval=None))
|
|
elif self.type == "process":
|
|
for proc in psutil.process_iter():
|
|
try:
|
|
if self.argument == proc.name():
|
|
self._state = STATE_ON
|
|
return
|
|
except psutil.NoSuchProcess as err:
|
|
_LOGGER.warning(
|
|
"Failed to load process with id: %s, old name: %s",
|
|
err.pid,
|
|
err.name,
|
|
)
|
|
self._state = STATE_OFF
|
|
elif self.type == "network_out" or self.type == "network_in":
|
|
counters = psutil.net_io_counters(pernic=True)
|
|
if self.argument in counters:
|
|
counter = counters[self.argument][IO_COUNTER[self.type]]
|
|
self._state = round(counter / 1024 ** 2, 1)
|
|
else:
|
|
self._state = None
|
|
elif self.type == "packets_out" or self.type == "packets_in":
|
|
counters = psutil.net_io_counters(pernic=True)
|
|
if self.argument in counters:
|
|
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 = dt_util.utcnow()
|
|
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:
|
|
for addr in addresses[self.argument]:
|
|
if addr.family == IF_ADDRS_FAMILY[self.type]:
|
|
self._state = addr.address
|
|
else:
|
|
self._state = None
|
|
elif self.type == "last_boot":
|
|
self._state = dt_util.as_local(
|
|
dt_util.utc_from_timestamp(psutil.boot_time())
|
|
).isoformat()
|
|
elif self.type == "load_1m":
|
|
self._state = round(os.getloadavg()[0], 2)
|
|
elif self.type == "load_5m":
|
|
self._state = round(os.getloadavg()[1], 2)
|
|
elif self.type == "load_15m":
|
|
self._state = round(os.getloadavg()[2], 2)
|