evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Only enable envoy inverters when the user has access (#49234)

Browse source
This commit is contained in:
J. Nick Koston 2021-04-14 23:17:32 -10:00 • committed by GitHub
parent d71f913a12
commit 2887eeb32f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 72 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

21
homeassistant/components/enphase_envoy/config_flow.py
View file

@ -35,7 +35,7 @@ async def validate_input(hass: HomeAssistant, data: dict[str, Any]) -> dict[str,
data[CONF_HOST],
data[CONF_USERNAME],
data[CONF_PASSWORD],
inverters=True,
inverters=False,
async_client=get_async_client(hass),
)
@ -59,6 +59,7 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
self.name = None
self.username = None
self.serial = None
self._reauth_entry = None
@callback
def _async_generate_schema(self):
@ -121,6 +122,13 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
return await self.async_step_user()
async def async_step_reauth(self, user_input):
"""Handle configuration by re-auth."""
self._reauth_entry = self.hass.config_entries.async_get_entry(
self.context["entry_id"]
)
return await self.async_step_user()
async def async_step_user(
self, user_input: dict[str, Any] | None = None
) -> dict[str, Any]:
@ -128,7 +136,10 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
errors = {}
if user_input is not None:
if user_input[CONF_HOST] in self._async_current_hosts():
if (
not self._reauth_entry
and user_input[CONF_HOST] in self._async_current_hosts()
):
return self.async_abort(reason="already_configured")
try:
await validate_input(self.hass, user_input)
@ -145,6 +156,12 @@ class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
data[CONF_NAME] = f"{ENVOY} {self.serial}"
else:
data[CONF_NAME] = self.name or ENVOY
if self._reauth_entry:
self.hass.config_entries.async_update_entry(
self._reauth_entry,
data=data,
)
return self.async_abort(reason="reauth_successful")
return self.async_create_entry(title=data[CONF_NAME], data=data)
if self.serial: