138 lines
4.4 KiB
Python
138 lines
4.4 KiB
Python
"""Support for Tesla cars."""
|
|
from collections import defaultdict
|
|
import logging
|
|
|
|
from teslajsonpy import Controller as teslaAPI, TeslaException
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.const import (
|
|
ATTR_BATTERY_LEVEL,
|
|
CONF_PASSWORD,
|
|
CONF_SCAN_INTERVAL,
|
|
CONF_USERNAME,
|
|
)
|
|
from homeassistant.helpers import aiohttp_client, config_validation as cv, discovery
|
|
from homeassistant.helpers.entity import Entity
|
|
from homeassistant.util import slugify
|
|
|
|
from .const import DOMAIN, TESLA_COMPONENTS
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
CONFIG_SCHEMA = vol.Schema(
|
|
{
|
|
DOMAIN: vol.Schema(
|
|
{
|
|
vol.Required(CONF_USERNAME): cv.string,
|
|
vol.Required(CONF_PASSWORD): cv.string,
|
|
vol.Optional(CONF_SCAN_INTERVAL, default=300): vol.All(
|
|
cv.positive_int, vol.Clamp(min=300)
|
|
),
|
|
}
|
|
)
|
|
},
|
|
extra=vol.ALLOW_EXTRA,
|
|
)
|
|
|
|
NOTIFICATION_ID = "tesla_integration_notification"
|
|
NOTIFICATION_TITLE = "Tesla integration setup"
|
|
|
|
|
|
async def async_setup(hass, base_config):
|
|
"""Set up of Tesla component."""
|
|
config = base_config.get(DOMAIN)
|
|
|
|
email = config.get(CONF_USERNAME)
|
|
password = config.get(CONF_PASSWORD)
|
|
update_interval = config.get(CONF_SCAN_INTERVAL)
|
|
if hass.data.get(DOMAIN) is None:
|
|
try:
|
|
websession = aiohttp_client.async_get_clientsession(hass)
|
|
controller = teslaAPI(
|
|
websession,
|
|
email=email,
|
|
password=password,
|
|
update_interval=update_interval,
|
|
)
|
|
await controller.connect(test_login=False)
|
|
hass.data[DOMAIN] = {"controller": controller, "devices": defaultdict(list)}
|
|
_LOGGER.debug("Connected to the Tesla API.")
|
|
except TeslaException as ex:
|
|
if ex.code == 401:
|
|
hass.components.persistent_notification.create(
|
|
"Error:<br />Please check username and password."
|
|
"You will need to restart Home Assistant after fixing.",
|
|
title=NOTIFICATION_TITLE,
|
|
notification_id=NOTIFICATION_ID,
|
|
)
|
|
else:
|
|
hass.components.persistent_notification.create(
|
|
"Error:<br />Can't communicate with Tesla API.<br />"
|
|
"Error code: {} Reason: {}"
|
|
"You will need to restart Home Assistant after fixing."
|
|
"".format(ex.code, ex.message),
|
|
title=NOTIFICATION_TITLE,
|
|
notification_id=NOTIFICATION_ID,
|
|
)
|
|
_LOGGER.error("Unable to communicate with Tesla API: %s", ex.message)
|
|
return False
|
|
all_devices = controller.get_homeassistant_components()
|
|
if not all_devices:
|
|
return False
|
|
|
|
for device in all_devices:
|
|
hass.data[DOMAIN]["devices"][device.hass_type].append(device)
|
|
|
|
for component in TESLA_COMPONENTS:
|
|
hass.async_create_task(
|
|
discovery.async_load_platform(hass, component, DOMAIN, {}, base_config)
|
|
)
|
|
return True
|
|
|
|
|
|
class TeslaDevice(Entity):
|
|
"""Representation of a Tesla device."""
|
|
|
|
def __init__(self, tesla_device, controller):
|
|
"""Initialise the Tesla device."""
|
|
self.tesla_device = tesla_device
|
|
self.controller = controller
|
|
self._name = self.tesla_device.name
|
|
self.tesla_id = slugify(self.tesla_device.uniq_name)
|
|
self._attributes = {}
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the device."""
|
|
return self._name
|
|
|
|
@property
|
|
def unique_id(self) -> str:
|
|
"""Return a unique ID."""
|
|
return self.tesla_id
|
|
|
|
@property
|
|
def should_poll(self):
|
|
"""Return the polling state."""
|
|
return self.tesla_device.should_poll
|
|
|
|
@property
|
|
def device_state_attributes(self):
|
|
"""Return the state attributes of the device."""
|
|
attr = self._attributes
|
|
if self.tesla_device.has_battery():
|
|
attr[ATTR_BATTERY_LEVEL] = self.tesla_device.battery_level()
|
|
return attr
|
|
|
|
async def async_added_to_hass(self):
|
|
"""Register state update callback."""
|
|
pass
|
|
|
|
async def async_will_remove_from_hass(self):
|
|
"""Prepare for unload."""
|
|
pass
|
|
|
|
async def async_update(self):
|
|
"""Update the state of the device."""
|
|
await self.tesla_device.async_update()
|