diff --git a/custom_components/irrigation_unlimited/service.py b/custom_components/irrigation_unlimited/service.py index 45152a6..52eaf4e 100644 --- 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 + )