75 lines
2.4 KiB
Python
75 lines
2.4 KiB
Python
"""Support for BMW notifications."""
|
|
import logging
|
|
|
|
from homeassistant.components.notify import (
|
|
ATTR_DATA,
|
|
ATTR_TARGET,
|
|
ATTR_TITLE,
|
|
ATTR_TITLE_DEFAULT,
|
|
BaseNotificationService,
|
|
)
|
|
from homeassistant.const import ATTR_LATITUDE, ATTR_LOCATION, ATTR_LONGITUDE, ATTR_NAME
|
|
|
|
from . import DOMAIN as BMW_DOMAIN
|
|
|
|
ATTR_LAT = "lat"
|
|
ATTR_LOCATION_ATTRIBUTES = ["street", "city", "postal_code", "country"]
|
|
ATTR_LON = "lon"
|
|
ATTR_SUBJECT = "subject"
|
|
ATTR_TEXT = "text"
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
def get_service(hass, config, discovery_info=None):
|
|
"""Get the BMW notification service."""
|
|
accounts = hass.data[BMW_DOMAIN]
|
|
_LOGGER.debug("Found BMW accounts: %s", ", ".join([a.name for a in accounts]))
|
|
svc = BMWNotificationService()
|
|
svc.setup(accounts)
|
|
return svc
|
|
|
|
|
|
class BMWNotificationService(BaseNotificationService):
|
|
"""Send Notifications to BMW."""
|
|
|
|
def __init__(self):
|
|
"""Set up the notification service."""
|
|
self.targets = {}
|
|
|
|
def setup(self, accounts):
|
|
"""Get the BMW vehicle(s) for the account(s)."""
|
|
for account in accounts:
|
|
self.targets.update({v.name: v for v in account.account.vehicles})
|
|
|
|
def send_message(self, message="", **kwargs):
|
|
"""Send a message or POI to the car."""
|
|
for _vehicle in kwargs[ATTR_TARGET]:
|
|
_LOGGER.debug("Sending message to %s", _vehicle.name)
|
|
|
|
# Extract params from data dict
|
|
title = kwargs.get(ATTR_TITLE, ATTR_TITLE_DEFAULT)
|
|
data = kwargs.get(ATTR_DATA)
|
|
|
|
# Check if message is a POI
|
|
if data is not None and ATTR_LOCATION in data:
|
|
location_dict = {
|
|
ATTR_LAT: data[ATTR_LOCATION][ATTR_LATITUDE],
|
|
ATTR_LON: data[ATTR_LOCATION][ATTR_LONGITUDE],
|
|
ATTR_NAME: message,
|
|
}
|
|
# Update dictionary with additional attributes if available
|
|
location_dict.update(
|
|
{
|
|
k: v
|
|
for k, v in data[ATTR_LOCATION].items()
|
|
if k in ATTR_LOCATION_ATTRIBUTES
|
|
}
|
|
)
|
|
|
|
_vehicle.remote_services.trigger_send_poi(location_dict)
|
|
else:
|
|
_vehicle.remote_services.trigger_send_message(
|
|
{ATTR_TEXT: message, ATTR_SUBJECT: title}
|
|
)
|