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

Use HassKey for importlib helper (#117116)

Browse source
This commit is contained in:
Marc Mueller 2024-05-09 00:42:28 +02:00 • committed by GitHub
parent a77add1b77
commit 04c0b7d3df
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
homeassistant/helpers/importlib.py
View file

@ -10,12 +10,15 @@ import sys
from types import ModuleType
from homeassistant.core import HomeAssistant
from homeassistant.util.hass_dict import HassKey
_LOGGER = logging.getLogger(__name__)
DATA_IMPORT_CACHE = "import_cache"
DATA_IMPORT_FUTURES = "import_futures"
DATA_IMPORT_FAILURES = "import_failures"
DATA_IMPORT_CACHE: HassKey[dict[str, ModuleType]] = HassKey("import_cache")
DATA_IMPORT_FUTURES: HassKey[dict[str, asyncio.Future[ModuleType]]] = HassKey(
"import_futures"
)
DATA_IMPORT_FAILURES: HassKey[dict[str, bool]] = HassKey("import_failures")
def _get_module(cache: dict[str, ModuleType], name: str) -> ModuleType:
@ -26,17 +29,15 @@ def _get_module(cache: dict[str, ModuleType], name: str) -> ModuleType:
async def async_import_module(hass: HomeAssistant, name: str) -> ModuleType:
"""Import a module or return it from the cache."""
cache: dict[str, ModuleType] = hass.data.setdefault(DATA_IMPORT_CACHE, {})
cache = hass.data.setdefault(DATA_IMPORT_CACHE, {})
if module := cache.get(name):
return module
failure_cache: dict[str, bool] = hass.data.setdefault(DATA_IMPORT_FAILURES, {})
failure_cache = hass.data.setdefault(DATA_IMPORT_FAILURES, {})
if name in failure_cache:
raise ModuleNotFoundError(f"{name} not found", name=name)
import_futures: dict[str, asyncio.Future[ModuleType]]
import_futures = hass.data.setdefault(DATA_IMPORT_FUTURES, {})
if future := import_futures.get(name):
return await future