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

Migrate switch device classes to StrEnum (#60658)

Browse source
This commit is contained in:
Franck Nijhof 2021-12-01 00:38:45 +01:00 • committed by GitHub
parent 98ce12c6ee
commit 542aef2fe1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 34 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

6
homeassistant/components/demo/switch.py
View file

@ -1,7 +1,7 @@
"""Demo platform that has two fake switches."""
from __future__ import annotations
from homeassistant.components.switch import SwitchEntity
from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
from homeassistant.const import DEVICE_DEFAULT_NAME
from homeassistant.helpers.entity import DeviceInfo
@ -19,7 +19,7 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info=
False,
"mdi:air-conditioner",
False,
device_class="outlet",
device_class=SwitchDeviceClass.OUTLET,
),
]
)
@ -42,7 +42,7 @@ class DemoSwitch(SwitchEntity):
state: bool,
icon: str | None,
assumed: bool,
device_class: str | None = None,
device_class: SwitchDeviceClass | None = None,
) -> None:
"""Initialize the Demo switch."""
self._attr_assumed_state = assumed