Merge 6eceee1f4e into af87fd0719

custom_components/sws12500/config_flow.py

@@ -1,15 +1,13 @@
 """Config flow for Sencor SWS 12500 Weather Station integration."""
 
-import logging
 from typing import Any
 
 import voluptuous as vol
 
 from homeassistant.config_entries import ConfigFlow, OptionsFlow
-from homeassistant.const import UnitOfPrecipitationDepth, UnitOfVolumetricFlux
 from homeassistant.core import callback
 from homeassistant.exceptions import HomeAssistantError
-import homeassistant.helpers.entity_registry as er
+from .utils import long_term_units_in_statistics_meta, migrate_data
 
 from .const import (
     API_ID,
@@ -17,18 +15,13 @@ from .const import (
     DEV_DBG,
     DOMAIN,
     INVALID_CREDENTIALS,
-    MIG_FROM,
-    MIG_TO,
-    SENSOR_TO_MIGRATE,
     SENSORS_TO_LOAD,
+    SENSOR_TO_MIGRATE,
     WINDY_API_KEY,
     WINDY_ENABLED,
     WINDY_LOGGER_ENABLED,
     WSLINK,
 )
-from .utils import long_term_units_in_statistics_meta, migrate_data
-
-_LOGGER = logging.getLogger(__name__)
 
 
 class CannotConnect(HomeAssistantError):
@@ -48,23 +41,16 @@ class ConfigOptionsFlowHandler(OptionsFlow):
 
         self.windy_data: dict[str, Any] = {}
         self.windy_data_schema = {}
-        self.user_data: dict[str, Any] = {}
+        self.user_data: dict[str, str] = {}
         self.user_data_schema = {}
         self.sensors: dict[str, Any] = {}
         self.migrate_schema = {}
-        self.migrate_sensor_select = {}
-        self.migrate_unit_selection = {}
-        self.count = 0
-        self.selected_sensor = ""
-
-        self.unit_values = [unit.value for unit in UnitOfVolumetricFlux]
-        self.unit_values.extend([unit.value for unit in UnitOfPrecipitationDepth])
 
         @property
         def config_entry(self):
             return self.hass.config_entries.async_get_entry(self.handler)
 
-    async def _get_entry_data(self):
+    def _get_entry_data(self):
         """Get entry data."""
 
         self.user_data: dict[str, Any] = {
@@ -75,10 +61,10 @@ class ConfigOptionsFlowHandler(OptionsFlow):
         }
 
         self.user_data_schema = {
-            vol.Required(API_ID, default=self.user_data.get(API_ID, "")): str,
+            vol.Required(API_ID, default=self.user_data[API_ID] or ""): str,
-            vol.Required(API_KEY, default=self.user_data.get(API_KEY, "")): str,
+            vol.Required(API_KEY, default=self.user_data[API_KEY] or ""): str,
-            vol.Optional(WSLINK, default=self.user_data.get(WSLINK, False)): bool,
+            vol.Optional(WSLINK, default=self.user_data[WSLINK]): bool or False,
-            vol.Optional(DEV_DBG, default=self.user_data.get(DEV_DBG, False)): bool,
+            vol.Optional(DEV_DBG, default=self.user_data[DEV_DBG]): bool or False,
         }
 
         self.sensors: dict[str, Any] = {
@@ -97,7 +83,7 @@ class ConfigOptionsFlowHandler(OptionsFlow):
 
         self.windy_data_schema = {
             vol.Optional(
-                WINDY_API_KEY, default=self.windy_data.get(WINDY_API_KEY, "")
+                WINDY_API_KEY, default=self.windy_data[WINDY_API_KEY] or ""
             ): str,
             vol.Optional(WINDY_ENABLED, default=self.windy_data[WINDY_ENABLED]): bool
             or False,
@@ -107,34 +93,12 @@ class ConfigOptionsFlowHandler(OptionsFlow):
             ): bool or False,
         }
 
-        self.migrate_sensor_select = {
+        self.migrate_schema = {
             vol.Required(SENSOR_TO_MIGRATE): vol.In(
-                await self.load_sensors_to_migrate() or {}
+                long_term_units_in_statistics_meta() or {}
             ),
-        }
-
-        self.migrate_unit_selection = {
-            vol.Required(MIG_FROM): vol.In(self.unit_values),
-            vol.Required(MIG_TO): vol.In(self.unit_values),
             vol.Optional("trigger_action", default=False): bool,
         }
-        # "mm/d", "mm/h", "mm", "in/d", "in/h", "in"
-
-    async def load_sensors_to_migrate(self) -> dict[str, Any]:
-        """Load sensors to migrate."""
-
-        sensor_statistics = await long_term_units_in_statistics_meta(self.hass)
-
-        entity_registry = er.async_get(self.hass)
-        sensors = entity_registry.entities.get_entries_for_config_entry_id(
-            self.config_entry.entry_id
-        )
-
-        return {
-            sensor.entity_id: f"{sensor.name or sensor.original_name} (current settings: {sensor.unit_of_measurement}, longterm stats unit: {sensor_statistics.get(sensor.entity_id)})"
-            for sensor in sensors
-            if sensor.unique_id in {"rain", "daily_rain"}
-        }
 
     async def async_step_init(self, user_input=None):
         """Manage the options - show menu first."""
@@ -146,7 +110,7 @@ class ConfigOptionsFlowHandler(OptionsFlow):
         """Manage basic options - credentials."""
         errors = {}
 
-        await self._get_entry_data()
+        self._get_entry_data()
 
         if user_input is None:
             return self.async_show_form(
@@ -183,7 +147,7 @@ class ConfigOptionsFlowHandler(OptionsFlow):
         """Manage windy options."""
         errors = {}
 
-        await self._get_entry_data()
+        self._get_entry_data()
 
         if user_input is None:
             return self.async_show_form(
@@ -196,7 +160,7 @@ class ConfigOptionsFlowHandler(OptionsFlow):
             errors[WINDY_API_KEY] = "windy_key_required"
             return self.async_show_form(
                 step_id="windy",
-                data_schema=vol.Schema(self.windy_data_schema),
+                data_schema=self.windy_data_schema,
                 errors=errors,
             )
 
@@ -211,17 +175,15 @@ class ConfigOptionsFlowHandler(OptionsFlow):
     async def async_step_migration(self, user_input=None):
         """Migrate sensors."""
 
+        # hj
         errors = {}
 
-        data_schema = vol.Schema(self.migrate_sensor_select)
+        self._get_entry_data()
-        data_schema.schema.update()
-
-        await self._get_entry_data()
 
         if user_input is None:
             return self.async_show_form(
                 step_id="migration",
-                data_schema=vol.Schema(self.migrate_sensor_select),
+                data_schema=vol.Schema(self.migrate_schema),
                 errors=errors,
                 description_placeholders={
                     "migration_status": "-",
@@ -229,116 +191,18 @@ class ConfigOptionsFlowHandler(OptionsFlow):
                 },
             )
 
-        self.selected_sensor = user_input.get(SENSOR_TO_MIGRATE)
-
-        return await self.async_step_migration_units()
-
-    async def async_step_migration_units(self, user_input=None):
-        """Migrate unit step."""
-
-        registry = er.async_get(self.hass)
-        sensor_entry = registry.async_get(self.selected_sensor)
-        sensor_stats = await long_term_units_in_statistics_meta(self.hass)
-
-        default_unit = sensor_entry.unit_of_measurement if sensor_entry else None
-
-        if default_unit not in self.unit_values:
-            default_unit = self.unit_values[0]
-
-        data_schema = vol.Schema({
-            vol.Required(MIG_FROM, default=default_unit): vol.In(self.unit_values),
-            vol.Required(MIG_TO): vol.In(self.unit_values),
-            vol.Optional("trigger_action", default=False): bool,
-        })
-
-        if user_input is None:
-            return self.async_show_form(
-                step_id="migration_units",
-                data_schema=data_schema,
-                errors={},
-                description_placeholders={
-                    "migration_sensor": sensor_entry.original_name,
-                    "migration_stats": sensor_stats.get(self.selected_sensor),
-                },
-            )
-
         if user_input.get("trigger_action"):
-            self.count = await migrate_data(
+            # Akce se vykoná po zaškrtnutí
-                self.hass,
+            count = await self.hass.async_add_executor_job(
-                self.selected_sensor,
+                migrate_data, user_input.get(SENSOR_TO_MIGRATE)
-                user_input.get(MIG_FROM),
-                user_input.get(MIG_TO),
             )
-
-            registry.async_update_entity(self.selected_sensor,
-                unit_of_measurement=user_input.get(MIG_TO),
-            )
-
-            state = self.hass.states.get(self.selected_sensor)
-            if state:
-                _LOGGER.info("State attributes before update: %s", state.attributes)
-                attributes = dict(state.attributes)
-                attributes["unit_of_measurement"] = user_input.get(MIG_TO)
-                self.hass.states.async_set(self.selected_sensor, state.state, attributes)
-                _LOGGER.info("State attributes after update: %s", attributes)
-
-                options = {**self.config_entry.options, "reload_sensor": self.selected_sensor}
-                self.hass.config_entries.async_update_entry(self.config_entry, options=options)
-
-            await self.hass.config_entries.async_reload(self.config_entry.entry_id)
-
-            await self.hass.async_block_till_done()
-
-            _LOGGER.info("Migration complete for sensor %s: %s row updated, new measurement unit: %s, ",
-                         self.selected_sensor,
-                         self.count,
-                         user_input.get(MIG_TO),
-                        )
-
-            await self._get_entry_data()
-            sensor_entry = er.async_get(self.hass).async_get(self.selected_sensor)
-            sensor_stat = await self.load_sensors_to_migrate()
-
             return self.async_show_form(
-                step_id="migration_complete",
+                step_id="migration",
-                data_schema=vol.Schema({}),
+                data_schema=vol.Schema(self.migrate_schema),
-                errors={},
-                description_placeholders={
-                    "migration_sensor": sensor_entry.unit_of_measurement,
-                    "migration_stats": sensor_stat.get(self.selected_sensor),
-                    "migration_count": self.count,
-                },
-            )
-
-        # retain windy data
-        user_input.update(self.windy_data)
-
-        # retain user_data
-        user_input.update(self.user_data)
-
-        # retain senors
-        user_input.update(self.sensors)
-
-        return self.async_create_entry(title=DOMAIN, data=user_input)
-
-    async def async_step_migration_complete(self, user_input=None):
-        """Migration complete."""
-
-        errors = {}
-
-        await self._get_entry_data()
-        sensor_entry = er.async_get(self.hass).async_get(self.selected_sensor)
-        sensor_stat = await self.load_sensors_to_migrate()
-
-        if user_input is None:
-            return self.async_show_form(
-                step_id="migration_complete",
-                data_schema=vol.Schema({}),
                 errors=errors,
                 description_placeholders={
-                    "migration_sensor": sensor_entry.unit_of_measurement,
+                    "migration_status": user_input.get(SENSOR_TO_MIGRATE),
-                    "migration_stats": sensor_stat.get(self.selected_sensor),
+                    "migration_count": count,
-                    "migration_count": self.count,
                 },
             )