Remove single use vars for service and attrs

mill1000 · Nov 2, 2023 · 4196172 · 4196172
1 parent 43b7a5c
commit 4196172
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/custom_components/midea_ac/climate.py b/custom_components/midea_ac/climate.py
@@ -49,12 +49,6 @@
     HVACMode.AUTO: AC.OperationalMode.AUTO,
 }
 
-_ATTR_FOLLOW_ME = "follow_me"
-_SERVICE_SET_FOLLOW_ME = "set_follow_me"
-_SERVICE_SET_FOLLOW_ME_SCHEMA = {
-    vol.Required(CONF_ENABLED): cv.boolean,
-}
-
 
 async def async_setup_entry(
     hass: HomeAssistant,
@@ -76,8 +70,10 @@ async def async_setup_entry(
     if helpers.property_exists(coordinator.device, "follow_me"):
         platform = entity_platform.async_get_current_platform()
         platform .async_register_entity_service(
-            _SERVICE_SET_FOLLOW_ME,
-            _SERVICE_SET_FOLLOW_ME_SCHEMA,
+            "set_follow_me",
+            {
+                vol.Required(CONF_ENABLED): cv.boolean,
+            },
             "async_set_follow_me",
         )
 
@@ -227,7 +223,7 @@ def extra_state_attributes(self) -> dict[str, str]:
         state_attributes = {}
 
         if hasattr(self._device, "follow_me"):
-            state_attributes[_ATTR_FOLLOW_ME] = getattr(
+            state_attributes["follow_me"] = getattr(
                 self._device, "follow_me")
 
         return state_attributes