evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 17 additions and 0 deletions
Show stats 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(