jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-13 00:07:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add switch checks to pylint plugin (#76909)

Browse Source
This commit is contained in:
epenet 2022-08-18 14:48:23 +02:00 committed by GitHub
parent 60c8d95a77
commit c7301a449b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
pylint/plugins/hass_enforce_type_hints.py
View File

@ -2036,6 +2036,25 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
],
),
],
"switch": [
ClassTypeHintMatch(
base_class="Entity",
matches=_ENTITY_MATCH,
),
ClassTypeHintMatch(
base_class="ToggleEntity",
matches=_TOGGLE_ENTITY_MATCH,
),
ClassTypeHintMatch(
base_class="SwitchEntity",
matches=[
TypeHintMatch(
function_name="device_class",
return_type=["SwitchDeviceClass", "str", None],
),
],
),
],
"update": [
ClassTypeHintMatch(
base_class="Entity",