EVNHANOI: Fix token expried

trvqhuy · Dec 17, 2024 · 1091a51 · 1091a51
1 parent bf2227d
commit 1091a51
Show file tree

Hide file tree

Showing 3 changed files with 1,596 additions and 1,583 deletions.
diff --git a/custom_components/nestup_evn/config_flow.py b/custom_components/nestup_evn/config_flow.py
@@ -1,195 +1,197 @@
-"""Config flow for EVN Data integration."""
-
-from __future__ import annotations
-
-import logging
-from typing import Any
-
-import voluptuous as vol
-
-from homeassistant import config_entries
-from homeassistant.data_entry_flow import FlowResult
-
-from . import nestup_evn
-from .const import (
-    CONF_AREA,
-    CONF_CUSTOMER_ID,
-    CONF_ERR_UNKNOWN,
-    CONF_MONTHLY_START,
-    CONF_PASSWORD,
-    CONF_SUCCESS,
-    CONF_USERNAME,
-    DOMAIN,
-)
-
-_LOGGER = logging.getLogger(__name__)
-
-CUSTOMER_ID_FIELD = (
-    lambda data: {
-        vol.Required(CONF_CUSTOMER_ID): vol.All(str, vol.Length(min=11, max=13))
-    }
-    if (data.get(CONF_CUSTOMER_ID) is None)
-    else {
-        vol.Required(CONF_CUSTOMER_ID, default=data.get(CONF_CUSTOMER_ID)): vol.All(
-            str, vol.Length(min=11, max=13)
-        )
-    }
-)
-
-AUTH_FIELD = (
-    lambda data: {vol.Required(CONF_USERNAME): str, vol.Required(CONF_PASSWORD): str}
-    if (data.get(CONF_USERNAME) is None)
-    else {
-        vol.Required(CONF_USERNAME, default=data.get(CONF_USERNAME)): str,
-        vol.Required(CONF_PASSWORD, default=data.get(CONF_PASSWORD)): str,
-    }
-)
-
-DATE_START_FIELD = {
-    vol.Required(CONF_MONTHLY_START, default=24): vol.All(int, vol.Range(min=1, max=28))
-}
-
-
-class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
-    """Handle the Config Flow for setting up this Integration."""
-
-    VERSION = 1
-
-    def __init__(self):
-        self._user_data = {}
-        self._api = None
-        self._errors = {}
-
-    async def async_step_fulfill_data(
-        self, user_input: dict[str, Any] | None = None
-    ) -> FlowResult:
-        """Handle Fulfill Data config flow by the user."""
-
-        self._errors = {}
-
-        if user_input is not None:
-
-            self._user_data.update(user_input)
-            self._api = nestup_evn.EVNAPI(self.hass, True)
-
-            verify_account = await self._try_auth()
-
-            if verify_account is not CONF_SUCCESS:
-                self._errors["base"] = verify_account
-            else:
-                verify_id = await self._verify_id()
-
-                if verify_id is not CONF_SUCCESS:
-                    self._errors["base"] = verify_id
-                else:
-                    await self.async_set_unique_id(self._user_data[CONF_CUSTOMER_ID])
-
-                    self._abort_if_unique_id_configured()
-
-                    return self.async_create_entry(
-                        title=self._user_data[CONF_CUSTOMER_ID], data=self._user_data
-                    )
-
-        data_schema = {}
-
-        data_schema = AUTH_FIELD(self._user_data)
-
-        if bool(self._errors):
-            data_schema |= CUSTOMER_ID_FIELD(self._user_data)
-
-        if self._user_data[CONF_AREA].get("date_needed"):
-            data_schema |= DATE_START_FIELD
-
-        return self.async_show_form(
-            step_id="fulfill_data",
-            data_schema=vol.Schema(data_schema),
-            errors=self._errors,
-            description_placeholders={
-                "customer_id": self._user_data[CONF_CUSTOMER_ID],
-                "evn_name": self._user_data[CONF_AREA].get("name"),
-            },
-        )
-
-    async def async_step_evn_info(
-        self, user_input: dict[str, Any] | None = None
-    ) -> FlowResult:
-        """Handle EVN-info config flow by the user."""
-
-        self._errors = {}
-
-        if user_input is not None:
-
-            return await self.async_step_fulfill_data()
-
-        evn_info = nestup_evn.get_evn_info(self._user_data[CONF_CUSTOMER_ID])
-
-        if evn_info.get("status") is CONF_SUCCESS:
-            self._user_data[CONF_AREA] = evn_info["evn_area"]
-
-            return self.async_show_form(
-                step_id="evn_info",
-                description_placeholders=evn_info,
-                errors=self._errors,
-            )
-        else:
-            return self.async_abort(
-                reason=evn_info.get("status"), description_placeholders=evn_info
-            )
-
-    async def async_step_customer_id(
-        self, user_input: dict[str, Any] | None = None
-    ) -> FlowResult:
-        """Handle customer-id config flow by the user."""
-
-        if user_input is not None:
-            self._user_data[CONF_CUSTOMER_ID] = user_input[CONF_CUSTOMER_ID].upper()
-            return await self.async_step_evn_info()
-
-        return self.async_show_form(
-            step_id="customer_id",
-            data_schema=vol.Schema(CUSTOMER_ID_FIELD(self._user_data)),
-            errors=self._errors,
-        )
-
-    async def async_step_user(
-        self, user_input: dict[str, Any] | None = None
-    ) -> FlowResult:
-        """Handle a flow initialized by the user."""
-        return await self.async_step_customer_id()
-
-    async def _try_auth(self):
-        """Try authenticating credentials given by the user config flow."""
-
-        try:
-            res = await self._api.login(
-                self._user_data.get(CONF_AREA),
-                self._user_data.get(CONF_USERNAME),
-                self._user_data.get(CONF_PASSWORD),
-                self._user_data.get(CONF_CUSTOMER_ID),
-            )
-        except Exception as e:
-            _LOGGER.exception(f"Unexpected exception: {e}")
-            return CONF_ERR_UNKNOWN
-        else:
-            if res != CONF_SUCCESS:
-                return res
-
-        return CONF_SUCCESS
-
-    async def _verify_id(self):
-        """Try verifying EVN Customer ID given by the user config flow."""
-
-        try:
-            res = await self._api.request_update(
-                self._user_data.get(CONF_AREA),
-                self._user_data.get(CONF_CUSTOMER_ID),
-                self._user_data.get(CONF_MONTHLY_START),
-            )
-        except Exception as e:
-            _LOGGER.exception(f"Unexpected exception: {e}")
-            return CONF_ERR_UNKNOWN
-        else:
-            if res["status"] != CONF_SUCCESS:
-                return res["status"]
-
-        return CONF_SUCCESS
+"""Config flow for EVN Data integration."""
+
+from __future__ import annotations
+
+import logging
+from typing import Any
+
+import voluptuous as vol
+
+from homeassistant import config_entries
+from homeassistant.data_entry_flow import FlowResult
+
+from . import nestup_evn
+from .const import (
+    CONF_AREA,
+    CONF_CUSTOMER_ID,
+    CONF_ERR_UNKNOWN,
+    CONF_MONTHLY_START,
+    CONF_PASSWORD,
+    CONF_SUCCESS,
+    CONF_USERNAME,
+    DOMAIN,
+)
+
+_LOGGER = logging.getLogger(__name__)
+
+CUSTOMER_ID_FIELD = (
+    lambda data: {
+        vol.Required(CONF_CUSTOMER_ID): vol.All(str, vol.Length(min=11, max=13))
+    }
+    if (data.get(CONF_CUSTOMER_ID) is None)
+    else {
+        vol.Required(CONF_CUSTOMER_ID, default=data.get(CONF_CUSTOMER_ID)): vol.All(
+            str, vol.Length(min=11, max=13)
+        )
+    }
+)
+
+AUTH_FIELD = (
+    lambda data: {vol.Required(CONF_USERNAME): str, vol.Required(CONF_PASSWORD): str}
+    if (data.get(CONF_USERNAME) is None)
+    else {
+        vol.Required(CONF_USERNAME, default=data.get(CONF_USERNAME)): str,
+        vol.Required(CONF_PASSWORD, default=data.get(CONF_PASSWORD)): str,
+    }
+)
+
+DATE_START_FIELD = {
+    vol.Required(CONF_MONTHLY_START, default=24): vol.All(int, vol.Range(min=1, max=28))
+}
+
+
+class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
+    """Handle the Config Flow for setting up this Integration."""
+
+    VERSION = 1
+
+    def __init__(self):
+        self._user_data = {}
+        self._api = None
+        self._errors = {}
+
+    async def async_step_fulfill_data(
+        self, user_input: dict[str, Any] | None = None
+    ) -> FlowResult:
+        """Handle Fulfill Data config flow by the user."""
+
+        self._errors = {}
+
+        if user_input is not None:
+
+            self._user_data.update(user_input)
+            self._api = nestup_evn.EVNAPI(self.hass, True)
+
+            verify_account = await self._try_auth()
+
+            if verify_account is not CONF_SUCCESS:
+                self._errors["base"] = verify_account
+            else:
+                verify_id = await self._verify_id()
+
+                if verify_id is not CONF_SUCCESS:
+                    self._errors["base"] = verify_id
+                else:
+                    await self.async_set_unique_id(self._user_data[CONF_CUSTOMER_ID])
+
+                    self._abort_if_unique_id_configured()
+
+                    return self.async_create_entry(
+                        title=self._user_data[CONF_CUSTOMER_ID], data=self._user_data
+                    )
+
+        data_schema = {}
+
+        data_schema = AUTH_FIELD(self._user_data)
+
+        if bool(self._errors):
+            data_schema |= CUSTOMER_ID_FIELD(self._user_data)
+
+        if self._user_data[CONF_AREA].get("date_needed"):
+            data_schema |= DATE_START_FIELD
+
+        return self.async_show_form(
+            step_id="fulfill_data",
+            data_schema=vol.Schema(data_schema),
+            errors=self._errors,
+            description_placeholders={
+                "customer_id": self._user_data[CONF_CUSTOMER_ID],
+                "evn_name": self._user_data[CONF_AREA].get("name"),
+            },
+        )
+
+    async def async_step_evn_info(
+        self, user_input: dict[str, Any] | None = None
+    ) -> FlowResult:
+        """Handle EVN-info config flow by the user."""
+
+        self._errors = {}
+
+        if user_input is not None:
+
+            return await self.async_step_fulfill_data()
+
+        evn_info = nestup_evn.get_evn_info(self._user_data[CONF_CUSTOMER_ID])
+
+        if evn_info.get("status") is CONF_SUCCESS:
+            self._user_data[CONF_AREA] = evn_info["evn_area"]
+
+            return self.async_show_form(
+                step_id="evn_info",
+                description_placeholders=evn_info,
+                errors=self._errors,
+            )
+        else:
+            return self.async_abort(
+                reason=evn_info.get("status"), description_placeholders=evn_info
+            )
+
+    async def async_step_customer_id(
+        self, user_input: dict[str, Any] | None = None
+    ) -> FlowResult:
+        """Handle customer-id config flow by the user."""
+
+        if user_input is not None:
+            self._user_data[CONF_CUSTOMER_ID] = user_input[CONF_CUSTOMER_ID].upper()
+            return await self.async_step_evn_info()
+
+        return self.async_show_form(
+            step_id="customer_id",
+            data_schema=vol.Schema(CUSTOMER_ID_FIELD(self._user_data)),
+            errors=self._errors,
+        )
+
+    async def async_step_user(
+        self, user_input: dict[str, Any] | None = None
+    ) -> FlowResult:
+        """Handle a flow initialized by the user."""
+        return await self.async_step_customer_id()
+
+    async def _try_auth(self):
+        """Try authenticating credentials given by the user config flow."""
+
+        try:
+            res = await self._api.login(
+                self._user_data.get(CONF_AREA),
+                self._user_data.get(CONF_USERNAME),
+                self._user_data.get(CONF_PASSWORD),
+                self._user_data.get(CONF_CUSTOMER_ID),
+            )
+        except Exception as e:
+            _LOGGER.exception(f"Unexpected exception: {e}")
+            return CONF_ERR_UNKNOWN
+        else:
+            if res != CONF_SUCCESS:
+                return res
+
+        return CONF_SUCCESS
+
+    async def _verify_id(self):
+        """Try verifying EVN Customer ID given by the user config flow."""
+
+        try:
+            res = await self._api.request_update(
+                self._user_data.get(CONF_AREA),
+                self._user_data.get(CONF_USERNAME),
+                self._user_data.get(CONF_PASSWORD),                
+                self._user_data.get(CONF_CUSTOMER_ID),
+                self._user_data.get(CONF_MONTHLY_START),
+            )
+        except Exception as e:
+            _LOGGER.exception(f"Unexpected exception: {e}")
+            return CONF_ERR_UNKNOWN
+        else:
+            if res["status"] != CONF_SUCCESS:
+                return res["status"]
+
+        return CONF_SUCCESS