From acebf1fb48d23bd086f5bb93a09793efa74ce9d2 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Tue, 24 Sep 2024 12:19:39 +0200
Subject: [PATCH] Adjust _ENTITY_COMPONENTS in hass-enforce-class-module
(#126603)
---
homeassistant/components/dominos/__init__.py | 2 +-
homeassistant/components/microsoft_face/__init__.py | 2 +-
homeassistant/components/plant/__init__.py | 2 +-
homeassistant/components/template/trigger_entity.py | 4 +++-
pylint/plugins/hass_enforce_class_module.py | 9 ++++++++-
5 files changed, 14 insertions(+), 5 deletions(-)
diff --git a/homeassistant/components/dominos/__init__.py b/homeassistant/components/dominos/__init__.py
index 609cb93ba0d..9b11b667e84 100644
--- a/homeassistant/components/dominos/__init__.py
+++ b/homeassistant/components/dominos/__init__.py
@@ -182,7 +182,7 @@ class DominosProductListView(http.HomeAssistantView):
return self.json(self.dominos.get_menu())
-class DominosOrder(Entity): # pylint: disable=hass-enforce-class-module
+class DominosOrder(Entity):
"""Represents a Dominos order entity."""
def __init__(self, order_info, dominos):
diff --git a/homeassistant/components/microsoft_face/__init__.py b/homeassistant/components/microsoft_face/__init__.py
index 6a7e2d42fd9..fa4de7f9c99 100644
--- a/homeassistant/components/microsoft_face/__init__.py
+++ b/homeassistant/components/microsoft_face/__init__.py
@@ -214,7 +214,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
return True
-class MicrosoftFaceGroupEntity(Entity): # pylint: disable=hass-enforce-class-module
+class MicrosoftFaceGroupEntity(Entity):
"""Person-Group state/data Entity."""
_attr_should_poll = False
diff --git a/homeassistant/components/plant/__init__.py b/homeassistant/components/plant/__init__.py
index b3e1084f501..c6e527290df 100644
--- a/homeassistant/components/plant/__init__.py
+++ b/homeassistant/components/plant/__init__.py
@@ -127,7 +127,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
return True
-class Plant(Entity): # pylint: disable=hass-enforce-class-module
+class Plant(Entity):
"""Plant monitors the well-being of a plant.
It also checks the measurements against
diff --git a/homeassistant/components/template/trigger_entity.py b/homeassistant/components/template/trigger_entity.py
index 697cd827b9e..df84ce057c3 100644
--- a/homeassistant/components/template/trigger_entity.py
+++ b/homeassistant/components/template/trigger_entity.py
@@ -9,7 +9,9 @@ from homeassistant.helpers.update_coordinator import CoordinatorEntity
from . import TriggerUpdateCoordinator
-class TriggerEntity(TriggerBaseEntity, CoordinatorEntity[TriggerUpdateCoordinator]):
+class TriggerEntity( # pylint: disable=hass-enforce-class-module
+ TriggerBaseEntity, CoordinatorEntity[TriggerUpdateCoordinator]
+):
"""Template entity based on trigger data."""
def __init__(
diff --git a/pylint/plugins/hass_enforce_class_module.py b/pylint/plugins/hass_enforce_class_module.py
index e48cae877a5..95527126a30 100644
--- a/pylint/plugins/hass_enforce_class_module.py
+++ b/pylint/plugins/hass_enforce_class_module.py
@@ -76,16 +76,23 @@ _MODULES: dict[str, set[str]] = {
}
_ENTITY_COMPONENTS: set[str] = {platform.value for platform in Platform}.union(
{
+ "alert",
"automation",
"counter",
+ "dominos",
"input_boolean",
+ "input_button",
"input_datetime",
"input_number",
+ "input_select",
"input_text",
+ "microsoft_face",
"person",
+ "plant",
+ "remember_the_milk",
+ "schedule",
"script",
"tag",
- "template",
"timer",
}
)
--
GitLab