Add scene checks to pylint plugin (#76908)

2022-08-17 18:44:08 +02:00 · 2022-08-17 18:44:08 +02:00 · 49c793b1a2
commit 49c793b1a2
parent 27b5ebb9d3
1 changed files with 21 additions and 0 deletions
--- a/pylint/plugins/hass_enforce_type_hints.py
+++ b/pylint/plugins/hass_enforce_type_hints.py
@ -1911,6 +1911,27 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
            ],
        ),
    ],
+    "scene": [
+        ClassTypeHintMatch(
+            base_class="Entity",
+            matches=_ENTITY_MATCH,
+        ),
+        ClassTypeHintMatch(
+            base_class="RestoreEntity",
+            matches=_RESTORE_ENTITY_MATCH,
+        ),
+        ClassTypeHintMatch(
+            base_class="Scene",
+            matches=[
+                TypeHintMatch(
+                    function_name="activate",
+                    kwargs_type="Any",
+                    return_type=None,
+                    has_async_counterpart=True,
+                ),
+            ],
+        ),
+    ],
    "select": [
        ClassTypeHintMatch(
            base_class="Entity",