Update config_flow.py

Author: HAEdwin

custom_components/apsystems_ecu_proxy/config_flow.py

@@ -10,7 +10,7 @@
 
 
 class ConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
-    """First start of integration settings."""
+    """Integration configuration."""
     VERSION = 1
     CONNECTION_CLASS = config_entries.CONN_CLASS_LOCAL_PUSH
 
@@ -39,7 +39,7 @@ async def async_step_user(self, user_input=None):
                 return self.async_show_form(
                     step_id="user",
                     data_schema=schema,
-                    errors={"base": "Could not connect to the specified EMA host."},
+                    errors={"base": "Could not connect to the specified EMA host"},
                 )
         return self.async_show_form(step_id="user", data_schema=schema)
 
@@ -56,13 +56,16 @@ async def async_step_init(self, user_input=None):
             if not self.config_entry.options 
             else self.config_entry.options
         )
-        _LOGGER.debug("async_step_init with options: %s", current_options)
+        _LOGGER.debug("async_step_init with configuration: %s", current_options)
 
         schema = vol.Schema({
-            vol.Required(key, default=current_options.get(key)): str if key != "send_to_ema" else bool 
+            vol.Required(key, default=current_options.get(key)): (
+                str if key != "send_to_ema" else bool
+            )
             for key in KEYS
         })
 
+
         if user_input is not None:
             ema_host = user_input["ema_host"]
             if await self.validate_ip(ema_host):
@@ -73,14 +76,20 @@ async def async_step_init(self, user_input=None):
                 return self.async_show_form(
                     step_id="init",
                     data_schema=schema,
-                    errors={"base": "Could not connect to the specified EMA host."},
+                    errors={"base": "Could not connect to the specified EMA host"},
                 )
         return self.async_show_form(step_id="init", data_schema=schema)
 
     @staticmethod
     async def validate_ip(ip_address: str) -> bool:
         try:
-            await asyncio.wait_for(asyncio.open_connection(ip_address, 8995), timeout=5.0)
+            reader, writer = await asyncio.wait_for(
+                asyncio.open_connection(ip_address, 8995),
+                timeout=3.0
+            )
+            # Close the connection neatly
+            writer.close()
+            await writer.wait_closed()
             return True
         except (OSError, asyncio.TimeoutError):
             return False