Improve type hints in scene (#78347)

This commit is contained in:
epenet 2022-09-14 11:18:23 +02:00 committed by GitHub
parent dce2569389
commit abc87b5dfa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -4,7 +4,7 @@ from __future__ import annotations
import functools as ft import functools as ft
import importlib import importlib
import logging import logging
from typing import Any, final from typing import Any, Final, final
import voluptuous as vol import voluptuous as vol
@ -17,13 +17,11 @@ from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import ConfigType from homeassistant.helpers.typing import ConfigType
from homeassistant.util import dt as dt_util from homeassistant.util import dt as dt_util
# mypy: allow-untyped-defs, no-check-untyped-defs DOMAIN: Final = "scene"
STATES: Final = "states"
DOMAIN = "scene"
STATES = "states"
def _hass_domain_validator(config): def _hass_domain_validator(config: dict[str, Any]) -> dict[str, Any]:
"""Validate platform in config for homeassistant domain.""" """Validate platform in config for homeassistant domain."""
if CONF_PLATFORM not in config: if CONF_PLATFORM not in config:
config = {CONF_PLATFORM: HA_DOMAIN, STATES: config} config = {CONF_PLATFORM: HA_DOMAIN, STATES: config}
@ -31,7 +29,7 @@ def _hass_domain_validator(config):
return config return config
def _platform_validator(config): def _platform_validator(config: dict[str, Any]) -> dict[str, Any]:
"""Validate it is a valid platform.""" """Validate it is a valid platform."""
try: try:
platform = importlib.import_module(f".{config[CONF_PLATFORM]}", __name__) platform = importlib.import_module(f".{config[CONF_PLATFORM]}", __name__)
@ -46,7 +44,7 @@ def _platform_validator(config):
if not hasattr(platform, "PLATFORM_SCHEMA"): if not hasattr(platform, "PLATFORM_SCHEMA"):
return config return config
return platform.PLATFORM_SCHEMA(config) return platform.PLATFORM_SCHEMA(config) # type: ignore[no-any-return]
PLATFORM_SCHEMA = vol.Schema( PLATFORM_SCHEMA = vol.Schema(