Add setup type hints to sisyphus

This commit is contained in:
epenet
2022-01-10 11:01:17 +00:00
parent 27073abc62
commit 43807f02c3
2 changed files with 26 additions and 2 deletions

View File

@@ -1,11 +1,16 @@
"""Support for the light on the Sisyphus Kinetic Art Table.""" """Support for the light on the Sisyphus Kinetic Art Table."""
from __future__ import annotations
import logging import logging
import aiohttp import aiohttp
from homeassistant.components.light import SUPPORT_BRIGHTNESS, LightEntity from homeassistant.components.light import SUPPORT_BRIGHTNESS, LightEntity
from homeassistant.const import CONF_HOST from homeassistant.const import CONF_HOST
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import PlatformNotReady from homeassistant.exceptions import PlatformNotReady
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from . import DATA_SISYPHUS from . import DATA_SISYPHUS
@@ -14,8 +19,15 @@ _LOGGER = logging.getLogger(__name__)
SUPPORTED_FEATURES = SUPPORT_BRIGHTNESS SUPPORTED_FEATURES = SUPPORT_BRIGHTNESS
async def async_setup_platform(hass, config, add_entities, discovery_info=None): async def async_setup_platform(
hass: HomeAssistant,
config: ConfigType,
add_entities: AddEntitiesCallback,
discovery_info: DiscoveryInfoType | None = None,
) -> None:
"""Set up a single Sisyphus table.""" """Set up a single Sisyphus table."""
if not discovery_info:
return
host = discovery_info[CONF_HOST] host = discovery_info[CONF_HOST]
try: try:
table_holder = hass.data[DATA_SISYPHUS][host] table_holder = hass.data[DATA_SISYPHUS][host]

View File

@@ -1,4 +1,6 @@
"""Support for track controls on the Sisyphus Kinetic Art Table.""" """Support for track controls on the Sisyphus Kinetic Art Table."""
from __future__ import annotations
import aiohttp import aiohttp
from sisyphus_control import Track from sisyphus_control import Track
@@ -21,7 +23,10 @@ from homeassistant.const import (
STATE_PAUSED, STATE_PAUSED,
STATE_PLAYING, STATE_PLAYING,
) )
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import PlatformNotReady from homeassistant.exceptions import PlatformNotReady
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from . import DATA_SISYPHUS from . import DATA_SISYPHUS
@@ -40,8 +45,15 @@ SUPPORTED_FEATURES = (
) )
async def async_setup_platform(hass, config, add_entities, discovery_info=None): async def async_setup_platform(
hass: HomeAssistant,
config: ConfigType,
add_entities: AddEntitiesCallback,
discovery_info: DiscoveryInfoType | None = None,
) -> None:
"""Set up a media player entity for a Sisyphus table.""" """Set up a media player entity for a Sisyphus table."""
if not discovery_info:
return
host = discovery_info[CONF_HOST] host = discovery_info[CONF_HOST]
try: try:
table_holder = hass.data[DATA_SISYPHUS][host] table_holder = hass.data[DATA_SISYPHUS][host]