Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

cannot chain

pull/112640/head
J. Nick Koston 2024-03-07 11:56:26 -10:00
parent 441573ce70
commit 1d8708ded3
No known key found for this signature in database
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
homeassistant/core.py
View File

@ -23,7 +23,6 @@ import datetime
import enum import enum
import functools import functools
import inspect import inspect
from itertools import chain
import logging import logging
import os import os
import pathlib import pathlib
@ -843,13 +842,13 @@ class HomeAssistant:
await asyncio.sleep(0) await asyncio.sleep(0)
start_time: float | None = None start_time: float | None = None
current_task = asyncio.current_task() current_task = asyncio.current_task()
to_wait: Iterable[asyncio.Future[Any]] = self._tasks
if wait_periodic_tasks:
to_wait = chain(self._tasks, self._periodic_tasks)
while tasks := [ while tasks := [
task task
for task in to_wait for task in (
self._tasks | self._periodic_tasks
if wait_periodic_tasks
else self._tasks
)
if task is not current_task and not cancelling(task) if task is not current_task and not cancelling(task)
]: ]:
await self._await_and_log_pending(tasks) await self._await_and_log_pending(tasks)
@ -980,7 +979,7 @@ class HomeAssistant:
self._tasks = set() self._tasks = set()
# Cancel all background tasks # Cancel all background tasks
for task in chain(self._background_tasks, self._periodic_tasks): for task in self._background_tasks | self._periodic_tasks:
self._tasks.add(task) self._tasks.add(task)
task.add_done_callback(self._tasks.remove) task.add_done_callback(self._tasks.remove)
task.cancel("Home Assistant is stopping") task.cancel("Home Assistant is stopping")