GeoRideHA/custom_components/georide/config_flow.py

""" Georide config flow """

import logging
from homeassistant import config_entries
import voluptuous as vol
import georideapilib.api as GeoRideApi
import georideapilib.exception as GeoRideException


from .const import CONF_EMAIL, CONF_PASSWORD, CONF_TOKEN, DOMAIN


_LOGGER = logging.getLogger(__name__)

@config_entries.HANDLERS.register(DOMAIN)
class GeoRideConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
    """GeoRide config flow """
    
    async def async_step_user(self, user_input=None): #pylint: disable=W0613
        """ handle info to help to configure GeoRide """

        if self._async_current_entries():
            return self.async_abort(reason="one_instance_allowed")

        return self.async_show_form(step_id='georide_login', data_schema=vol.Schema({
            vol.Required(CONF_EMAIL): vol.All(str, vol.Length(min=3)),
            vol.Required(CONF_PASSWORD): vol.All(str)
        }))

    async def async_step_import(self, user_input=None): #pylint: disable=W0613
        """Import a config flow from configuration."""
        if self._async_current_entries():
            return self.async_abort(reason="one_instance_allowed")

        return self.async_show_form(step_id='georide_login', data_schema=vol.Schema({
            vol.Required(CONF_EMAIL): vol.All(str, vol.Length(min=3)),
            vol.Required(CONF_PASSWORD): vol.All(str)
        }))

    async def async_step_georide_login(self, user_input):
        """ try to seupt GeoRide Account """

        schema = vol.Schema({
            vol.Required(CONF_EMAIL): vol.All(str, vol.Length(min=3)),
            vol.Required(CONF_PASSWORD): vol.All(str)
        })
        
        if user_input is None:
            return self.async_show_form(step_id='georide_login', data_schema=schema)
        
        email = user_input[CONF_EMAIL]
        password = user_input[CONF_PASSWORD]
        
        try:
            account = await self.hass.async_add_executor_job(GeoRideApi.get_authorisation_token, email, password)
            data = {
                CONF_EMAIL: email,
                CONF_PASSWORD: password,
                CONF_TOKEN: account.auth_token
            }
            return self.async_create_entry(title=email, data=data)
        except (GeoRideException.SeverException, GeoRideException.LoginException):
            _LOGGER.error("Invalid credentials provided, config not created")
            errors = {"base":  "faulty_credentials"}
            return self.async_show_form(step_id="georide_login", data_schema=schema, errors=errors)
        except: 
            _LOGGER.error("Unknown error")
            errors = {"base": "unkonwn"}
            return self.async_show_form(step_id="georide_login", data_schema=schema, errors=errors)