home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-07-30 02:38:10 +02:00
Code Issues Packages Projects Releases Wiki Activity

Use HassKey for helpers (2) (#117013)

Browse Source
This commit is contained in:
Marc Mueller
2024-05-07 18:24:13 +02:00
committed by GitHub
parent c50a340cbc
commit 8f614fb06d
13 changed files with 69 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
homeassistant/helpers/recorder.py
View File

@ -1,12 +1,15 @@
"""Helpers to check recorder."""
from __future__ import annotations
import asyncio
from dataclasses import dataclass, field
from typing import Any
from homeassistant.core import HomeAssistant, callback
from homeassistant.util.hass_dict import HassKey
DOMAIN = "recorder"
DOMAIN: HassKey[RecorderData] = HassKey("recorder")
@dataclass(slots=True)
@ -14,7 +17,7 @@ class RecorderData:
"""Recorder data stored in hass.data."""
recorder_platforms: dict[str, Any] = field(default_factory=dict)
db_connected: asyncio.Future = field(default_factory=asyncio.Future)
db_connected: asyncio.Future[bool] = field(default_factory=asyncio.Future)
def async_migration_in_progress(hass: HomeAssistant) -> bool:
@ -40,5 +43,4 @@ async def async_wait_recorder(hass: HomeAssistant) -> bool:
"""
if DOMAIN not in hass.data:
return False
db_connected: asyncio.Future[bool] = hass.data[DOMAIN].db_connected
return await db_connected
return await hass.data[DOMAIN].db_connected