Added options flow to reconfigure Sonarr integration to match Radarr Integration

Author: mkanet

custom_components/sonarr_upcoming_media/__init__.py

@@ -33,12 +33,12 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
             setup_client,
             hass,
             config_entry.data[CONF_API_KEY],
-            config_entry.data[CONF_DAYS],
+            config_entry.options.get(CONF_DAYS, config_entry.data[CONF_DAYS]),
             config_entry.data[CONF_HOST],
             config_entry.data[CONF_PORT],
             config_entry.data[CONF_SSL],
             config_entry.data[CONF_URLBASE],
-            config_entry.data[CONF_MAX],
+            config_entry.options.get(CONF_MAX, config_entry.data[CONF_MAX]),
         )
     except FailedToLogin as err:
         raise ConfigEntryNotReady("Failed to Log-in") from err
@@ -52,6 +52,7 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
     hass.data.setdefault(DOMAIN, {})[config_entry.entry_id] = coordinator
 
     await hass.config_entries.async_forward_entry_setups(config_entry, PLATFORMS)
+    config_entry.async_on_unload(config_entry.add_update_listener(update_listener))
 
     return True
 
@@ -64,3 +65,7 @@ async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) ->
         if not hass.data[DOMAIN]:
             del hass.data[DOMAIN]
     return unload_ok
+
+async def update_listener(hass: HomeAssistant, config_entry: ConfigEntry) -> None:
+    """Handle options update."""
+    await hass.config_entries.async_reload(config_entry.entry_id)

custom_components/sonarr_upcoming_media/config_flow.py

@@ -23,6 +23,10 @@
     SonarrCannotBeReached
 )
 
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import callback
+from .options_flow import SonarrOptionFlow
+
 def valid_max():
     return lambda x: type(x) == int and x > 0
 
@@ -40,6 +44,11 @@ def valid_max():
 
 class SonarrConfigFlow(ConfigFlow, domain=DOMAIN):
     """Config flow for the Sonarr integration."""
+    @staticmethod
+    @callback
+    def async_get_options_flow(config_entry: ConfigEntry) -> SonarrOptionFlow:
+        return SonarrOptionFlow(config_entry)
+
     async def async_step_user(
         self, user_input: dict[str, Any] | None = None
     ):
@@ -67,4 +76,4 @@ async def async_step_user(
                 return self.async_create_entry(title=user_input[CONF_NAME], data=user_input)
 
         schema = self.add_suggested_values_to_schema(SONARR_SCHEMA, user_input)
-        return self.async_show_form(step_id="user", data_schema=schema, errors=errors)
+        return self.async_show_form(step_id="user", data_schema=schema, errors=errors)

custom_components/sonarr_upcoming_media/manifest.json

@@ -8,6 +8,5 @@
     "iot_class": "local_polling",
     "issue_tracker": "https://github.com/custom-components/sensor.sonarr_upcoming_media/issues",
     "requirements": [],
-    "version": "0.4.3"
+    "version": "0.4.5"
   }
-  

custom_components/sonarr_upcoming_media/options_flow.py

@@ -0,0 +1,36 @@
+from typing import Any
+from homeassistant.config_entries import OptionsFlow, ConfigEntry, ConfigFlowResult
+import voluptuous as vol
+
+from .const import (
+    CONF_DAYS,
+    CONF_MAX,
+    DOMAIN,
+    )
+
+import logging
+_LOGGER = logging.getLogger(__name__)
+class SonarrOptionFlow(OptionsFlow):
+    def __init__(self, config_entry: ConfigEntry) -> None:
+        self._config_entry = config_entry
+
+
+    async def async_step_init(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult:
+        errors = {}
+
+        client = self.hass.data[DOMAIN][self._config_entry.entry_id]._client
+
+        if user_input is not None:
+            # Validate and process user input here
+            return self.async_create_entry(title="", data=user_input)
+
+        SONARR_SCHEMA = vol.Schema({
+            vol.Required(CONF_DAYS, default=self._config_entry.options.get(CONF_DAYS, self._config_entry.data[CONF_DAYS])): vol.All(vol.Coerce(int), vol.Range(min=1)),
+            vol.Required(CONF_MAX, default=self._config_entry.options.get(CONF_MAX, self._config_entry.data[CONF_MAX])): vol.All(vol.Coerce(int), vol.Range(min=0)),
+        })
+
+        # Display a form to gather user input
+        return self.async_show_form(step_id="init", data_schema=SONARR_SCHEMA, errors=errors)
+
+def keys(d) -> list:
+    return [i for i in d.keys()]

custom_components/sonarr_upcoming_media/strings.json

@@ -23,5 +23,15 @@
       "failed_to_login": "Failed to Log-in",
       "cannot_be_reached": "Sonarr cannot be reached"
     }
+  },
+  "options": {
+    "step": {
+      "init": {
+        "data": {
+          "days": "How many days look ahead for the upcoming sensor",
+          "max": "Max number of items in sensor"
+        }
+      }
+    }
   }
-}
+}