6fc517fdbe
* 0.2.0 of the btsmarthub_devicelist package makes it compatable with BT's home hub 2. The API has changed in the new version so this change also makes the component code compatible with the changes to the library. * Update homeassistant/components/bt_smarthub/device_tracker.py Co-Authored-By: Franck Nijhof <frenck@frenck.nl> * Update homeassistant/components/bt_smarthub/device_tracker.py Co-Authored-By: Franck Nijhof <frenck@frenck.nl> * Remove dep on config in BTSmartHubScanner This should make BTSmartHubScanner easier to test as you can pass in a mock smarthub_client * Black format bt_smarthub Co-authored-by: Franck Nijhof <frenck@frenck.nl>
102 lines
3 KiB
Python
102 lines
3 KiB
Python
"""Support for BT Smart Hub (Sometimes referred to as BT Home Hub 6)."""
|
|
import logging
|
|
|
|
from btsmarthub_devicelist import BTSmartHub
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.device_tracker import (
|
|
DOMAIN,
|
|
PLATFORM_SCHEMA,
|
|
DeviceScanner,
|
|
)
|
|
from homeassistant.const import CONF_HOST
|
|
import homeassistant.helpers.config_validation as cv
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
CONF_DEFAULT_IP = "192.168.1.254"
|
|
CONF_SMARTHUB_MODEL = "smarthub_model"
|
|
|
|
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
|
|
{
|
|
vol.Optional(CONF_HOST, default=CONF_DEFAULT_IP): cv.string,
|
|
vol.Optional(CONF_SMARTHUB_MODEL): vol.In([1, 2]),
|
|
}
|
|
)
|
|
|
|
|
|
def get_scanner(hass, config):
|
|
"""Return a BT Smart Hub scanner if successful."""
|
|
info = config[DOMAIN]
|
|
smarthub_client = BTSmartHub(
|
|
router_ip=info[CONF_HOST], smarthub_model=info.get(CONF_SMARTHUB_MODEL)
|
|
)
|
|
|
|
scanner = BTSmartHubScanner(smarthub_client)
|
|
|
|
return scanner if scanner.success_init else None
|
|
|
|
|
|
class BTSmartHubScanner(DeviceScanner):
|
|
"""This class queries a BT Smart Hub."""
|
|
|
|
def __init__(self, smarthub_client):
|
|
"""Initialise the scanner."""
|
|
self.smarthub = smarthub_client
|
|
self.last_results = {}
|
|
self.success_init = False
|
|
|
|
# Test the router is accessible
|
|
data = self.get_bt_smarthub_data()
|
|
if data:
|
|
self.success_init = True
|
|
else:
|
|
_LOGGER.info("Failed to connect to %s", self.smarthub.router_ip)
|
|
|
|
def scan_devices(self):
|
|
"""Scan for new devices and return a list with found device IDs."""
|
|
self._update_info()
|
|
return [client["mac"] for client in self.last_results]
|
|
|
|
def get_device_name(self, device):
|
|
"""Return the name of the given device or None if we don't know."""
|
|
if not self.last_results:
|
|
return None
|
|
for client in self.last_results:
|
|
if client["mac"] == device:
|
|
return client["host"]
|
|
return None
|
|
|
|
def _update_info(self):
|
|
"""Ensure the information from the BT Smart Hub is up to date."""
|
|
if not self.success_init:
|
|
return
|
|
|
|
_LOGGER.info("Scanning")
|
|
data = self.get_bt_smarthub_data()
|
|
if not data:
|
|
_LOGGER.warning("Error scanning devices")
|
|
return
|
|
|
|
clients = list(data.values())
|
|
self.last_results = clients
|
|
|
|
def get_bt_smarthub_data(self):
|
|
"""Retrieve data from BT Smart Hub and return parsed result."""
|
|
|
|
# Request data from bt smarthub into a list of dicts.
|
|
data = self.smarthub.get_devicelist(only_active_devices=True)
|
|
|
|
# Renaming keys from parsed result.
|
|
devices = {}
|
|
for device in data:
|
|
try:
|
|
devices[device["UserHostName"]] = {
|
|
"ip": device["IPAddress"],
|
|
"mac": device["PhysAddress"],
|
|
"host": device["UserHostName"],
|
|
"status": device["Active"],
|
|
}
|
|
except KeyError:
|
|
pass
|
|
return devices
|