get back fpl original folder

custom_components/fpl/config_flow.py

@@ -0,0 +1,113 @@
+"""Home Assistant Fpl integration Config Flow"""
+from collections import OrderedDict
+
+import voluptuous as vol
+
+from homeassistant import config_entries
+from homeassistant.helpers.aiohttp_client import async_create_clientsession
+from homeassistant.core import callback
+
+from .const import DOMAIN, CONF_USERNAME, CONF_PASSWORD, CONF_NAME
+
+from .fplapi import (
+    LOGIN_RESULT_OK,
+    LOGIN_RESULT_FAILURE,
+    LOGIN_RESULT_INVALIDUSER,
+    LOGIN_RESULT_INVALIDPASSWORD,
+    FplApi,
+)
+
+
+@callback
+def configured_instances(hass):
+    """Return a set of configured SimpliSafe instances."""
+    entites = []
+    for entry in hass.config_entries.async_entries(DOMAIN):
+        entites.append(f"{entry.data.get(CONF_USERNAME)}")
+    return set(entites)
+
+
+class FplFlowHandler(config_entries.ConfigFlow, domain=DOMAIN):
+    """Fpl Config Flow Handler"""
+
+    VERSION = 1
+    CONNECTION_CLASS = config_entries.CONN_CLASS_CLOUD_POLL
+
+    def __init__(self):
+        """Initialize."""
+        self._errors = {}
+
+    async def async_step_user(
+        self, user_input={}
+    ):  # pylint: disable=dangerous-default-value
+        """Handle a flow initialized by the user."""
+        self._errors = {}
+
+        # if self._async_current_entries():
+        #    return self.async_abort(reason="single_instance_allowed")
+
+        # if self.hass.data.get(DOMAIN):
+        #    return self.async_abort(reason="single_instance_allowed")
+
+        if user_input is not None:
+            username = user_input[CONF_USERNAME]
+            password = user_input[CONF_PASSWORD]
+
+            if username not in configured_instances(self.hass):
+                session = async_create_clientsession(self.hass)
+                api = FplApi(username, password, session)
+                result = await api.login()
+
+                if result == LOGIN_RESULT_OK:
+
+                    accounts = await api.async_get_open_accounts()
+                    await api.logout()
+
+                    user_input["accounts"] = accounts
+
+                    return self.async_create_entry(title=username, data=user_input)
+
+                if result == LOGIN_RESULT_INVALIDUSER:
+                    self._errors[CONF_USERNAME] = "invalid_username"
+
+                if result == LOGIN_RESULT_INVALIDPASSWORD:
+                    self._errors[CONF_PASSWORD] = "invalid_password"
+
+                if result == LOGIN_RESULT_FAILURE:
+                    self._errors["base"] = "failure"
+
+            else:
+                self._errors[CONF_NAME] = "name_exists"
+
+            return await self._show_config_form(user_input)
+
+        return await self._show_config_form(user_input)
+
+    async def _show_config_form(self, user_input):
+        """Show the configuration form to edit location data."""
+        username = ""
+        password = ""
+
+        if user_input is not None:
+            if CONF_USERNAME in user_input:
+                username = user_input[CONF_USERNAME]
+            if CONF_PASSWORD in user_input:
+                password = user_input[CONF_PASSWORD]
+
+        data_schema = OrderedDict()
+        data_schema[vol.Required(CONF_USERNAME, default=username)] = str
+        data_schema[vol.Required(CONF_PASSWORD, default=password)] = str
+
+        return self.async_show_form(
+            step_id="user", data_schema=vol.Schema(data_schema), errors=self._errors
+        )
+
+    async def async_step_import(self, user_input):  # pylint: disable=unused-argument
+        """Import a config entry.
+        Special type of import, we're not actually going to store any data.
+        Instead, we're going to rely on the values that are in config file.
+        """
+        if self._async_current_entries():
+            return self.async_abort(reason="single_instance_allowed")
+
+        return self.async_create_entry(title="configuration.yaml", data={})