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

Add sensor checks to pylint plugin (#76876)

Browse source
This commit is contained in:
epenet 2022-08-16 19:22:15 +02:00 • committed by GitHub
parent 5331af2143
commit 2630ff8fce
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 40 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
pylint/plugins/hass_enforce_type_hints.py
View file

@ -1863,6 +1863,46 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
],
),
],
"sensor": [
ClassTypeHintMatch(
base_class="Entity",
matches=_ENTITY_MATCH,
),
ClassTypeHintMatch(
base_class="SensorEntity",
matches=[
TypeHintMatch(
function_name="device_class",
return_type=["SensorDeviceClass", "str", None],
),
TypeHintMatch(
function_name="state_class",
return_type=["SensorStateClass", "str", None],
),
TypeHintMatch(
function_name="last_reset",
return_type=["datetime", None],
),
TypeHintMatch(
function_name="native_value",
return_type=[
"StateType",
"str",
"int",
"float",
None,
"date",
"datetime",
"Decimal",
],
),
TypeHintMatch(
function_name="native_unit_of_measurement",
return_type=["str", None],
),
],
),
],
"siren": [
ClassTypeHintMatch(
base_class="Entity",