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

Improve

Browse Source
This commit is contained in:
epenet
2024-05-27 10:02:55 +00:00
parent 21a3667c28
commit 8dffc98533
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
pylint/plugins/hass_imports.py
View File

@@ -395,19 +395,17 @@ _OBSOLETE_IMPORT: dict[str, list[ObsoleteImportMatch]] = {
}
# Blacklist of imports that should be using the namespace
_FORCE_NAMESPACE_IMPORT: dict[tuple[str, str], str] = {
("homeassistant.helpers.area_registry", "async_get"): "ar.async_get",
(
"homeassistant.helpers.device_registry",
_FORCE_NAMESPACE_IMPORT: dict[tuple[str, str], set[str]] = {
("homeassistant.helpers.area_registry", "ar"): {"async_get"},
("homeassistant.helpers.device_registry", "dr"): {
"async_get",
"async_entries_for_config_entry",
): "dr.async_entries_for_config_entry",
("homeassistant.helpers.device_registry", "async_get"): "dr.async_get",
(
"homeassistant.helpers.entity_registry",
},
("homeassistant.helpers.entity_registry", "er"): {
"async_get",
"async_entries_for_config_entry",
): "er.async_entries_for_config_entry",
("homeassistant.helpers.entity_registry", "async_get"): "er.async_get",
("homeassistant.helpers.issue_registry", "async_get"): "ir.async_get",
},
("homeassistant.helpers.issue_registry", "ir"): {"async_get"},
}
@@ -552,9 +550,11 @@ class HassImportsFormatChecker(BaseChecker):
self, node: nodes.ImportFrom, module: str, name: str
) -> None:
for key, value in _FORCE_NAMESPACE_IMPORT.items():
if module == key[0] and name == key[1]:
if module == key[0] and name in value:
self.add_message(
"hass-helper-namespace-import", node=node, args=(name, value)
"hass-helper-namespace-import",
node=node,
args=(name, f"{key[1]}.{name}"),
)