mirror of
https://github.com/jpawlowski/hass.tibber_prices.git
synced 2026-03-30 13:23:41 +00:00
87 lines
3.6 KiB
Python
87 lines
3.6 KiB
Python
"""TibberPricesEntity class."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
|
|
from homeassistant.helpers.update_coordinator import CoordinatorEntity
|
|
|
|
from .const import ATTRIBUTION, DOMAIN, get_home_type_translation
|
|
from .coordinator import TibberPricesDataUpdateCoordinator
|
|
|
|
|
|
class TibberPricesEntity(CoordinatorEntity[TibberPricesDataUpdateCoordinator]):
|
|
"""TibberPricesEntity class."""
|
|
|
|
_attr_attribution = ATTRIBUTION
|
|
_attr_has_entity_name = True
|
|
|
|
def __init__(self, coordinator: TibberPricesDataUpdateCoordinator) -> None:
|
|
"""Initialize."""
|
|
super().__init__(coordinator)
|
|
|
|
# Get user profile information from coordinator
|
|
user_profile = self.coordinator.get_user_profile()
|
|
|
|
# Check if this is a main entry or subentry
|
|
is_subentry = bool(self.coordinator.config_entry.data.get("home_id"))
|
|
|
|
# Initialize variables
|
|
home_name = "Tibber Home"
|
|
home_id = self.coordinator.config_entry.unique_id
|
|
home_type = None
|
|
|
|
if is_subentry:
|
|
# For subentries, show specific home information
|
|
home_data = self.coordinator.config_entry.data.get("home_data", {})
|
|
home_id = self.coordinator.config_entry.data.get("home_id")
|
|
|
|
# Get home details
|
|
address = home_data.get("address", {})
|
|
address1 = address.get("address1", "")
|
|
city = address.get("city", "")
|
|
app_nickname = home_data.get("appNickname", "")
|
|
home_type = home_data.get("type", "")
|
|
|
|
# Compose home name
|
|
home_name = app_nickname or address1 or f"Tibber Home {home_id}"
|
|
if city:
|
|
home_name = f"{home_name}, {city}"
|
|
|
|
# Add user information if available
|
|
if user_profile and user_profile.get("name"):
|
|
home_name = f"{home_name} ({user_profile['name']})"
|
|
elif user_profile:
|
|
# For main entry, show user profile information
|
|
user_name = user_profile.get("name", "Tibber User")
|
|
user_email = user_profile.get("email", "")
|
|
home_name = f"Tibber - {user_name}"
|
|
if user_email:
|
|
home_name = f"{home_name} ({user_email})"
|
|
elif coordinator.data:
|
|
# Fallback to original logic if user data not available yet
|
|
try:
|
|
address1 = str(coordinator.data.get("address", {}).get("address1", ""))
|
|
city = str(coordinator.data.get("address", {}).get("city", ""))
|
|
app_nickname = str(coordinator.data.get("appNickname", ""))
|
|
home_type = str(coordinator.data.get("type", ""))
|
|
# Compose a nice name
|
|
home_name = "Tibber " + (app_nickname or address1 or "Home")
|
|
if city:
|
|
home_name = f"{home_name}, {city}"
|
|
except (KeyError, IndexError, TypeError):
|
|
home_name = "Tibber Home"
|
|
|
|
# Get translated home type using the configured language
|
|
language = coordinator.hass.config.language or "en"
|
|
translated_model = get_home_type_translation(home_type, language) if home_type else "Unknown"
|
|
|
|
self._attr_device_info = DeviceInfo(
|
|
entry_type=DeviceEntryType.SERVICE,
|
|
identifiers={(DOMAIN, coordinator.config_entry.unique_id or coordinator.config_entry.entry_id)},
|
|
name=home_name,
|
|
manufacturer="Tibber",
|
|
model=translated_model,
|
|
model_id=home_type if home_type else None,
|
|
serial_number=home_id if home_id else None,
|
|
configuration_url="https://developer.tibber.com/explorer",
|
|
)
|