jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 03:07:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Simplify custom component loading (#128813)

Browse Source
This commit is contained in:
Paulus Schoutsen 2024-10-20 03:47:27 -04:00 committed by GitHub
parent e8acb48b1e
commit 28ff138370
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 10 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
homeassistant/loader.py
View File

@ -283,9 +283,7 @@ def manifest_from_legacy_module(domain: str, module: ModuleType) -> Manifest:
}
async def _async_get_custom_components(
hass: HomeAssistant,
) -> dict[str, Integration]:
def _get_custom_components(hass: HomeAssistant) -> dict[str, Integration]:
"""Return list of custom integrations."""
if hass.config.recovery_mode or hass.config.safe_mode:
return {}
@ -295,21 +293,14 @@ async def _async_get_custom_components(
except ImportError:
return {}
def get_sub_directories(paths: list[str]) -> list[pathlib.Path]:
"""Return all sub directories in a set of paths."""
return [
dirs = [
entry
for path in paths
for path in custom_components.__path__
for entry in pathlib.Path(path).iterdir()
if entry.is_dir()
]
dirs = await hass.async_add_executor_job(
get_sub_directories, custom_components.__path__
)
integrations = await hass.async_add_executor_job(
_resolve_integrations_from_root,
integrations = _resolve_integrations_from_root(
hass,
custom_components,
[comp.name for comp in dirs],
@ -330,7 +321,7 @@ async def async_get_custom_components(
if comps_or_future is None:
future = hass.data[DATA_CUSTOM_COMPONENTS] = hass.loop.create_future()
comps = await _async_get_custom_components(hass)
comps = await hass.async_add_executor_job(_get_custom_components, hass)
hass.data[DATA_CUSTOM_COMPONENTS] = comps
future.set_result(comps)

2
tests/test_loader.py
View File

@ -818,7 +818,7 @@ async def test_get_custom_components(hass: HomeAssistant) -> None:
test_1_integration = _get_test_integration(hass, "test_1", False)
test_2_integration = _get_test_integration(hass, "test_2", True)
name = "homeassistant.loader._async_get_custom_components"
name = "homeassistant.loader._get_custom_components"
with patch(name) as mock_get:
mock_get.return_value = {
"test_1": test_1_integration,