Small cleanups to rainmachine get_client_controller (#50250)

2021-05-07 13:07:06 -05:00 · 2021-05-07 13:07:06 -05:00 · 3a36a976ee
parent 80b05c39cc
commit 3a36a976ee
1 changed files with 2 additions and 3 deletions
--- a/homeassistant/components/rainmachine/config_flow.py
+++ b/homeassistant/components/rainmachine/config_flow.py
@ -22,9 +22,8 @@ DATA_SCHEMA = vol.Schema(


 def get_client_controller(client):
-    """Enumerate controllers to find the first mac."""
-    for controller in client.controllers.values():
-        return controller
+    """Return the first local controller."""
+    return next(iter(client.controllers.values()))


 async def async_get_controller(hass, ip_address, password, port, ssl):