mirror of
https://github.com/home-assistant/core.git
synced 2026-04-17 23:53:49 +01:00
124 lines
3.7 KiB
Python
124 lines
3.7 KiB
Python
"""The Subaru integration."""
|
|
|
|
import logging
|
|
|
|
from subarulink import Controller as SubaruAPI, InvalidCredentials, SubaruException
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import (
|
|
CONF_COUNTRY,
|
|
CONF_DEVICE_ID,
|
|
CONF_PASSWORD,
|
|
CONF_PIN,
|
|
CONF_USERNAME,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import ConfigEntryNotReady
|
|
from homeassistant.helpers import aiohttp_client
|
|
from homeassistant.helpers.device_registry import DeviceInfo
|
|
|
|
from .const import (
|
|
DOMAIN,
|
|
ENTRY_CONTROLLER,
|
|
ENTRY_COORDINATOR,
|
|
ENTRY_VEHICLES,
|
|
FETCH_INTERVAL,
|
|
MANUFACTURER,
|
|
PLATFORMS,
|
|
UPDATE_INTERVAL,
|
|
VEHICLE_API_GEN,
|
|
VEHICLE_HAS_EV,
|
|
VEHICLE_HAS_REMOTE_SERVICE,
|
|
VEHICLE_HAS_REMOTE_START,
|
|
VEHICLE_HAS_SAFETY_SERVICE,
|
|
VEHICLE_LAST_UPDATE,
|
|
VEHICLE_MODEL_NAME,
|
|
VEHICLE_MODEL_YEAR,
|
|
VEHICLE_NAME,
|
|
VEHICLE_VIN,
|
|
)
|
|
from .coordinator import SubaruDataUpdateCoordinator
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Set up Subaru from a config entry."""
|
|
config = entry.data
|
|
websession = aiohttp_client.async_create_clientsession(hass)
|
|
try:
|
|
controller = SubaruAPI(
|
|
websession,
|
|
config[CONF_USERNAME],
|
|
config[CONF_PASSWORD],
|
|
config[CONF_DEVICE_ID],
|
|
config[CONF_PIN],
|
|
None,
|
|
config[CONF_COUNTRY],
|
|
update_interval=UPDATE_INTERVAL,
|
|
fetch_interval=FETCH_INTERVAL,
|
|
)
|
|
_LOGGER.debug("Using subarulink %s", controller.version)
|
|
await controller.connect()
|
|
except InvalidCredentials:
|
|
_LOGGER.error("Invalid account")
|
|
return False
|
|
except SubaruException as err:
|
|
raise ConfigEntryNotReady(err.message) from err
|
|
|
|
vehicle_info = {}
|
|
for vin in controller.get_vehicles():
|
|
if controller.get_subscription_status(vin):
|
|
vehicle_info[vin] = get_vehicle_info(controller, vin)
|
|
|
|
coordinator = SubaruDataUpdateCoordinator(
|
|
hass, entry, controller=controller, vehicle_info=vehicle_info
|
|
)
|
|
|
|
await coordinator.async_refresh()
|
|
|
|
hass.data.setdefault(DOMAIN, {})
|
|
hass.data[DOMAIN][entry.entry_id] = {
|
|
ENTRY_CONTROLLER: controller,
|
|
ENTRY_COORDINATOR: coordinator,
|
|
ENTRY_VEHICLES: vehicle_info,
|
|
}
|
|
|
|
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
|
|
|
|
return True
|
|
|
|
|
|
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Unload a config entry."""
|
|
unload_ok = await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
|
|
if unload_ok:
|
|
hass.data[DOMAIN].pop(entry.entry_id)
|
|
return unload_ok
|
|
|
|
|
|
def get_vehicle_info(controller, vin):
|
|
"""Obtain vehicle identifiers and capabilities."""
|
|
return {
|
|
VEHICLE_VIN: vin,
|
|
VEHICLE_MODEL_NAME: controller.get_model_name(vin),
|
|
VEHICLE_MODEL_YEAR: controller.get_model_year(vin),
|
|
VEHICLE_NAME: controller.vin_to_name(vin),
|
|
VEHICLE_HAS_EV: controller.get_ev_status(vin),
|
|
VEHICLE_API_GEN: controller.get_api_gen(vin),
|
|
VEHICLE_HAS_REMOTE_START: controller.get_res_status(vin),
|
|
VEHICLE_HAS_REMOTE_SERVICE: controller.get_remote_status(vin),
|
|
VEHICLE_HAS_SAFETY_SERVICE: controller.get_safety_status(vin),
|
|
VEHICLE_LAST_UPDATE: 0,
|
|
}
|
|
|
|
|
|
def get_device_info(vehicle_info):
|
|
"""Return DeviceInfo object based on vehicle info."""
|
|
return DeviceInfo(
|
|
identifiers={(DOMAIN, vehicle_info[VEHICLE_VIN])},
|
|
manufacturer=MANUFACTURER,
|
|
model=f"{vehicle_info[VEHICLE_MODEL_YEAR]} {vehicle_info[VEHICLE_MODEL_NAME]}",
|
|
name=vehicle_info[VEHICLE_NAME],
|
|
)
|