Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-12-24 12:59:34 +00:00
Code Activity

Add RestoreSensor to sensor checks in pylint (#76916)

Browse Source
This commit is contained in:
epenet
2022-08-17 18:43:28 +02:00
committed by GitHub
parent fc6c66fe6c
commit 27b5ebb9d3
1 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
pylint/plugins/hass_enforce_type_hints.py
View File

@@ -1949,6 +1949,10 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
base_class="Entity",
matches=_ENTITY_MATCH,
),
ClassTypeHintMatch(
base_class="RestoreEntity",
matches=_RESTORE_ENTITY_MATCH,
),
ClassTypeHintMatch(
base_class="SensorEntity",
matches=[
@@ -1983,6 +1987,19 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
),
],
),
ClassTypeHintMatch(
base_class="RestoreSensor",
matches=[
TypeHintMatch(
function_name="extra_restore_state_data",
return_type="SensorExtraStoredData",
),
TypeHintMatch(
function_name="async_get_last_sensor_data",
return_type=["SensorExtraStoredData", None],
),
],
),
],
"siren": [
ClassTypeHintMatch(