Update entity.py

danielperna84 · Sep 30, 2024 · 4c950cf · 4c950cf
1 parent 7dcd3dc
commit 4c950cf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/hahomematic/platforms/entity.py b/hahomematic/platforms/entity.py
@@ -846,7 +846,7 @@ async def wrapper(*args: Any, **kwargs: Any) -> Any:
                 )
             return return_value
 
-        setattr(func, "__service", True)
+        setattr(func, "ha_service", True)
         return wrapper  # type: ignore[return-value]
 
     return decorator
@@ -874,7 +874,7 @@ async def wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
                     )
                 raise
 
-        setattr(func, "__service", True)
+        setattr(wrapper, "ha_service", True)
         return wrapper
 
     return decorator
@@ -887,5 +887,5 @@ def get_service_calls(obj: object) -> dict[str, Callable]:
         for name in dir(obj)
         if not name.startswith("_")
         and callable(getattr(obj, name))
-        and hasattr(getattr(obj, name), "__service")
+        and hasattr(getattr(obj, name), "ha_service")
     }