96 lines
2.7 KiB
Python
96 lines
2.7 KiB
Python
"""
|
|
Support for Tesla cars.
|
|
|
|
For more details about this component, please refer to the documentation at
|
|
https://home-assistant.io/components/tesla/
|
|
"""
|
|
from collections import defaultdict
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.const import (
|
|
ATTR_BATTERY_LEVEL, CONF_USERNAME, CONF_PASSWORD, CONF_SCAN_INTERVAL)
|
|
from homeassistant.helpers import discovery
|
|
from homeassistant.helpers import config_validation as cv
|
|
from homeassistant.helpers.entity import Entity
|
|
from homeassistant.util import slugify
|
|
|
|
REQUIREMENTS = ['teslajsonpy==0.0.11']
|
|
|
|
DOMAIN = 'tesla'
|
|
|
|
TESLA_ID_FORMAT = '{}_{}'
|
|
TESLA_ID_LIST_SCHEMA = vol.Schema([int])
|
|
|
|
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)
|
|
|
|
TESLA_COMPONENTS = [
|
|
'sensor', 'lock', 'climate', 'binary_sensor', 'device_tracker'
|
|
]
|
|
|
|
|
|
def setup(hass, base_config):
|
|
"""Set up of Tesla platform."""
|
|
from teslajsonpy.controller import Controller as teslaApi
|
|
|
|
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:
|
|
hass.data[DOMAIN] = {
|
|
'controller': teslaApi(email, password, update_interval),
|
|
'devices': defaultdict(list)
|
|
}
|
|
|
|
all_devices = hass.data[DOMAIN]['controller'].list_vehicles()
|
|
|
|
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:
|
|
discovery.load_platform(hass, component, DOMAIN, {}, base_config)
|
|
|
|
return True
|
|
|
|
|
|
class TeslaDevice(Entity):
|
|
"""Representation of a Tesla device."""
|
|
|
|
def __init__(self, tesla_device, controller):
|
|
"""Initialisation of 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)
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the device."""
|
|
return self._name
|
|
|
|
@property
|
|
def should_poll(self):
|
|
"""Get polling requirement from tesla device."""
|
|
return self.tesla_device.should_poll
|
|
|
|
@property
|
|
def device_state_attributes(self):
|
|
"""Return the state attributes of the device."""
|
|
attr = {}
|
|
|
|
if self.tesla_device.has_battery():
|
|
attr[ATTR_BATTERY_LEVEL] = self.tesla_device.battery_level()
|
|
return attr
|