Move translation lookup to __init__

custom_components/sun2/__init__.py

@@ -1,6 +1,8 @@
 """Sun2 integration."""
 from __future__ import annotations
 
+from typing import cast
+
 import voluptuous as vol
 
 from homeassistant.config_entries import ConfigEntry, SOURCE_IMPORT
@@ -15,11 +17,12 @@
 from homeassistant.core import Event, HomeAssistant
 import homeassistant.helpers.config_validation as cv
 from homeassistant.helpers.dispatcher import dispatcher_send
+from homeassistant.helpers.translation import async_get_translations
 from homeassistant.helpers.typing import ConfigType
 
 from .binary_sensor import SUN2_BINARY_SENSOR_SCHEMA
 from .const import DOMAIN, SIG_HA_LOC_UPDATED
-from .helpers import LOC_PARAMS, LocData, LocParams
+from .helpers import LOC_PARAMS, LocData, LocParams, Sun2Data
 from .sensor import ELEVATION_AT_TIME_SCHEMA, TIME_AT_ELEVATION_SCHEMA
 
 
@@ -60,11 +63,16 @@ def _unique_locations_names(configs: list[dict]) -> list[dict]:
 
 async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
     """Setup composite integration."""
-    hass.data[DOMAIN] = {}
+    hass.data[DOMAIN] = Sun2Data(
+        locations={},
+        translations=await async_get_translations(
+            hass, hass.config.language, "misc", [DOMAIN], False
+        ),
+    )
 
     def update_local_loc_data(event: Event | None = None) -> None:
         """Update local location data from HA's config."""
-        hass.data[DOMAIN][None] = loc_data = LocData(
+        cast(Sun2Data, hass.data[DOMAIN]).locations[None] = loc_data = LocData(
             LocParams(
                 hass.config.elevation,
                 hass.config.latitude,

custom_components/sun2/config_flow.py

@@ -1,14 +1,14 @@
 """Config flow for Sun2 integration."""
 from __future__ import annotations
 
-from typing import Any
+from typing import cast, Any
 
 from homeassistant.config_entries import ConfigFlow
 from homeassistant.const import CONF_LOCATION, CONF_UNIQUE_ID
 from homeassistant.data_entry_flow import FlowResult
-from homeassistant.helpers.translation import async_get_translations
 
 from .const import DOMAIN
+from .helpers import Sun2Data
 
 
 class Sun2ConfigFlow(ConfigFlow, domain=DOMAIN):
@@ -18,11 +18,11 @@ class Sun2ConfigFlow(ConfigFlow, domain=DOMAIN):
 
     async def async_step_import(self, data: dict[str, Any]) -> FlowResult:
         """Import config entry from configuration."""
-        translations = await async_get_translations(
-            self.hass, self.hass.config.language, "service_name", [DOMAIN], False
-        )
         location = data.pop(CONF_LOCATION, self.hass.config.location_name)
-        title = f"{location} {translations[f'component.{DOMAIN}.service_name']}"
+        service_name = cast(Sun2Data, self.hass.data[DOMAIN]).translations[
+            f"component.{DOMAIN}.misc.service_name"
+        ]
+        title = f"{location} {service_name}"
         if existing_entry := await self.async_set_unique_id(data.pop(CONF_UNIQUE_ID)):
             self.hass.config_entries.async_update_entry(
                 existing_entry, title=title, options=data

custom_components/sun2/helpers.py

@@ -70,6 +70,14 @@ def __init__(self, lp: LocParams) -> None:
         object.__setattr__(self, "tzi", dt_util.get_time_zone(lp.time_zone))
 
 
+@dataclass
+class Sun2Data:
+    """Sun2 shared data."""
+
+    locations: dict[LocParams | None, LocData]
+    translations: dict[str, str]
+
+
 def get_loc_params(config: ConfigType) -> LocParams | None:
     """Get location parameters from configuration."""
     try:
@@ -140,6 +148,10 @@ def __init__(
         self._loc_params = loc_params
         self.async_on_remove(self._cancel_update)
 
+    @property
+    def _sun2_data(self) -> Sun2Data:
+        return cast(Sun2Data, self.hass.data[DOMAIN])
+
     async def async_update(self) -> None:
         """Update state."""
         if not self._loc_data:
@@ -162,9 +174,9 @@ def _get_loc_data(self) -> LocData:
         loc_params = None -> Use location parameters from HA's config.
         """
         try:
-            loc_data = cast(LocData, self.hass.data[DOMAIN][self._loc_params])
+            loc_data = self._sun2_data.locations[self._loc_params]
         except KeyError:
-            loc_data = self.hass.data[DOMAIN][self._loc_params] = LocData(
+            loc_data = self._sun2_data.locations[self._loc_params] = LocData(
                 cast(LocParams, self._loc_params)
             )
 

custom_components/sun2/translations/en.json

@@ -1,6 +1,8 @@
 {
     "title": "Sun2",
-    "service_name": "Sun",
+    "misc": {
+        "service_name": "Sun"
+    },
     "entity": {
         "sensor": {
             "astronomical_daylight": {

custom_components/sun2/translations/nl.json

@@ -1,6 +1,8 @@
 {
     "title": "Zon2",
-    "service_name": "Zon",
+    "misc": {
+        "service_name": "Zon"
+    },
     "entity": {
         "sensor": {
             "astronomical_daylight": {