mirror of
https://github.com/home-assistant/core.git
synced 2025-12-24 12:59:34 +00:00
130 lines
3.3 KiB
Python
130 lines
3.3 KiB
Python
"""Support for Obihai Sensors."""
|
|
import logging
|
|
|
|
from datetime import timedelta
|
|
import voluptuous as vol
|
|
|
|
from pyobihai import PyObihai
|
|
|
|
from homeassistant.components.sensor import PLATFORM_SCHEMA
|
|
from homeassistant.const import (
|
|
CONF_HOST,
|
|
CONF_PASSWORD,
|
|
CONF_USERNAME,
|
|
DEVICE_CLASS_TIMESTAMP,
|
|
)
|
|
|
|
from homeassistant.helpers.entity import Entity
|
|
import homeassistant.helpers.config_validation as cv
|
|
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
SCAN_INTERVAL = timedelta(seconds=5)
|
|
|
|
OBIHAI = "Obihai"
|
|
DEFAULT_USERNAME = "admin"
|
|
DEFAULT_PASSWORD = "admin"
|
|
|
|
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,
|
|
}
|
|
)
|
|
|
|
|
|
def setup_platform(hass, config, add_entities, discovery_info=None):
|
|
"""Set up the Obihai sensor platform."""
|
|
|
|
username = config[CONF_USERNAME]
|
|
password = config[CONF_PASSWORD]
|
|
host = config[CONF_HOST]
|
|
|
|
sensors = []
|
|
|
|
pyobihai = PyObihai(host, username, password)
|
|
|
|
login = pyobihai.check_account()
|
|
if not login:
|
|
_LOGGER.error("Invalid credentials")
|
|
return
|
|
|
|
serial = pyobihai.get_device_serial()
|
|
|
|
services = pyobihai.get_state()
|
|
|
|
line_services = pyobihai.get_line_state()
|
|
|
|
call_direction = pyobihai.get_call_direction()
|
|
|
|
for key in services:
|
|
sensors.append(ObihaiServiceSensors(pyobihai, serial, key))
|
|
|
|
for key in line_services:
|
|
sensors.append(ObihaiServiceSensors(pyobihai, serial, key))
|
|
|
|
for key in call_direction:
|
|
sensors.append(ObihaiServiceSensors(pyobihai, serial, key))
|
|
|
|
add_entities(sensors)
|
|
|
|
|
|
class ObihaiServiceSensors(Entity):
|
|
"""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 state(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 DEVICE_CLASS_TIMESTAMP
|
|
return None
|
|
|
|
def update(self):
|
|
"""Update the sensor."""
|
|
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 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)
|