"""Support for Obihai Sensors.""" from __future__ import annotations from datetime import timedelta import voluptuous as vol from homeassistant.components.sensor import ( PLATFORM_SCHEMA, SensorDeviceClass, SensorEntity, ) from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME from homeassistant.core import HomeAssistant from homeassistant.helpers import issue_registry import homeassistant.helpers.config_validation as cv from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from .connectivity import ObihaiConnection from .const import DEFAULT_PASSWORD, DEFAULT_USERNAME, DOMAIN, OBIHAI SCAN_INTERVAL = timedelta(seconds=5) PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( { vol.Required(CONF_HOST): cv.string, vol.Optional(CONF_USERNAME, default=DEFAULT_USERNAME): cv.string, vol.Optional(CONF_PASSWORD, default=DEFAULT_PASSWORD): cv.string, } ) # DEPRECATED async def async_setup_platform( hass: HomeAssistant, config: ConfigType, async_add_entities: AddEntitiesCallback, discovery_info: DiscoveryInfoType | None = None, ) -> None: """Set up the Obihai sensor platform.""" issue_registry.async_create_issue( hass, DOMAIN, "manual_migration", breaks_in_ha_version="2023.6.0", is_fixable=False, severity=issue_registry.IssueSeverity.ERROR, translation_key="manual_migration", ) hass.async_create_task( hass.config_entries.flow.async_init( DOMAIN, context={"source": SOURCE_IMPORT}, data=config, ) ) async def async_setup_entry( hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback ) -> None: """Set up the Obihai sensor entries.""" username = entry.data[CONF_USERNAME] password = entry.data[CONF_PASSWORD] host = entry.data[CONF_HOST] requester = ObihaiConnection(host, username, password) await hass.async_add_executor_job(requester.update) sensors = [] for key in requester.services: sensors.append(ObihaiServiceSensors(requester.pyobihai, requester.serial, key)) if requester.line_services is not None: for key in requester.line_services: sensors.append( ObihaiServiceSensors(requester.pyobihai, requester.serial, key) ) for key in requester.call_direction: sensors.append(ObihaiServiceSensors(requester.pyobihai, requester.serial, key)) async_add_entities(sensors, update_before_add=True) class ObihaiServiceSensors(SensorEntity): """Get the status of each Obihai Lines.""" def __init__(self, pyobihai, serial, service_name): """Initialize monitor sensor.""" self._service_name = service_name self._state = None self._name = f"{OBIHAI} {self._service_name}" self._pyobihai = pyobihai self._unique_id = f"{serial}-{self._service_name}" @property def name(self): """Return the name of the sensor.""" return self._name @property def native_value(self): """Return the state of the sensor.""" return self._state @property def available(self): """Return if sensor is available.""" if self._state is not None: return True return False @property def unique_id(self): """Return the unique ID.""" return self._unique_id @property def device_class(self): """Return the device class for uptime sensor.""" if self._service_name == "Last Reboot": return SensorDeviceClass.TIMESTAMP return None @property def icon(self): """Return an icon.""" if self._service_name == "Call Direction": if self._state == "No Active Calls": return "mdi:phone-off" if self._state == "Inbound Call": return "mdi:phone-incoming" return "mdi:phone-outgoing" if "Caller Info" in self._service_name: return "mdi:phone-log" if "Port" in self._service_name: if self._state == "Ringing": return "mdi:phone-ring" if self._state == "Off Hook": return "mdi:phone-in-talk" return "mdi:phone-hangup" if "Service Status" in self._service_name: if "OBiTALK Service Status" in self._service_name: return "mdi:phone-check" if self._state == "0": return "mdi:phone-hangup" return "mdi:phone-in-talk" if "Reboot Required" in self._service_name: if self._state == "false": return "mdi:restart-off" return "mdi:restart-alert" return "mdi:phone" def update(self) -> None: """Update the sensor.""" if not self._pyobihai.check_account(): self._state = None return services = self._pyobihai.get_state() if self._service_name in services: self._state = services.get(self._service_name) services = self._pyobihai.get_line_state() if services is not None and self._service_name in services: self._state = services.get(self._service_name) call_direction = self._pyobihai.get_call_direction() if self._service_name in call_direction: self._state = call_direction.get(self._service_name)