evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Migrate script integration to create eager tasks (#113189)

Browse source 
Along with #113183 the script execution may be able to
synchronously without having to be scheduled on the event loop
This commit is contained in:
J. Nick Koston 2024-03-13 18:23:25 -10:00 • committed by GitHub
parent 9d1c683a70
commit 4f113f256f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
homeassistant/components/script/__init__.py
View file

@ -56,6 +56,7 @@ from homeassistant.helpers.service import async_set_service_schema
from homeassistant.helpers.trace import trace_get, trace_path
from homeassistant.helpers.typing import ConfigType
from homeassistant.loader import bind_hass
from homeassistant.util.async_ import create_eager_task
from homeassistant.util.dt import parse_datetime
from .config import ScriptConfig
@ -227,7 +228,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
await asyncio.wait(
[
asyncio.create_task(script_entity.async_turn_off())
create_eager_task(script_entity.async_turn_off())
for script_entity in script_entities
]
)
@ -576,7 +577,7 @@ class ScriptEntity(BaseScriptEntity, RestoreEntity):
script_stack_cv.set([])
self._changed.clear()
self.hass.async_create_task(coro)
self.hass.async_create_task(coro, eager_start=True)
# Wait for first state change so we can guarantee that
# it is written to the State Machine before we return.
await self._changed.wait()