4c67670566
* Update System Bridge to support version 4.x.x and above Update systembridgeconnector to version 4.0.0.dev4 Update system_bridgeconnector version to 4.0.0.dev6 Refactor WebSocket client handling in config_flow.py Update strings Update data handling Add default field values to SystemBridgeCoordinatorData Add version check and issue creation for unsupported System Bridge versions Update coordinator.py to set disks and memory to None Update system bridge coordinator to use token instead of API key Update systembridgeconnector version to 4.0.0.dev7 Update systembridgeconnector version to 4.0.0.dev8 Update systembridgeconnector version to 4.0.0.dev9 Changes Update units Fix GPU memory calculation in sensor.py Update GPU memory unit of measurement Add translation keys for binary sensor names Cleanup Add async_migrate_entry function for entry migration Update systembridgeconnector version to 4.0.0.dev10 Update systembridgeconnector version to 4.0.0.dev11 Add version check and authentication handling Update token description in strings.json Fix skipping partitions without data in system_bridge sensor Update systembridgeconnector version to 4.0.0.dev12 Update systembridgeconnector version to 4.0.0 Add check for unsupported version of System Bridge Update systembridgeconnector version to 4.0.1 Update debug log message in async_setup_entry function Remove debug log statement Fixes Update key to token Update tests Update tests Remove unused import in test_config_flow.py Remove added missing translations for another PR Refactor CPU power per CPU calculation Make one liner into lambda Refactors Fix exception type in async_setup_entry function Move checks to class and set minor version Remove unnecessary comment in gpu_memory_free function Remove translation_key for memory_used_percentage sensor Reverse string change Update token placeholder in strings.json Remove suggested_display_precision from sensor descriptions Remove suggested_display_precision from GPU sensor setup Refactor sensor code * Update migrate entry * Refactor GPU-related functions to use a decorator * Move per cpu functions to use decorator * Refactor functions to use decorators for data availability * Remove CONF_API_KEY from config entry data * Add test for migration * Refactor import statement in test_config_flow.py
76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
"""Support for System Bridge notification service."""
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Any
|
|
|
|
from systembridgemodels.notification import Notification
|
|
|
|
from homeassistant.components.notify import (
|
|
ATTR_DATA,
|
|
ATTR_TITLE,
|
|
ATTR_TITLE_DEFAULT,
|
|
BaseNotificationService,
|
|
)
|
|
from homeassistant.const import ATTR_ICON, CONF_ENTITY_ID
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
|
|
|
|
from .const import DOMAIN
|
|
from .coordinator import SystemBridgeDataUpdateCoordinator
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
ATTR_ACTIONS = "actions"
|
|
ATTR_AUDIO = "audio"
|
|
ATTR_IMAGE = "image"
|
|
ATTR_TIMEOUT = "timeout"
|
|
|
|
|
|
async def async_get_service(
|
|
hass: HomeAssistant,
|
|
config: ConfigType,
|
|
discovery_info: DiscoveryInfoType | None = None,
|
|
) -> SystemBridgeNotificationService | None:
|
|
"""Get the System Bridge notification service."""
|
|
if discovery_info is None:
|
|
return None
|
|
|
|
coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][
|
|
discovery_info[CONF_ENTITY_ID]
|
|
]
|
|
|
|
return SystemBridgeNotificationService(coordinator)
|
|
|
|
|
|
class SystemBridgeNotificationService(BaseNotificationService):
|
|
"""Implement the notification service for System Bridge."""
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator: SystemBridgeDataUpdateCoordinator,
|
|
) -> None:
|
|
"""Initialize the service."""
|
|
self._coordinator: SystemBridgeDataUpdateCoordinator = coordinator
|
|
|
|
async def async_send_message(
|
|
self,
|
|
message: str = "",
|
|
**kwargs: Any,
|
|
) -> None:
|
|
"""Send a message."""
|
|
data = kwargs.get(ATTR_DATA, {}) or {}
|
|
|
|
notification = Notification(
|
|
actions=data.get(ATTR_ACTIONS),
|
|
audio=data.get(ATTR_AUDIO),
|
|
icon=data.get(ATTR_ICON),
|
|
image=data.get(ATTR_IMAGE),
|
|
message=message,
|
|
timeout=data.get(ATTR_TIMEOUT),
|
|
title=kwargs.get(ATTR_TITLE, data.get(ATTR_TITLE, ATTR_TITLE_DEFAULT)),
|
|
)
|
|
|
|
_LOGGER.debug("Sending notification: %s", notification)
|
|
|
|
await self._coordinator.websocket_client.send_notification(notification)
|