homeassistant-stubs 2025.1.4__py3-none-any.whl → 2025.2.0__py3-none-any.whl

homeassistant-stubs/components/tplink/light.pyi

@@ -1,17 +1,19 @@
 from . import TPLinkConfigEntry as TPLinkConfigEntry, legacy_device_id as legacy_device_id
+from .const import DOMAIN as DOMAIN
 from .coordinator import TPLinkDataUpdateCoordinator as TPLinkDataUpdateCoordinator
-from .entity import CoordinatedTPLinkEntity as CoordinatedTPLinkEntity, async_refresh_after as async_refresh_after
+from .entity import CoordinatedTPLinkModuleEntity as CoordinatedTPLinkModuleEntity, TPLinkModuleEntityDescription as TPLinkModuleEntityDescription, async_refresh_after as async_refresh_after
 from _typeshed import Incomplete
 from collections.abc import Sequence
-from homeassistant.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS, ATTR_COLOR_TEMP_KELVIN as ATTR_COLOR_TEMP_KELVIN, ATTR_EFFECT as ATTR_EFFECT, ATTR_HS_COLOR as ATTR_HS_COLOR, ATTR_TRANSITION as ATTR_TRANSITION, ColorMode as ColorMode, EFFECT_OFF as EFFECT_OFF, LightEntity as LightEntity, LightEntityFeature as LightEntityFeature, filter_supported_color_modes as filter_supported_color_modes
+from dataclasses import dataclass
+from homeassistant.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS, ATTR_COLOR_TEMP_KELVIN as ATTR_COLOR_TEMP_KELVIN, ATTR_EFFECT as ATTR_EFFECT, ATTR_HS_COLOR as ATTR_HS_COLOR, ATTR_TRANSITION as ATTR_TRANSITION, ColorMode as ColorMode, EFFECT_OFF as EFFECT_OFF, LightEntity as LightEntity, LightEntityDescription as LightEntityDescription, LightEntityFeature as LightEntityFeature, filter_supported_color_modes as filter_supported_color_modes
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
-from homeassistant.helpers import entity_platform as entity_platform
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from homeassistant.helpers.typing import VolDictType as VolDictType
 from kasa import Device as Device
-from kasa.interfaces import Light as Light, LightEffect
 from typing import Any
 
+PARALLEL_UPDATES: int
 _LOGGER: Incomplete
 SERVICE_RANDOM_EFFECT: str
 SERVICE_SEQUENCE_EFFECT: str
@@ -26,19 +28,26 @@ RANDOM_EFFECT_DICT: VolDictType
 
 @callback
 def _async_build_base_effect(brightness: int, duration: int, transition: int, segments: list[int]) -> dict[str, Any]: ...
+def _get_backwards_compatible_light_unique_id(device: Device, entity_description: TPLinkModuleEntityDescription) -> str: ...
+
+@dataclass(frozen=True, kw_only=True)
+class TPLinkLightEntityDescription(LightEntityDescription, TPLinkModuleEntityDescription):
+    unique_id_fn = _get_backwards_compatible_light_unique_id
+
+LIGHT_DESCRIPTIONS: tuple[TPLinkLightEntityDescription, ...]
+LIGHT_EFFECT_DESCRIPTIONS: tuple[TPLinkLightEntityDescription, ...]
+
 async def async_setup_entry(hass: HomeAssistant, config_entry: TPLinkConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
-class TPLinkLightEntity(CoordinatedTPLinkEntity, LightEntity):
+class TPLinkLightEntity(CoordinatedTPLinkModuleEntity, LightEntity):
     _attr_supported_features: Incomplete
     _fixed_color_mode: ColorMode | None
-    _parent: Incomplete
+    entity_description: TPLinkLightEntityDescription
     _light_module: Incomplete
-    _attr_name: Incomplete
     _attr_min_color_temp_kelvin: Incomplete
     _attr_max_color_temp_kelvin: Incomplete
     _attr_supported_color_modes: Incomplete
-    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, *, light_module: Light, parent: Device | None = None) -> None: ...
-    def _get_unique_id(self) -> str: ...
+    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, description: TPLinkLightEntityDescription, *, parent: Device | None = None) -> None: ...
     @callback
     def _async_extract_brightness_transition(self, **kwargs: Any) -> tuple[int | None, int | None]: ...
     async def _async_set_hsv(self, hs_color: tuple[int, int], brightness: int | None, transition: int | None) -> None: ...
@@ -58,9 +67,11 @@ class TPLinkLightEntity(CoordinatedTPLinkEntity, LightEntity):
     def _async_update_attrs(self) -> bool: ...
 
 class TPLinkLightEffectEntity(TPLinkLightEntity):
-    _effect_module: Incomplete
-    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, *, light_module: Light, effect_module: LightEffect) -> None: ...
     _attr_supported_features: Incomplete
+    _effect_module: Incomplete
+    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, description: TPLinkLightEntityDescription, *, parent: Device | None = None) -> None: ...
+    async def async_added_to_hass(self) -> None: ...
+    def _register_effects_services(self) -> None: ...
     _attr_effect: Incomplete
     _attr_color_mode: Incomplete
     _attr_effect_list: Incomplete

homeassistant-stubs/components/tplink/number.pyi

@@ -13,6 +13,7 @@ _LOGGER: Incomplete
 @dataclass(frozen=True, kw_only=True)
 class TPLinkNumberEntityDescription(NumberEntityDescription, TPLinkFeatureEntityDescription): ...
 
+PARALLEL_UPDATES: int
 NUMBER_DESCRIPTIONS: Final[Incomplete]
 NUMBER_DESCRIPTIONS_MAP: Incomplete
 

homeassistant-stubs/components/tplink/select.pyi

@@ -11,6 +11,7 @@ from typing import Final
 @dataclass(frozen=True, kw_only=True)
 class TPLinkSelectEntityDescription(SelectEntityDescription, TPLinkFeatureEntityDescription): ...
 
+PARALLEL_UPDATES: int
 SELECT_DESCRIPTIONS: Final[Incomplete]
 SELECT_DESCRIPTIONS_MAP: Incomplete
 

homeassistant-stubs/components/tplink/sensor.pyi

@@ -1,16 +1,21 @@
 from . import TPLinkConfigEntry as TPLinkConfigEntry
 from .const import UNIT_MAPPING as UNIT_MAPPING
-from .deprecate import async_cleanup_deprecated as async_cleanup_deprecated
 from .entity import CoordinatedTPLinkFeatureEntity as CoordinatedTPLinkFeatureEntity, TPLinkFeatureEntityDescription as TPLinkFeatureEntityDescription
 from _typeshed import Incomplete
+from collections.abc import Callable as Callable
 from dataclasses import dataclass
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
+from homeassistant.const import UnitOfTime as UnitOfTime
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from typing import Any
 
 @dataclass(frozen=True, kw_only=True)
-class TPLinkSensorEntityDescription(SensorEntityDescription, TPLinkFeatureEntityDescription): ...
+class TPLinkSensorEntityDescription(SensorEntityDescription, TPLinkFeatureEntityDescription):
+    convert_fn: Callable[[Any], Any] | None = ...
 
+PARALLEL_UPDATES: int
+_TOTAL_SECONDS_METHOD_CALLER: Incomplete
 SENSOR_DESCRIPTIONS: tuple[TPLinkSensorEntityDescription, ...]
 SENSOR_DESCRIPTIONS_MAP: Incomplete
 

homeassistant-stubs/components/tplink/siren.pyi

@@ -1,25 +1,40 @@
-from . import TPLinkConfigEntry as TPLinkConfigEntry
+from . import TPLinkConfigEntry as TPLinkConfigEntry, legacy_device_id as legacy_device_id
+from .const import DOMAIN as DOMAIN
 from .coordinator import TPLinkDataUpdateCoordinator as TPLinkDataUpdateCoordinator
-from .entity import CoordinatedTPLinkEntity as CoordinatedTPLinkEntity, async_refresh_after as async_refresh_after
+from .entity import CoordinatedTPLinkModuleEntity as CoordinatedTPLinkModuleEntity, TPLinkModuleEntityDescription as TPLinkModuleEntityDescription, async_refresh_after as async_refresh_after
 from _typeshed import Incomplete
-from homeassistant.components.siren import SirenEntity as SirenEntity, SirenEntityFeature as SirenEntityFeature
+from collections.abc import Callable as Callable
+from dataclasses import dataclass
+from homeassistant.components.siren import ATTR_DURATION as ATTR_DURATION, ATTR_TONE as ATTR_TONE, ATTR_VOLUME_LEVEL as ATTR_VOLUME_LEVEL, SirenEntity as SirenEntity, SirenEntityDescription as SirenEntityDescription, SirenEntityFeature as SirenEntityFeature, SirenTurnOnServiceParameters as SirenTurnOnServiceParameters
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.exceptions import ServiceValidationError as ServiceValidationError
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from kasa import Device as Device
-from kasa.smart.modules.alarm import Alarm as Alarm
 from typing import Any
 
+PARALLEL_UPDATES: int
+
+@dataclass(frozen=True, kw_only=True)
+class TPLinkSirenEntityDescription(SirenEntityDescription, TPLinkModuleEntityDescription):
+    unique_id_fn: Callable[[Device, TPLinkModuleEntityDescription], str] = ...
+
+SIREN_DESCRIPTIONS: tuple[TPLinkSirenEntityDescription, ...]
+
 async def async_setup_entry(hass: HomeAssistant, config_entry: TPLinkConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
-class TPLinkSirenEntity(CoordinatedTPLinkEntity, SirenEntity):
+class TPLinkSirenEntity(CoordinatedTPLinkModuleEntity, SirenEntity):
     _attr_name: Incomplete
     _attr_supported_features: Incomplete
-    _alarm_module: Alarm
-    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator) -> None: ...
+    entity_description: TPLinkSirenEntityDescription
+    _alarm_module: Incomplete
+    _alarm_volume_max: Incomplete
+    _alarm_duration_max: Incomplete
+    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, description: TPLinkSirenEntityDescription, *, parent: Device | None = None) -> None: ...
     @async_refresh_after
     async def async_turn_on(self, **kwargs: Any) -> None: ...
     @async_refresh_after
     async def async_turn_off(self, **kwargs: Any) -> None: ...
     _attr_is_on: Incomplete
+    _attr_available_tones: Incomplete
     @callback
     def _async_update_attrs(self) -> bool: ...

homeassistant-stubs/components/tplink/switch.pyi

@@ -12,6 +12,7 @@ _LOGGER: Incomplete
 @dataclass(frozen=True, kw_only=True)
 class TPLinkSwitchEntityDescription(SwitchEntityDescription, TPLinkFeatureEntityDescription): ...
 
+PARALLEL_UPDATES: int
 SWITCH_DESCRIPTIONS: tuple[TPLinkSwitchEntityDescription, ...]
 SWITCH_DESCRIPTIONS_MAP: Incomplete
 

homeassistant-stubs/components/tplink/vacuum.pyi

@@ -0,0 +1,43 @@
+from . import TPLinkConfigEntry as TPLinkConfigEntry
+from .coordinator import TPLinkDataUpdateCoordinator as TPLinkDataUpdateCoordinator
+from .entity import CoordinatedTPLinkModuleEntity as CoordinatedTPLinkModuleEntity, TPLinkModuleEntityDescription as TPLinkModuleEntityDescription, async_refresh_after as async_refresh_after
+from _typeshed import Incomplete
+from dataclasses import dataclass
+from homeassistant.components.vacuum import StateVacuumEntity as StateVacuumEntity, StateVacuumEntityDescription as StateVacuumEntityDescription, VacuumActivity as VacuumActivity, VacuumEntityFeature as VacuumEntityFeature
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from kasa import Device as Device
+from kasa.smart.modules.clean import Clean as Clean
+from typing import Any
+
+PARALLEL_UPDATES: int
+STATUS_TO_ACTIVITY: Incomplete
+
+@dataclass(frozen=True, kw_only=True)
+class TPLinkVacuumEntityDescription(StateVacuumEntityDescription, TPLinkModuleEntityDescription): ...
+
+VACUUM_DESCRIPTIONS: tuple[TPLinkVacuumEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, config_entry: TPLinkConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class TPLinkVacuumEntity(CoordinatedTPLinkModuleEntity, StateVacuumEntity):
+    _attr_supported_features: Incomplete
+    entity_description: TPLinkVacuumEntityDescription
+    _vacuum_module: Clean
+    _speaker_module: Incomplete
+    _attr_fan_speed_list: Incomplete
+    def __init__(self, device: Device, coordinator: TPLinkDataUpdateCoordinator, description: TPLinkVacuumEntityDescription, *, parent: Device) -> None: ...
+    @async_refresh_after
+    async def async_start(self) -> None: ...
+    @async_refresh_after
+    async def async_pause(self) -> None: ...
+    @async_refresh_after
+    async def async_return_to_base(self, **kwargs: Any) -> None: ...
+    @async_refresh_after
+    async def async_set_fan_speed(self, fan_speed: str, **kwargs: Any) -> None: ...
+    async def async_locate(self, **kwargs: Any) -> None: ...
+    @property
+    def battery_level(self) -> int | None: ...
+    _attr_activity: Incomplete
+    _attr_fan_speed: Incomplete
+    def _async_update_attrs(self) -> bool: ...

homeassistant-stubs/components/tradfri/config_flow.pyi

@@ -1,9 +1,9 @@
 from .const import CONF_GATEWAY_ID as CONF_GATEWAY_ID, CONF_IDENTITY as CONF_IDENTITY, CONF_KEY as CONF_KEY, DOMAIN as DOMAIN
 from _typeshed import Incomplete
-from homeassistant.components import zeroconf as zeroconf
 from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult
 from homeassistant.const import CONF_HOST as CONF_HOST
 from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.service_info.zeroconf import ATTR_PROPERTIES_ID as ATTR_PROPERTIES_ID, ZeroconfServiceInfo as ZeroconfServiceInfo
 from typing import Any
 
 KEY_SECURITY_CODE: str
@@ -18,7 +18,7 @@ class FlowHandler(ConfigFlow, domain=DOMAIN):
     def __init__(self) -> None: ...
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_auth(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
-    async def async_step_homekit(self, discovery_info: zeroconf.ZeroconfServiceInfo) -> ConfigFlowResult: ...
+    async def async_step_homekit(self, discovery_info: ZeroconfServiceInfo) -> ConfigFlowResult: ...
     async def _entry_from_data(self, data: dict[str, Any]) -> ConfigFlowResult: ...
 
 async def authenticate(hass: HomeAssistant, host: str, security_code: str) -> dict[str, str | bool]: ...

homeassistant-stubs/components/trafikverket_camera/__init__.pyi

@@ -11,5 +11,5 @@ _LOGGER: Incomplete
 TVCameraConfigEntry = ConfigEntry[TVDataUpdateCoordinator]
 
 async def async_setup_entry(hass: HomeAssistant, entry: TVCameraConfigEntry) -> bool: ...
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: TVCameraConfigEntry) -> bool: ...
+async def async_migrate_entry(hass: HomeAssistant, entry: TVCameraConfigEntry) -> bool: ...

homeassistant-stubs/components/trafikverket_camera/camera.pyi

@@ -10,6 +10,8 @@ from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from typing import Any
 
+PARALLEL_UPDATES: int
+
 async def async_setup_entry(hass: HomeAssistant, entry: TVCameraConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class TVCamera(TrafikverketCameraEntity, Camera):

homeassistant-stubs/components/trafikverket_ferry/__init__.pyi

@@ -6,4 +6,4 @@ from homeassistant.core import HomeAssistant as HomeAssistant
 TVFerryConfigEntry = ConfigEntry[TVDataUpdateCoordinator]
 
 async def async_setup_entry(hass: HomeAssistant, entry: TVFerryConfigEntry) -> bool: ...
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: TVFerryConfigEntry) -> bool: ...

homeassistant-stubs/components/trafikverket_ferry/coordinator.pyi

@@ -23,5 +23,5 @@ class TVDataUpdateCoordinator(DataUpdateCoordinator):
     _to: str
     _time: time | None
     _weekdays: list[str]
-    def __init__(self, hass: HomeAssistant) -> None: ...
+    def __init__(self, hass: HomeAssistant, config_entry: TVFerryConfigEntry) -> None: ...
     async def _async_update_data(self) -> dict[str, Any]: ...

homeassistant-stubs/components/trafikverket_ferry/sensor.pyi

@@ -20,6 +20,7 @@ ATTR_TO: str
 ATTR_MODIFIED_TIME: str
 ATTR_OTHER_INFO: str
 SCAN_INTERVAL: Incomplete
+PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class TrafikverketSensorEntityDescription(SensorEntityDescription):

homeassistant-stubs/components/trafikverket_train/__init__.pyi

@@ -1,13 +1,16 @@
-from .const import PLATFORMS as PLATFORMS
+from .const import CONF_FROM as CONF_FROM, CONF_TO as CONF_TO, PLATFORMS as PLATFORMS
 from .coordinator import TVDataUpdateCoordinator as TVDataUpdateCoordinator
 from _typeshed import Incomplete
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import CONF_API_KEY as CONF_API_KEY
 from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed
+from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 
 TVTrainConfigEntry = ConfigEntry[TVDataUpdateCoordinator]
 _LOGGER: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, entry: TVTrainConfigEntry) -> bool: ...
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None: ...
+async def async_unload_entry(hass: HomeAssistant, entry: TVTrainConfigEntry) -> bool: ...
+async def update_listener(hass: HomeAssistant, entry: TVTrainConfigEntry) -> None: ...
 async def async_migrate_entry(hass: HomeAssistant, entry: TVTrainConfigEntry) -> bool: ...

homeassistant-stubs/components/trafikverket_train/config_flow.pyi

@@ -1,12 +1,12 @@
 from .const import CONF_FILTER_PRODUCT as CONF_FILTER_PRODUCT, CONF_FROM as CONF_FROM, CONF_TIME as CONF_TIME, CONF_TO as CONF_TO, DOMAIN as DOMAIN
-from .util import next_departuredate as next_departuredate
 from _typeshed import Incomplete
 from collections.abc import Mapping
-from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow
+from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow, SOURCE_RECONFIGURE as SOURCE_RECONFIGURE
 from homeassistant.const import CONF_API_KEY as CONF_API_KEY, CONF_NAME as CONF_NAME, CONF_WEEKDAY as CONF_WEEKDAY, WEEKDAYS as WEEKDAYS
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
-from homeassistant.helpers.selector import SelectSelector as SelectSelector, SelectSelectorConfig as SelectSelectorConfig, SelectSelectorMode as SelectSelectorMode, TextSelector as TextSelector, TimeSelector as TimeSelector
+from homeassistant.helpers.selector import SelectOptionDict as SelectOptionDict, SelectSelector as SelectSelector, SelectSelectorConfig as SelectSelectorConfig, SelectSelectorMode as SelectSelectorMode, TextSelector as TextSelector, TimeSelector as TimeSelector
+from pytrafikverket import StationInfoModel as StationInfoModel
 from typing import Any
 
 _LOGGER: Incomplete
@@ -14,17 +14,23 @@ OPTION_SCHEMA: Incomplete
 DATA_SCHEMA: Incomplete
 DATA_SCHEMA_REAUTH: Incomplete
 
-async def validate_input(hass: HomeAssistant, api_key: str, train_from: str, train_to: str, train_time: str | None, weekdays: list[str], product_filter: str | None) -> dict[str, str]: ...
+async def validate_station(hass: HomeAssistant, api_key: str, train_station: str, field: str) -> tuple[list[StationInfoModel], dict[str, str]]: ...
 
 class TVTrainConfigFlow(ConfigFlow, domain=DOMAIN):
     VERSION: int
     MINOR_VERSION: int
+    _from_stations: list[StationInfoModel]
+    _to_stations: list[StationInfoModel]
+    _data: dict[str, Any]
     @staticmethod
     @callback
     def async_get_options_flow(config_entry: ConfigEntry) -> TVTrainOptionsFlowHandler: ...
     async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> ConfigFlowResult: ...
     async def async_step_reauth_confirm(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_step_reconfigure(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_step_initial(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
+    async def async_step_select_stations(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
 
 class TVTrainOptionsFlowHandler(OptionsFlow):
     async def async_step_init(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/trafikverket_train/coordinator.pyi

@@ -9,7 +9,7 @@ from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed as ConfigEntryAuthFailed
 from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
-from pytrafikverket.models import StationInfoModel as StationInfoModel, TrainStopModel as TrainStopModel
+from pytrafikverket import StationInfoModel as StationInfoModel, TrainStopModel as TrainStopModel
 
 @dataclass
 class TrainData:
@@ -40,6 +40,6 @@ class TVDataUpdateCoordinator(DataUpdateCoordinator[TrainData]):
     _time: time | None
     _weekdays: list[str]
     _filter_product: str | None
-    def __init__(self, hass: HomeAssistant) -> None: ...
+    def __init__(self, hass: HomeAssistant, config_entry: TVTrainConfigEntry) -> None: ...
     async def _async_setup(self) -> None: ...
     async def _async_update_data(self) -> TrainData: ...

homeassistant-stubs/components/trafikverket_train/sensor.pyi

@@ -15,6 +15,7 @@ from homeassistant.helpers.update_coordinator import CoordinatorEntity as Coordi
 from typing import Any
 
 ATTR_PRODUCT_FILTER: str
+PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class TrafikverketSensorEntityDescription(SensorEntityDescription):

homeassistant-stubs/components/trafikverket_weatherstation/__init__.pyi

@@ -6,4 +6,4 @@ from homeassistant.core import HomeAssistant as HomeAssistant
 TVWeatherConfigEntry = ConfigEntry[TVDataUpdateCoordinator]
 
 async def async_setup_entry(hass: HomeAssistant, entry: TVWeatherConfigEntry) -> bool: ...
-async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: TVWeatherConfigEntry) -> bool: ...

homeassistant-stubs/components/trafikverket_weatherstation/coordinator.pyi

@@ -15,5 +15,5 @@ class TVDataUpdateCoordinator(DataUpdateCoordinator[WeatherStationInfoModel]):
     config_entry: TVWeatherConfigEntry
     _weather_api: Incomplete
     _station: Incomplete
-    def __init__(self, hass: HomeAssistant) -> None: ...
+    def __init__(self, hass: HomeAssistant, config_entry: TVWeatherConfigEntry) -> None: ...
     async def _async_update_data(self) -> WeatherStationInfoModel: ...

homeassistant-stubs/components/trafikverket_weatherstation/diagnostics.pyi

@@ -0,0 +1,5 @@
+from . import TVWeatherConfigEntry as TVWeatherConfigEntry
+from homeassistant.core import HomeAssistant as HomeAssistant
+from typing import Any
+
+async def async_get_config_entry_diagnostics(hass: HomeAssistant, entry: TVWeatherConfigEntry) -> dict[str, Any]: ...

homeassistant-stubs/components/trafikverket_weatherstation/sensor.pyi

@@ -15,6 +15,7 @@ from homeassistant.helpers.update_coordinator import CoordinatorEntity as Coordi
 from pytrafikverket.models import WeatherStationInfoModel as WeatherStationInfoModel
 
 PRECIPITATION_TYPE: Incomplete
+PARALLEL_UPDATES: int
 
 @dataclass(frozen=True, kw_only=True)
 class TrafikverketSensorEntityDescription(SensorEntityDescription):

homeassistant-stubs/components/tts/__init__.pyi

@@ -10,10 +10,10 @@ from homeassistant.components.http import HomeAssistantView
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.restore_state import RestoreEntity
 from homeassistant.helpers.typing import ConfigType
-from propcache import cached_property
+from propcache.api import cached_property
 from typing import Any, TypedDict, final
 
-__all__ = ['async_default_engine', 'async_get_media_source_audio', 'async_support_options', 'ATTR_AUDIO_OUTPUT', 'ATTR_PREFERRED_FORMAT', 'ATTR_PREFERRED_SAMPLE_RATE', 'ATTR_PREFERRED_SAMPLE_CHANNELS', 'ATTR_PREFERRED_SAMPLE_BYTES', 'CONF_LANG', 'DEFAULT_CACHE_DIR', 'generate_media_source_id', 'PLATFORM_SCHEMA_BASE', 'PLATFORM_SCHEMA', 'SampleFormat', 'Provider', 'TtsAudioType', 'Voice']
+__all__ = ['ATTR_AUDIO_OUTPUT', 'ATTR_PREFERRED_FORMAT', 'ATTR_PREFERRED_SAMPLE_BYTES', 'ATTR_PREFERRED_SAMPLE_CHANNELS', 'ATTR_PREFERRED_SAMPLE_RATE', 'CONF_LANG', 'DEFAULT_CACHE_DIR', 'PLATFORM_SCHEMA', 'PLATFORM_SCHEMA_BASE', 'Provider', 'SampleFormat', 'TtsAudioType', 'Voice', 'async_default_engine', 'async_get_media_source_audio', 'async_support_options', 'generate_media_source_id']
 
 ATTR_AUDIO_OUTPUT: str
 ATTR_PREFERRED_FORMAT: str

homeassistant-stubs/components/unifi/config_flow.pyi

@@ -6,11 +6,11 @@ from .hub import UnifiHub as UnifiHub, get_unifi_api as get_unifi_api
 from _typeshed import Incomplete
 from aiounifi.interfaces.sites import Sites as Sites
 from collections.abc import Mapping
-from homeassistant.components import ssdp as ssdp
 from homeassistant.config_entries import ConfigEntryState as ConfigEntryState, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow, SOURCE_REAUTH as SOURCE_REAUTH
 from homeassistant.const import CONF_HOST as CONF_HOST, CONF_PASSWORD as CONF_PASSWORD, CONF_PORT as CONF_PORT, CONF_USERNAME as CONF_USERNAME, CONF_VERIFY_SSL as CONF_VERIFY_SSL
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.device_registry import format_mac as format_mac
+from homeassistant.helpers.service_info.ssdp import ATTR_UPNP_MODEL_DESCRIPTION as ATTR_UPNP_MODEL_DESCRIPTION, ATTR_UPNP_SERIAL as ATTR_UPNP_SERIAL, SsdpServiceInfo as SsdpServiceInfo
 from typing import Any
 
 DEFAULT_PORT: int
@@ -30,7 +30,7 @@ class UnifiFlowHandler(ConfigFlow, domain=UNIFI_DOMAIN):
     async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_site(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...
     async def async_step_reauth(self, entry_data: Mapping[str, Any]) -> ConfigFlowResult: ...
-    async def async_step_ssdp(self, discovery_info: ssdp.SsdpServiceInfo) -> ConfigFlowResult: ...
+    async def async_step_ssdp(self, discovery_info: SsdpServiceInfo) -> ConfigFlowResult: ...
 
 class UnifiOptionsFlowHandler(OptionsFlow):
     hub: UnifiHub

homeassistant-stubs/components/unifi/device_tracker.pyi

@@ -13,7 +13,7 @@ from homeassistant.components.device_tracker import ScannerEntity as ScannerEnti
 from homeassistant.core import Event as core_Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from propcache import cached_property
+from propcache.api import cached_property
 from typing import Any
 
 LOGGER: Incomplete

homeassistant-stubs/components/unifi/switch.pyi

@@ -36,6 +36,7 @@ async def async_outlet_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> N
 async def async_poe_port_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 async def async_port_forward_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 async def async_traffic_rule_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
+async def async_traffic_route_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 async def async_wlan_control_fn(hub: UnifiHub, obj_id: str, target: bool) -> None: ...
 
 @dataclass(frozen=True, kw_only=True)

homeassistant-stubs/components/unifiprotect/__init__.pyi

@@ -4,7 +4,7 @@ from .discovery import async_start_discovery as async_start_discovery
 from .migrate import async_migrate_data as async_migrate_data
 from .services import async_setup_services as async_setup_services
 from .utils import _async_unifi_mac_from_hass as _async_unifi_mac_from_hass, async_create_api_client as async_create_api_client, async_get_devices as async_get_devices
-from .views import ThumbnailProxyView as ThumbnailProxyView, VideoEventProxyView as VideoEventProxyView, VideoProxyView as VideoProxyView
+from .views import SnapshotProxyView as SnapshotProxyView, ThumbnailProxyView as ThumbnailProxyView, VideoEventProxyView as VideoEventProxyView, VideoProxyView as VideoProxyView
 from _typeshed import Incomplete
 from homeassistant.const import EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP
 from homeassistant.core import HomeAssistant as HomeAssistant

homeassistant-stubs/components/unifiprotect/config_flow.pyi

@@ -4,11 +4,12 @@ from .discovery import async_start_discovery as async_start_discovery
 from .utils import _async_resolve as _async_resolve, _async_short_mac as _async_short_mac, _async_unifi_mac_from_hass as _async_unifi_mac_from_hass
 from _typeshed import Incomplete
 from collections.abc import Mapping
-from homeassistant.components import dhcp as dhcp, ssdp as ssdp
 from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigEntryState as ConfigEntryState, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow, SOURCE_IGNORE as SOURCE_IGNORE
 from homeassistant.const import CONF_HOST as CONF_HOST, CONF_ID as CONF_ID, CONF_PASSWORD as CONF_PASSWORD, CONF_PORT as CONF_PORT, CONF_USERNAME as CONF_USERNAME, CONF_VERIFY_SSL as CONF_VERIFY_SSL
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.aiohttp_client import async_create_clientsession as async_create_clientsession, async_get_clientsession as async_get_clientsession
+from homeassistant.helpers.service_info.dhcp import DhcpServiceInfo as DhcpServiceInfo
+from homeassistant.helpers.service_info.ssdp import SsdpServiceInfo as SsdpServiceInfo
 from homeassistant.helpers.storage import STORAGE_DIR as STORAGE_DIR
 from homeassistant.helpers.typing import DiscoveryInfoType as DiscoveryInfoType
 from homeassistant.loader import async_get_integration as async_get_integration
@@ -27,8 +28,8 @@ class ProtectFlowHandler(ConfigFlow, domain=DOMAIN):
     VERSION: int
     _discovered_device: dict[str, str]
     def __init__(self) -> None: ...
-    async def async_step_dhcp(self, discovery_info: dhcp.DhcpServiceInfo) -> ConfigFlowResult: ...
-    async def async_step_ssdp(self, discovery_info: ssdp.SsdpServiceInfo) -> ConfigFlowResult: ...
+    async def async_step_dhcp(self, discovery_info: DhcpServiceInfo) -> ConfigFlowResult: ...
+    async def async_step_ssdp(self, discovery_info: SsdpServiceInfo) -> ConfigFlowResult: ...
     async def _async_discovery_handoff(self) -> ConfigFlowResult: ...
     async def async_step_integration_discovery(self, discovery_info: DiscoveryInfoType) -> ConfigFlowResult: ...
     async def async_step_discovery_confirm(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/unifiprotect/light.pyi

@@ -1,7 +1,7 @@
 from .data import ProtectDeviceType as ProtectDeviceType, UFPConfigEntry as UFPConfigEntry
 from .entity import ProtectDeviceEntity as ProtectDeviceEntity
 from _typeshed import Incomplete
-from homeassistant.components.light import ATTR_BRIGHTNESS as ATTR_BRIGHTNESS, ColorMode as ColorMode, LightEntity as LightEntity
+from homeassistant.components.light import ColorMode as ColorMode, LightEntity as LightEntity
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 from typing import Any

homeassistant-stubs/components/unifiprotect/views.pyi

@@ -1,6 +1,7 @@
 from .data import ProtectData as ProtectData, async_get_data_for_entry_id as async_get_data_for_entry_id, async_get_data_for_nvr_id as async_get_data_for_nvr_id
 from _typeshed import Incomplete
 from aiohttp import web
+from datetime import datetime
 from homeassistant.components.http import HomeAssistantView as HomeAssistantView
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from http import HTTPStatus
@@ -12,6 +13,8 @@ _LOGGER: Incomplete
 @callback
 def async_generate_thumbnail_url(event_id: str, nvr_id: str, width: int | None = None, height: int | None = None) -> str: ...
 @callback
+def async_generate_snapshot_url(nvr_id: str, camera_id: str, timestamp: datetime, width: int | None = None, height: int | None = None) -> str: ...
+@callback
 def async_generate_event_video_url(event: Event) -> str: ...
 @callback
 def async_generate_proxy_event_video_url(nvr_id: str, event_id: str) -> str: ...
@@ -39,6 +42,11 @@ class ThumbnailProxyView(ProtectProxyView):
     name: str
     async def get(self, request: web.Request, nvr_id: str, event_id: str) -> web.Response: ...
 
+class SnapshotProxyView(ProtectProxyView):
+    url: str
+    name: str
+    async def get(self, request: web.Request, nvr_id: str, camera_id: str, timestamp: str) -> web.Response: ...
+
 class VideoProxyView(ProtectProxyView):
     url: str
     name: str

homeassistant-stubs/components/upcloud/__init__.pyi

@@ -1,8 +1,6 @@
-import dataclasses
-from .const import CONFIG_ENTRY_UPDATE_SIGNAL_TEMPLATE as CONFIG_ENTRY_UPDATE_SIGNAL_TEMPLATE, DATA_UPCLOUD as DATA_UPCLOUD, DEFAULT_SCAN_INTERVAL as DEFAULT_SCAN_INTERVAL
-from .coordinator import UpCloudDataUpdateCoordinator as UpCloudDataUpdateCoordinator
+from .const import CONFIG_ENTRY_UPDATE_SIGNAL_TEMPLATE as CONFIG_ENTRY_UPDATE_SIGNAL_TEMPLATE, DEFAULT_SCAN_INTERVAL as DEFAULT_SCAN_INTERVAL
+from .coordinator import UpCloudConfigEntry as UpCloudConfigEntry, UpCloudDataUpdateCoordinator as UpCloudDataUpdateCoordinator
 from _typeshed import Incomplete
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import CONF_PASSWORD as CONF_PASSWORD, CONF_SCAN_INTERVAL as CONF_SCAN_INTERVAL, CONF_USERNAME as CONF_USERNAME, Platform as Platform
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.exceptions import ConfigEntryNotReady as ConfigEntryNotReady
@@ -11,11 +9,7 @@ from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_d
 _LOGGER: Incomplete
 PLATFORMS: Incomplete
 
-@dataclasses.dataclass
-class UpCloudHassData:
-    coordinators: dict[str, UpCloudDataUpdateCoordinator] = dataclasses.field(default_factory=dict)
-
-def _config_entry_update_signal_name(config_entry: ConfigEntry) -> str: ...
-async def _async_signal_options_update(hass: HomeAssistant, config_entry: ConfigEntry) -> None: ...
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_unload_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool: ...
+def _config_entry_update_signal_name(config_entry: UpCloudConfigEntry) -> str: ...
+async def _async_signal_options_update(hass: HomeAssistant, config_entry: UpCloudConfigEntry) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: UpCloudConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: UpCloudConfigEntry) -> bool: ...

homeassistant-stubs/components/upcloud/binary_sensor.pyi

@@ -1,13 +1,11 @@
-from .const import DATA_UPCLOUD as DATA_UPCLOUD
+from .coordinator import UpCloudConfigEntry as UpCloudConfigEntry
 from .entity import UpCloudServerEntity as UpCloudServerEntity
 from _typeshed import Incomplete
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import CONF_USERNAME as CONF_USERNAME
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
 
-async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, config_entry: UpCloudConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class UpCloudBinarySensor(UpCloudServerEntity, BinarySensorEntity):
     _attr_device_class: Incomplete

homeassistant-stubs/components/upcloud/config_flow.pyi

@@ -1,6 +1,7 @@
 from .const import DEFAULT_SCAN_INTERVAL as DEFAULT_SCAN_INTERVAL, DOMAIN as DOMAIN
+from .coordinator import UpCloudConfigEntry as UpCloudConfigEntry
 from _typeshed import Incomplete
-from homeassistant.config_entries import ConfigEntry as ConfigEntry, ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow
+from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult, OptionsFlow as OptionsFlow
 from homeassistant.const import CONF_PASSWORD as CONF_PASSWORD, CONF_SCAN_INTERVAL as CONF_SCAN_INTERVAL, CONF_USERNAME as CONF_USERNAME
 from homeassistant.core import callback as callback
 from typing import Any
@@ -16,7 +17,7 @@ class UpCloudConfigFlow(ConfigFlow, domain=DOMAIN):
     def _async_show_form(self, step_id: str, user_input: dict[str, Any] | None = None, errors: dict[str, str] | None = None) -> ConfigFlowResult: ...
     @staticmethod
     @callback
-    def async_get_options_flow(config_entry: ConfigEntry) -> UpCloudOptionsFlow: ...
+    def async_get_options_flow(config_entry: UpCloudConfigEntry) -> UpCloudOptionsFlow: ...
 
 class UpCloudOptionsFlow(OptionsFlow):
     async def async_step_init(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/upcloud/const.pyi

@@ -1,6 +1,5 @@
 from _typeshed import Incomplete
 
 DOMAIN: str
-DATA_UPCLOUD: str
 DEFAULT_SCAN_INTERVAL: Incomplete
 CONFIG_ENTRY_UPDATE_SIGNAL_TEMPLATE: Incomplete