"""Sensor for monitoring the size of a file.""" import datetime import logging import os import voluptuous as vol from homeassistant.helpers.entity import Entity import homeassistant.helpers.config_validation as cv from homeassistant.components.sensor import PLATFORM_SCHEMA _LOGGER = logging.getLogger(__name__) CONF_FILE_PATHS = "file_paths" ICON = "mdi:file" PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( {vol.Required(CONF_FILE_PATHS): vol.All(cv.ensure_list, [cv.isfile])} ) def setup_platform(hass, config, add_entities, discovery_info=None): """Set up the file size sensor.""" sensors = [] for path in config.get(CONF_FILE_PATHS): if not hass.config.is_allowed_path(path): _LOGGER.error("Filepath %s is not valid or allowed", path) continue sensors.append(Filesize(path)) if sensors: add_entities(sensors, True) class Filesize(Entity): """Encapsulates file size information.""" def __init__(self, path): """Initialize the data object.""" self._path = path # Need to check its a valid path self._size = None self._last_updated = None self._name = path.split("/")[-1] self._unit_of_measurement = "MB" def update(self): """Update the sensor.""" statinfo = os.stat(self._path) self._size = statinfo.st_size last_updated = datetime.datetime.fromtimestamp(statinfo.st_mtime) self._last_updated = last_updated.isoformat() @property def name(self): """Return the name of the sensor.""" return self._name @property def state(self): """Return the size of the file in MB.""" decimals = 2 state_mb = round(self._size / 1e6, decimals) return state_mb @property def icon(self): """Icon to use in the frontend, if any.""" return ICON @property def device_state_attributes(self): """Return other details about the sensor state.""" attr = { "path": self._path, "last_updated": self._last_updated, "bytes": self._size, } return attr @property def unit_of_measurement(self): """Return the unit of measurement of this entity, if any.""" return self._unit_of_measurement