Register services

rgc99 · Nov 30, 2023 · 522ace2 · 522ace2
1 parent 7f2bff1
commit 522ace2
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/custom_components/irrigation_unlimited/service.py b/custom_components/irrigation_unlimited/service.py
@@ -30,6 +30,8 @@
     SERVICE_TOGGLE,
     SERVICE_LOAD_SCHEDULE,
     SERVICE_SUSPEND,
+    SERVICE_SKIP,
+    SERVICE_PAUSE,
 )
 
 
@@ -62,6 +64,13 @@ def register_platform_services(platform: entity_platform.EntityPlatform) -> None
     platform.async_register_entity_service(
         SERVICE_SUSPEND, SUSPEND_SCHEMA, async_entity_service_handler
     )
+    platform.async_register_entity_service(
+        SERVICE_SKIP, ENTITY_SCHEMA, async_entity_service_handler
+    )
+
+    platform.async_register_entity_service(
+        SERVICE_PAUSE, ENTITY_SCHEMA, async_entity_service_handler
+    )