From 1937d803c5f9fba91b4f9c810fa376d1dd8a13ff Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Mon, 27 Mar 2023 20:08:20 +0200
Subject: [PATCH] Add RestoreEntity pylint checks to all platforms (#90020)
---
pylint/plugins/hass_enforce_type_hints.py | 84 +++++++++++++++++++++++
1 file changed, 84 insertions(+)
diff --git a/pylint/plugins/hass_enforce_type_hints.py b/pylint/plugins/hass_enforce_type_hints.py
index 4fb471e2145..a84d578cf52 100644
--- a/pylint/plugins/hass_enforce_type_hints.py
+++ b/pylint/plugins/hass_enforce_type_hints.py
@@ -710,6 +710,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="AlarmControlPanelEntity",
matches=[
@@ -793,6 +797,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="BinarySensorEntity",
matches=[
@@ -836,6 +844,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="CalendarEntity",
matches=[
@@ -860,6 +872,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="Camera",
matches=[
@@ -974,6 +990,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="ClimateEntity",
matches=[
@@ -1147,6 +1167,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="CoverEntity",
matches=[
@@ -1246,6 +1270,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="BaseTrackerEntity",
matches=[
@@ -1319,6 +1347,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="ToggleEntity",
matches=_TOGGLE_ENTITY_MATCH,
@@ -1400,6 +1432,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="GeolocationEvent",
matches=[
@@ -1427,6 +1463,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="ImageProcessingEntity",
matches=[
@@ -1470,6 +1510,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="ToggleEntity",
matches=_TOGGLE_ENTITY_MATCH,
@@ -1525,6 +1569,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="ToggleEntity",
matches=_TOGGLE_ENTITY_MATCH,
@@ -1624,6 +1672,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="LockEntity",
matches=[
@@ -1714,6 +1766,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="MediaPlayerEntity",
matches=[
@@ -2118,6 +2174,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="ToggleEntity",
matches=_TOGGLE_ENTITY_MATCH,
@@ -2185,6 +2245,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="SelectEntity",
matches=[
@@ -2275,6 +2339,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="ToggleEntity",
matches=_TOGGLE_ENTITY_MATCH,
@@ -2334,6 +2402,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="ToggleEntity",
matches=_TOGGLE_ENTITY_MATCH,
@@ -2445,6 +2517,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="ToggleEntity",
matches=_TOGGLE_ENTITY_MATCH,
@@ -2580,6 +2656,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="WaterHeaterEntity",
matches=[
@@ -2661,6 +2741,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
+ ClassTypeHintMatch(
+ base_class="RestoreEntity",
+ matches=_RESTORE_ENTITY_MATCH,
+ ),
ClassTypeHintMatch(
base_class="WeatherEntity",
matches=[
--
GitLab