jpawlowski/hass.tibber_prices
1
0
Fork 0
mirror of https://github.com/jpawlowski/hass.tibber_prices.git synced 2026-03-30 05:13:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

fix

Browse source
This commit is contained in:
Julian Pawlowski 2025-05-24 17:35:55 +00:00
parent 86c8073a51
commit 5bb8af942e
1 changed files with 15 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
custom_components/tibber_prices/config_flow.py
View file

@ -4,7 +4,7 @@ from __future__ import annotations
import voluptuous as vol import voluptuous as vol
from homeassistant import config_entries from homeassistant.config_entries import ConfigEntry, ConfigFlow, ConfigFlowResult, OptionsFlow
from homeassistant.const import CONF_ACCESS_TOKEN from homeassistant.const import CONF_ACCESS_TOKEN
from homeassistant.helpers import selector from homeassistant.helpers import selector
from homeassistant.helpers.aiohttp_client import async_create_clientsession from homeassistant.helpers.aiohttp_client import async_create_clientsession
@ -27,7 +27,7 @@ from .const import (
) )
class TibberPricesFlowHandler(config_entries.ConfigFlow, domain=DOMAIN): class TibberPricesFlowHandler(ConfigFlow, domain=DOMAIN):
"""Config flow for tibber_prices.""" """Config flow for tibber_prices."""
VERSION = 1 VERSION = 1
@ -35,18 +35,18 @@ class TibberPricesFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
def __init__(self) -> None: def __init__(self) -> None:
"""Initialize the config flow.""" """Initialize the config flow."""
super().__init__() super().__init__()
self._reauth_entry: config_entries.ConfigEntry | None = None self._reauth_entry: ConfigEntry | None = None
self._pending_user_input: dict | None = None self._pending_user_input: dict | None = None
@staticmethod @staticmethod
def async_get_options_flow( def async_get_options_flow(
config_entry: config_entries.ConfigEntry, config_entry: ConfigEntry,
) -> config_entries.OptionsFlow: ) -> OptionsFlow:
"""Get the options flow for this handler.""" """Get the options flow for this handler."""
return TibberPricesOptionsFlowHandler(config_entry) return TibberPricesOptionsFlowHandler(config_entry)
@staticmethod @staticmethod
def async_get_reauth_flow(entry: config_entries.ConfigEntry) -> config_entries.ConfigFlow: def async_get_reauth_flow(entry: ConfigEntry) -> ConfigFlow:
"""Return the reauth flow handler for this integration.""" """Return the reauth flow handler for this integration."""
return TibberPricesReauthFlowHandler(entry) return TibberPricesReauthFlowHandler(entry)
@ -57,7 +57,7 @@ class TibberPricesFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
async def async_step_user( async def async_step_user(
self, self,
user_input: dict | None = None, user_input: dict | None = None,
) -> config_entries.ConfigFlowResult: ) -> ConfigFlowResult:
"""Handle a flow initialized by the user. Only ask for access token.""" """Handle a flow initialized by the user. Only ask for access token."""
_errors = {} _errors = {}
if user_input is not None: if user_input is not None:
@ -97,7 +97,7 @@ class TibberPricesFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
errors=_errors, errors=_errors,
) )
async def async_step_finish(self, user_input: dict | None = None) -> config_entries.ConfigFlowResult: async def async_step_finish(self, user_input: dict | None = None) -> ConfigFlowResult:
"""Show a finish screen after successful setup, then create entry on submit.""" """Show a finish screen after successful setup, then create entry on submit."""
if self._pending_user_input is not None and user_input is None: if self._pending_user_input is not None and user_input is None:
# First visit: show home selection # First visit: show home selection
@ -158,16 +158,16 @@ class TibberPricesFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
return result["viewer"] return result["viewer"]
class TibberPricesReauthFlowHandler(config_entries.ConfigFlow): class TibberPricesReauthFlowHandler(ConfigFlow):
"""Handle a reauthentication flow for tibber_prices.""" """Handle a reauthentication flow for tibber_prices."""
def __init__(self, entry: config_entries.ConfigEntry) -> None: def __init__(self, entry: ConfigEntry) -> None:
"""Initialize the reauth flow handler.""" """Initialize the reauth flow handler."""
self._entry = entry self._entry = entry
self._errors: dict[str, str] = {} self._errors: dict[str, str] = {}
self._pending_user_input: dict | None = None self._pending_user_input: dict | None = None
async def async_step_user(self, user_input: dict | None = None) -> config_entries.ConfigFlowResult: async def async_step_user(self, user_input: dict | None = None) -> ConfigFlowResult:
"""Prompt for a new access token, then go to finish for home selection.""" """Prompt for a new access token, then go to finish for home selection."""
if user_input is not None: if user_input is not None:
try: try:
@ -203,7 +203,7 @@ class TibberPricesReauthFlowHandler(config_entries.ConfigFlow):
errors=self._errors, errors=self._errors,
) )
async def async_step_finish(self, user_input: dict | None = None) -> config_entries.ConfigFlowResult: async def async_step_finish(self, user_input: dict | None = None) -> ConfigFlowResult:
"""Show home selection, then update config entry.""" """Show home selection, then update config entry."""
if self._pending_user_input is not None and user_input is None: if self._pending_user_input is not None and user_input is None:
viewer = self._pending_user_input["viewer"] viewer = self._pending_user_input["viewer"]
@ -238,14 +238,14 @@ class TibberPricesReauthFlowHandler(config_entries.ConfigFlow):
return self.async_abort(reason="setup_complete") return self.async_abort(reason="setup_complete")
class TibberPricesOptionsFlowHandler(config_entries.OptionsFlow): class TibberPricesOptionsFlowHandler(OptionsFlow):
"""Tibber Prices config flow options handler.""" """Tibber Prices config flow options handler."""
def __init__(self, config_entry: config_entries.ConfigEntry) -> None: # noqa: ARG002 def __init__(self, config_entry: ConfigEntry) -> None: # noqa: ARG002
"""Initialize options flow.""" """Initialize options flow."""
super().__init__() super().__init__()
async def async_step_init(self, user_input: dict | None = None) -> config_entries.ConfigFlowResult: async def async_step_init(self, user_input: dict | None = None) -> ConfigFlowResult:
"""Manage the options.""" """Manage the options."""
errors: dict[str, str] = {} errors: dict[str, str] = {}