home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-06 14:15:12 +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 # Blacklist of imports that should be using the namespace
_FORCE_NAMESPACE_IMPORT: dict[tuple[str, str], str] = { _FORCE_NAMESPACE_IMPORT: dict[tuple[str, str], set[str]] = {
("homeassistant.helpers.area_registry", "async_get"): "ar.async_get", ("homeassistant.helpers.area_registry", "ar"): {"async_get"},
( ("homeassistant.helpers.device_registry", "dr"): {
"homeassistant.helpers.device_registry", "async_get",
"async_entries_for_config_entry", "async_entries_for_config_entry",
): "dr.async_entries_for_config_entry", },
("homeassistant.helpers.device_registry", "async_get"): "dr.async_get", ("homeassistant.helpers.entity_registry", "er"): {
( "async_get",
"homeassistant.helpers.entity_registry",
"async_entries_for_config_entry", "async_entries_for_config_entry",
): "er.async_entries_for_config_entry", },
("homeassistant.helpers.entity_registry", "async_get"): "er.async_get", ("homeassistant.helpers.issue_registry", "ir"): {"async_get"},
("homeassistant.helpers.issue_registry", "async_get"): "ir.async_get",
} }
@@ -552,9 +550,11 @@ class HassImportsFormatChecker(BaseChecker):
self, node: nodes.ImportFrom, module: str, name: str self, node: nodes.ImportFrom, module: str, name: str
) -> None: ) -> None:
for key, value in _FORCE_NAMESPACE_IMPORT.items(): 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( self.add_message(
"hass-helper-namespace-import", node=node, args=(name, value) "hass-helper-namespace-import",
node=node,
args=(name, f"{key[1]}.{name}"),
) )