From abc87b5dfabfc2381cba15bd6461d4f9a8efb585 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 14 Sep 2022 11:18:23 +0200 Subject: [PATCH] Improve type hints in scene (#78347) --- homeassistant/components/scene/__init__.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/homeassistant/components/scene/__init__.py b/homeassistant/components/scene/__init__.py index 5dea5965d43..813f5138ed1 100644 --- a/homeassistant/components/scene/__init__.py +++ b/homeassistant/components/scene/__init__.py @@ -4,7 +4,7 @@ from __future__ import annotations import functools as ft import importlib import logging -from typing import Any, final +from typing import Any, Final, final import voluptuous as vol @@ -17,13 +17,11 @@ from homeassistant.helpers.restore_state import RestoreEntity from homeassistant.helpers.typing import ConfigType from homeassistant.util import dt as dt_util -# mypy: allow-untyped-defs, no-check-untyped-defs - -DOMAIN = "scene" -STATES = "states" +DOMAIN: Final = "scene" +STATES: Final = "states" -def _hass_domain_validator(config): +def _hass_domain_validator(config: dict[str, Any]) -> dict[str, Any]: """Validate platform in config for homeassistant domain.""" if CONF_PLATFORM not in config: config = {CONF_PLATFORM: HA_DOMAIN, STATES: config} @@ -31,7 +29,7 @@ def _hass_domain_validator(config): return config -def _platform_validator(config): +def _platform_validator(config: dict[str, Any]) -> dict[str, Any]: """Validate it is a valid platform.""" try: platform = importlib.import_module(f".{config[CONF_PLATFORM]}", __name__) @@ -46,7 +44,7 @@ def _platform_validator(config): if not hasattr(platform, "PLATFORM_SCHEMA"): return config - return platform.PLATFORM_SCHEMA(config) + return platform.PLATFORM_SCHEMA(config) # type: ignore[no-any-return] PLATFORM_SCHEMA = vol.Schema(