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

Add siren checks to pylint plugin (#76460)

Browse source
This commit is contained in:
epenet 2022-08-09 16:55:52 +02:00 • committed by GitHub
parent b19ace9124
commit bcdf880364
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 52 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
pylint/plugins/hass_enforce_type_hints.py
View file

@ -1477,6 +1477,58 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
],
),
],
"select": [
ClassTypeHintMatch(
base_class="Entity",
matches=_ENTITY_MATCH,
),
ClassTypeHintMatch(
base_class="SelectEntity",
matches=[
TypeHintMatch(
function_name="capability_attributes",
return_type="dict[str, Any]",
),
TypeHintMatch(
function_name="options",
return_type="list[str]",
),
TypeHintMatch(
function_name="current_option",
return_type=["str", None],
),
TypeHintMatch(
function_name="select_option",
return_type=None,
),
TypeHintMatch(
function_name="select_option",
arg_types={1: "str"},
return_type=None,
has_async_counterpart=True,
),
],
),
],
"siren": [
ClassTypeHintMatch(
base_class="Entity",
matches=_ENTITY_MATCH,
),
ClassTypeHintMatch(
base_class="ToggleEntity",
matches=_TOGGLE_ENTITY_MATCH,
),
ClassTypeHintMatch(
base_class="SirenEntity",
matches=[
TypeHintMatch(
function_name="available_tones",
return_type=["dict[int, str]", "list[int | str]", None],
),
],
),
],
}