homeassistant-stubs 2024.5.4__py3-none-any.whl → 2024.6.0__py3-none-any.whl

homeassistant-stubs/components/husqvarna_automower/lawn_mower.pyi

@@ -0,0 +1,28 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import AutomowerDataUpdateCoordinator as AutomowerDataUpdateCoordinator
+from .entity import AutomowerControlEntity as AutomowerControlEntity
+from _typeshed import Incomplete
+from homeassistant.components.lawn_mower import LawnMowerActivity as LawnMowerActivity, LawnMowerEntity as LawnMowerEntity, LawnMowerEntityFeature as LawnMowerEntityFeature
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+
+SUPPORT_STATE_SERVICES: Incomplete
+DOCKED_ACTIVITIES: Incomplete
+MOWING_ACTIVITIES: Incomplete
+PAUSED_STATES: Incomplete
+_LOGGER: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AutomowerLawnMowerEntity(AutomowerControlEntity, LawnMowerEntity):
+    _attr_name: Incomplete
+    _attr_supported_features = SUPPORT_STATE_SERVICES
+    _attr_unique_id: Incomplete
+    def __init__(self, mower_id: str, coordinator: AutomowerDataUpdateCoordinator) -> None: ...
+    @property
+    def activity(self) -> LawnMowerActivity: ...
+    async def async_start_mowing(self) -> None: ...
+    async def async_pause(self) -> None: ...
+    async def async_dock(self) -> None: ...

homeassistant-stubs/components/husqvarna_automower/number.pyi

@@ -0,0 +1,67 @@
+from .const import DOMAIN as DOMAIN, EXECUTION_TIME_DELAY as EXECUTION_TIME_DELAY
+from .coordinator import AutomowerDataUpdateCoordinator as AutomowerDataUpdateCoordinator
+from .entity import AutomowerControlEntity as AutomowerControlEntity
+from _typeshed import Incomplete
+from aioautomower.model import MowerAttributes as MowerAttributes, WorkArea as WorkArea
+from aioautomower.session import AutomowerSession as AutomowerSession
+from collections.abc import Awaitable, Callable as Callable
+from dataclasses import dataclass
+from homeassistant.components.number import NumberEntity as NumberEntity, NumberEntityDescription as NumberEntityDescription
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, Platform as Platform
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from typing import Any
+
+_LOGGER: Incomplete
+
+def _async_get_cutting_height(data: MowerAttributes) -> int: ...
+def _work_area_translation_key(work_area_id: int) -> str: ...
+async def async_set_work_area_cutting_height(coordinator: AutomowerDataUpdateCoordinator, mower_id: str, cheight: float, work_area_id: int) -> None: ...
+async def async_set_cutting_height(session: AutomowerSession, mower_id: str, cheight: float) -> None: ...
+
+@dataclass(frozen=True, kw_only=True)
+class AutomowerNumberEntityDescription(NumberEntityDescription):
+    exists_fn: Callable[[MowerAttributes], bool] = ...
+    value_fn: Callable[[MowerAttributes], int]
+    set_value_fn: Callable[[AutomowerSession, str, float], Awaitable[Any]]
+    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, max_value, min_value, mode, native_max_value, native_min_value, native_step, native_unit_of_measurement, step, exists_fn, value_fn, set_value_fn) -> None: ...
+
+NUMBER_TYPES: tuple[AutomowerNumberEntityDescription, ...]
+
+@dataclass(frozen=True, kw_only=True)
+class AutomowerWorkAreaNumberEntityDescription(NumberEntityDescription):
+    value_fn: Callable[[WorkArea], int]
+    translation_key_fn: Callable[[int], str]
+    set_value_fn: Callable[[AutomowerDataUpdateCoordinator, str, float, int], Awaitable[Any]]
+    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, max_value, min_value, mode, native_max_value, native_min_value, native_step, native_unit_of_measurement, step, value_fn, translation_key_fn, set_value_fn) -> None: ...
+
+WORK_AREA_NUMBER_TYPES: tuple[AutomowerWorkAreaNumberEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AutomowerNumberEntity(AutomowerControlEntity, NumberEntity):
+    entity_description: AutomowerNumberEntityDescription
+    _attr_unique_id: Incomplete
+    def __init__(self, mower_id: str, coordinator: AutomowerDataUpdateCoordinator, description: AutomowerNumberEntityDescription) -> None: ...
+    @property
+    def native_value(self) -> float: ...
+    async def async_set_native_value(self, value: float) -> None: ...
+
+class AutomowerWorkAreaNumberEntity(AutomowerControlEntity, NumberEntity):
+    entity_description: AutomowerWorkAreaNumberEntityDescription
+    coordinator: Incomplete
+    work_area_id: Incomplete
+    _attr_unique_id: Incomplete
+    _attr_translation_placeholders: Incomplete
+    def __init__(self, mower_id: str, coordinator: AutomowerDataUpdateCoordinator, description: AutomowerWorkAreaNumberEntityDescription, work_area_id: int) -> None: ...
+    @property
+    def work_area(self) -> WorkArea: ...
+    @property
+    def translation_key(self) -> str: ...
+    @property
+    def native_value(self) -> float: ...
+    async def async_set_native_value(self, value: float) -> None: ...
+
+def async_remove_entities(hass: HomeAssistant, coordinator: AutomowerDataUpdateCoordinator, config_entry: ConfigEntry, mower_id: str) -> None: ...

homeassistant-stubs/components/husqvarna_automower/select.pyi

@@ -0,0 +1,25 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import AutomowerDataUpdateCoordinator as AutomowerDataUpdateCoordinator
+from .entity import AutomowerControlEntity as AutomowerControlEntity
+from _typeshed import Incomplete
+from homeassistant.components.select import SelectEntity as SelectEntity
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import EntityCategory as EntityCategory
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+
+_LOGGER: Incomplete
+HEADLIGHT_MODES: list
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AutomowerSelectEntity(AutomowerControlEntity, SelectEntity):
+    _attr_options = HEADLIGHT_MODES
+    _attr_entity_category: Incomplete
+    _attr_translation_key: str
+    _attr_unique_id: Incomplete
+    def __init__(self, mower_id: str, coordinator: AutomowerDataUpdateCoordinator) -> None: ...
+    @property
+    def current_option(self) -> str: ...
+    async def async_select_option(self, option: str) -> None: ...

homeassistant-stubs/components/husqvarna_automower/sensor.pyi

@@ -0,0 +1,35 @@
+from .const import DOMAIN as DOMAIN
+from .coordinator import AutomowerDataUpdateCoordinator as AutomowerDataUpdateCoordinator
+from .entity import AutomowerBaseEntity as AutomowerBaseEntity
+from _typeshed import Incomplete
+from aioautomower.model import MowerAttributes as MowerAttributes
+from collections.abc import Callable as Callable
+from dataclasses import dataclass
+from datetime import datetime
+from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription, SensorStateClass as SensorStateClass
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import EntityCategory as EntityCategory, PERCENTAGE as PERCENTAGE, UnitOfLength as UnitOfLength, UnitOfTime as UnitOfTime
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.typing import StateType as StateType
+
+_LOGGER: Incomplete
+ERROR_KEY_LIST: Incomplete
+RESTRICTED_REASONS: list
+
+@dataclass(frozen=True, kw_only=True)
+class AutomowerSensorEntityDescription(SensorEntityDescription):
+    exists_fn: Callable[[MowerAttributes], bool] = ...
+    value_fn: Callable[[MowerAttributes], StateType | datetime]
+    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, exists_fn, value_fn) -> None: ...
+
+SENSOR_TYPES: tuple[AutomowerSensorEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AutomowerSensorEntity(AutomowerBaseEntity, SensorEntity):
+    entity_description: AutomowerSensorEntityDescription
+    _attr_unique_id: Incomplete
+    def __init__(self, mower_id: str, coordinator: AutomowerDataUpdateCoordinator, description: AutomowerSensorEntityDescription) -> None: ...
+    @property
+    def native_value(self) -> StateType | datetime: ...

homeassistant-stubs/components/husqvarna_automower/switch.pyi

@@ -0,0 +1,49 @@
+from .const import DOMAIN as DOMAIN, EXECUTION_TIME_DELAY as EXECUTION_TIME_DELAY
+from .coordinator import AutomowerDataUpdateCoordinator as AutomowerDataUpdateCoordinator
+from .entity import AutomowerControlEntity as AutomowerControlEntity
+from _typeshed import Incomplete
+from aioautomower.model import StayOutZones as StayOutZones, Zone as Zone
+from homeassistant.components.switch import SwitchEntity as SwitchEntity
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import Platform as Platform
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.exceptions import HomeAssistantError as HomeAssistantError
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from typing import Any
+
+_LOGGER: Incomplete
+ERROR_ACTIVITIES: Incomplete
+ERROR_STATES: Incomplete
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class AutomowerScheduleSwitchEntity(AutomowerControlEntity, SwitchEntity):
+    _attr_translation_key: str
+    _attr_unique_id: Incomplete
+    def __init__(self, mower_id: str, coordinator: AutomowerDataUpdateCoordinator) -> None: ...
+    @property
+    def is_on(self) -> bool: ...
+    @property
+    def available(self) -> bool: ...
+    async def async_turn_off(self, **kwargs: Any) -> None: ...
+    async def async_turn_on(self, **kwargs: Any) -> None: ...
+
+class AutomowerStayOutZoneSwitchEntity(AutomowerControlEntity, SwitchEntity):
+    _attr_translation_key: str
+    coordinator: Incomplete
+    stay_out_zone_uid: Incomplete
+    _attr_unique_id: Incomplete
+    _attr_translation_placeholders: Incomplete
+    def __init__(self, coordinator: AutomowerDataUpdateCoordinator, mower_id: str, stay_out_zone_uid: str) -> None: ...
+    @property
+    def stay_out_zones(self) -> StayOutZones: ...
+    @property
+    def stay_out_zone(self) -> Zone: ...
+    @property
+    def is_on(self) -> bool: ...
+    @property
+    def available(self) -> bool: ...
+    async def async_turn_off(self, **kwargs: Any) -> None: ...
+    async def async_turn_on(self, **kwargs: Any) -> None: ...
+
+def async_remove_entities(hass: HomeAssistant, coordinator: AutomowerDataUpdateCoordinator, config_entry: ConfigEntry, mower_id: str) -> None: ...

homeassistant-stubs/components/hydrawise/binary_sensor.pyi

@@ -2,18 +2,28 @@ from .const import DOMAIN as DOMAIN
 from .coordinator import HydrawiseDataUpdateCoordinator as HydrawiseDataUpdateCoordinator
 from .entity import HydrawiseEntity as HydrawiseEntity
 from _typeshed import Incomplete
+from collections.abc import Callable as Callable
+from dataclasses import dataclass
 from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, BinarySensorEntityDescription as BinarySensorEntityDescription
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from pydrawise.schema import Zone as Zone
 
-BINARY_SENSOR_STATUS: Incomplete
-BINARY_SENSOR_TYPES: tuple[BinarySensorEntityDescription, ...]
-BINARY_SENSOR_KEYS: list[str]
+@dataclass(frozen=True, kw_only=True)
+class HydrawiseBinarySensorEntityDescription(BinarySensorEntityDescription):
+    value_fn: Callable[[HydrawiseBinarySensor], bool | None]
+    always_available: bool = ...
+    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, value_fn, always_available) -> None: ...
+
+CONTROLLER_BINARY_SENSORS: tuple[HydrawiseBinarySensorEntityDescription, ...]
+RAIN_SENSOR_BINARY_SENSOR: tuple[HydrawiseBinarySensorEntityDescription, ...]
+ZONE_BINARY_SENSORS: tuple[HydrawiseBinarySensorEntityDescription, ...]
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class HydrawiseBinarySensor(HydrawiseEntity, BinarySensorEntity):
+    entity_description: HydrawiseBinarySensorEntityDescription
     _attr_is_on: Incomplete
     def _update_attrs(self) -> None: ...
+    @property
+    def available(self) -> bool: ...

homeassistant-stubs/components/hydrawise/coordinator.pyi

@@ -3,17 +3,20 @@ from dataclasses import dataclass
 from datetime import timedelta
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
-from pydrawise import HydrawiseBase as HydrawiseBase
-from pydrawise.schema import Controller as Controller, User as User, Zone as Zone
+from homeassistant.util.dt import now as now
+from pydrawise import Hydrawise as Hydrawise
+from pydrawise.schema import Controller as Controller, ControllerWaterUseSummary, Sensor as Sensor, User as User, Zone as Zone
 
 @dataclass
 class HydrawiseData:
     user: User
     controllers: dict[int, Controller]
     zones: dict[int, Zone]
-    def __init__(self, user, controllers, zones) -> None: ...
+    sensors: dict[int, Sensor]
+    daily_water_use: dict[int, ControllerWaterUseSummary]
+    def __init__(self, user, controllers, zones, sensors, daily_water_use) -> None: ...
 
 class HydrawiseDataUpdateCoordinator(DataUpdateCoordinator[HydrawiseData]):
-    api: HydrawiseBase
-    def __init__(self, hass: HomeAssistant, api: HydrawiseBase, scan_interval: timedelta) -> None: ...
+    api: Hydrawise
+    def __init__(self, hass: HomeAssistant, api: Hydrawise, scan_interval: timedelta) -> None: ...
     async def _async_update_data(self) -> HydrawiseData: ...

homeassistant-stubs/components/hydrawise/entity.pyi

@@ -5,17 +5,24 @@ from homeassistant.core import callback as callback
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
 from homeassistant.helpers.entity import EntityDescription as EntityDescription
 from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
-from pydrawise.schema import Controller as Controller, Zone as Zone
+from pydrawise.schema import Controller as Controller, Sensor as Sensor, Zone as Zone
 
 class HydrawiseEntity(CoordinatorEntity[HydrawiseDataUpdateCoordinator]):
     _attr_attribution: str
     _attr_has_entity_name: bool
     entity_description: Incomplete
     controller: Incomplete
-    zone: Incomplete
+    zone_id: Incomplete
+    sensor_id: Incomplete
     _device_id: Incomplete
     _attr_unique_id: Incomplete
     _attr_device_info: Incomplete
-    def __init__(self, coordinator: HydrawiseDataUpdateCoordinator, description: EntityDescription, controller: Controller, zone: Zone | None = None) -> None: ...
+    def __init__(self, coordinator: HydrawiseDataUpdateCoordinator, description: EntityDescription, controller: Controller, *, zone_id: int | None = None, sensor_id: int | None = None) -> None: ...
+    @property
+    def zone(self) -> Zone: ...
+    @property
+    def sensor(self) -> Sensor: ...
     def _update_attrs(self) -> None: ...
     def _handle_coordinator_update(self) -> None: ...
+    @property
+    def available(self) -> bool: ...

homeassistant-stubs/components/hydrawise/sensor.pyi

@@ -2,21 +2,38 @@ from .const import DOMAIN as DOMAIN
 from .coordinator import HydrawiseDataUpdateCoordinator as HydrawiseDataUpdateCoordinator
 from .entity import HydrawiseEntity as HydrawiseEntity
 from _typeshed import Incomplete
+from collections.abc import Callable as Callable
+from dataclasses import dataclass
+from datetime import datetime
 from homeassistant.components.sensor import SensorDeviceClass as SensorDeviceClass, SensorEntity as SensorEntity, SensorEntityDescription as SensorEntityDescription
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
-from homeassistant.const import UnitOfTime as UnitOfTime
+from homeassistant.const import UnitOfTime as UnitOfTime, UnitOfVolume as UnitOfVolume
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from pydrawise.schema import Zone as Zone
+from typing import Any
 
-SENSOR_TYPES: tuple[SensorEntityDescription, ...]
-SENSOR_KEYS: list[str]
-TWO_YEAR_SECONDS: Incomplete
-WATERING_TIME_ICON: str
+@dataclass(frozen=True, kw_only=True)
+class HydrawiseSensorEntityDescription(SensorEntityDescription):
+    value_fn: Callable[[HydrawiseSensor], Any]
+    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, value_fn) -> None: ...
+
+def _get_zone_watering_time(sensor: HydrawiseSensor) -> int: ...
+def _get_zone_next_cycle(sensor: HydrawiseSensor) -> datetime | None: ...
+def _get_zone_daily_active_water_use(sensor: HydrawiseSensor) -> float: ...
+def _get_controller_daily_active_water_use(sensor: HydrawiseSensor) -> float: ...
+def _get_controller_daily_inactive_water_use(sensor: HydrawiseSensor) -> float | None: ...
+def _get_controller_daily_total_water_use(sensor: HydrawiseSensor) -> float | None: ...
+
+FLOW_CONTROLLER_SENSORS: tuple[HydrawiseSensorEntityDescription, ...]
+FLOW_ZONE_SENSORS: tuple[SensorEntityDescription, ...]
+ZONE_SENSORS: tuple[HydrawiseSensorEntityDescription, ...]
+FLOW_MEASUREMENT_KEYS: Incomplete
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class HydrawiseSensor(HydrawiseEntity, SensorEntity):
-    zone: Zone
+    entity_description: HydrawiseSensorEntityDescription
+    @property
+    def icon(self) -> str | None: ...
     _attr_native_value: Incomplete
     def _update_attrs(self) -> None: ...

homeassistant-stubs/components/hydrawise/switch.pyi

@@ -1,19 +1,29 @@
 from .const import DEFAULT_WATERING_TIME as DEFAULT_WATERING_TIME, DOMAIN as DOMAIN
 from .coordinator import HydrawiseDataUpdateCoordinator as HydrawiseDataUpdateCoordinator
 from .entity import HydrawiseEntity as HydrawiseEntity
+from collections.abc import Callable as Callable, Coroutine
+from dataclasses import dataclass
 from homeassistant.components.switch import SwitchDeviceClass as SwitchDeviceClass, SwitchEntity as SwitchEntity, SwitchEntityDescription as SwitchEntityDescription
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
-from pydrawise.schema import Zone as Zone
+from pydrawise import Hydrawise as Hydrawise, Zone as Zone
 from typing import Any
 
-SWITCH_TYPES: tuple[SwitchEntityDescription, ...]
+@dataclass(frozen=True, kw_only=True)
+class HydrawiseSwitchEntityDescription(SwitchEntityDescription):
+    turn_on_fn: Callable[[Hydrawise, Zone], Coroutine[Any, Any, None]]
+    turn_off_fn: Callable[[Hydrawise, Zone], Coroutine[Any, Any, None]]
+    value_fn: Callable[[Zone], bool]
+    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, turn_on_fn, turn_off_fn, value_fn) -> None: ...
+
+SWITCH_TYPES: tuple[HydrawiseSwitchEntityDescription, ...]
 SWITCH_KEYS: list[str]
 
 async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class HydrawiseSwitch(HydrawiseEntity, SwitchEntity):
+    entity_description: HydrawiseSwitchEntityDescription
     zone: Zone
     _attr_is_on: bool
     async def async_turn_on(self, **kwargs: Any) -> None: ...

homeassistant-stubs/components/ibeacon/__init__.pyi

@@ -2,8 +2,10 @@ from .const import DOMAIN as DOMAIN, PLATFORMS as PLATFORMS
 from .coordinator import IBeaconCoordinator as IBeaconCoordinator
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.core import HomeAssistant as HomeAssistant
-from homeassistant.helpers.device_registry import DeviceEntry as DeviceEntry, async_get as async_get
+from homeassistant.helpers.device_registry import DeviceEntry as DeviceEntry
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
+IBeaconConfigEntry = ConfigEntry[IBeaconCoordinator]
+
+async def async_setup_entry(hass: HomeAssistant, entry: IBeaconConfigEntry) -> bool: ...
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: ...
-async def async_remove_config_entry_device(hass: HomeAssistant, config_entry: ConfigEntry, device_entry: DeviceEntry) -> bool: ...
+async def async_remove_config_entry_device(hass: HomeAssistant, config_entry: IBeaconConfigEntry, device_entry: DeviceEntry) -> bool: ...

homeassistant-stubs/components/ibeacon/device_tracker.pyi

@@ -1,17 +1,17 @@
-from .const import DOMAIN as DOMAIN, SIGNAL_IBEACON_DEVICE_NEW as SIGNAL_IBEACON_DEVICE_NEW
+from . import IBeaconConfigEntry as IBeaconConfigEntry
+from .const import SIGNAL_IBEACON_DEVICE_NEW as SIGNAL_IBEACON_DEVICE_NEW
 from .coordinator import IBeaconCoordinator as IBeaconCoordinator
 from .entity import IBeaconEntity as IBeaconEntity
 from _typeshed import Incomplete
 from homeassistant.components.device_tracker import SourceType as SourceType
 from homeassistant.components.device_tracker.config_entry import BaseTrackerEntity as BaseTrackerEntity
-from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import STATE_HOME as STATE_HOME, STATE_NOT_HOME as STATE_NOT_HOME
 from homeassistant.core import 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 ibeacon_ble import iBeaconAdvertisement as iBeaconAdvertisement
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: IBeaconConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class IBeaconTrackerEntity(IBeaconEntity, BaseTrackerEntity):
     _attr_name: Incomplete

homeassistant-stubs/components/ibeacon/sensor.pyi

@@ -1,11 +1,11 @@
-from .const import DOMAIN as DOMAIN, SIGNAL_IBEACON_DEVICE_NEW as SIGNAL_IBEACON_DEVICE_NEW
+from . import IBeaconConfigEntry as IBeaconConfigEntry
+from .const import SIGNAL_IBEACON_DEVICE_NEW as SIGNAL_IBEACON_DEVICE_NEW
 from .coordinator import IBeaconCoordinator as IBeaconCoordinator
 from .entity import IBeaconEntity as IBeaconEntity
 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.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import SIGNAL_STRENGTH_DECIBELS_MILLIWATT as SIGNAL_STRENGTH_DECIBELS_MILLIWATT, UnitOfLength as UnitOfLength
 from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect as async_dispatcher_connect
@@ -19,7 +19,7 @@ class IBeaconSensorEntityDescription(SensorEntityDescription):
 
 SENSOR_DESCRIPTIONS: Incomplete
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+async def async_setup_entry(hass: HomeAssistant, entry: IBeaconConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
 
 class IBeaconSensorEntity(IBeaconEntity, SensorEntity):
     entity_description: IBeaconSensorEntityDescription

homeassistant-stubs/components/idasen_desk/__init__.pyi

@@ -1,27 +1,17 @@
-import logging
 from .const import DOMAIN as DOMAIN
+from .coordinator import IdasenDeskCoordinator as IdasenDeskCoordinator
 from _typeshed import Incomplete
 from homeassistant.components import bluetooth as bluetooth
+from homeassistant.components.bluetooth.match import ADDRESS as ADDRESS, BluetoothCallbackMatcher as BluetoothCallbackMatcher
 from homeassistant.config_entries import ConfigEntry as ConfigEntry
 from homeassistant.const import ATTR_NAME as ATTR_NAME, CONF_ADDRESS as CONF_ADDRESS, EVENT_HOMEASSISTANT_STOP as EVENT_HOMEASSISTANT_STOP, Platform as Platform
 from homeassistant.core import Event as Event, HomeAssistant as HomeAssistant, callback as callback
 from homeassistant.exceptions import ConfigEntryNotReady as ConfigEntryNotReady
 from homeassistant.helpers.device_registry import DeviceInfo as DeviceInfo
-from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
 
 PLATFORMS: list[Platform]
 _LOGGER: Incomplete
 
-class IdasenDeskCoordinator(DataUpdateCoordinator[int | None]):
-    _address: Incomplete
-    _expected_connected: bool
-    _connection_lost: bool
-    desk: Incomplete
-    def __init__(self, hass: HomeAssistant, logger: logging.Logger, name: str, address: str) -> None: ...
-    async def async_connect(self) -> bool: ...
-    async def async_disconnect(self) -> None: ...
-    def async_set_updated_data(self, data: int | None) -> None: ...
-
 class DeskData:
     address: str
     device_info: DeviceInfo

homeassistant-stubs/components/idasen_desk/coordinator.pyi

@@ -0,0 +1,19 @@
+import logging
+from _typeshed import Incomplete
+from homeassistant.components import bluetooth as bluetooth
+from homeassistant.core import HomeAssistant as HomeAssistant, callback as callback
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator
+
+_LOGGER: Incomplete
+
+class IdasenDeskCoordinator(DataUpdateCoordinator[int | None]):
+    _address: Incomplete
+    _expected_connected: bool
+    _connection_lost: bool
+    _disconnect_lock: Incomplete
+    desk: Incomplete
+    def __init__(self, hass: HomeAssistant, logger: logging.Logger, name: str, address: str) -> None: ...
+    async def async_connect(self) -> bool: ...
+    async def async_disconnect(self) -> None: ...
+    async def async_ensure_connection_state(self) -> None: ...
+    def async_set_updated_data(self, data: int | None) -> None: ...

homeassistant-stubs/components/imgw_pib/__init__.pyi

@@ -0,0 +1,21 @@
+from .const import CONF_STATION_ID as CONF_STATION_ID
+from .coordinator import ImgwPibDataUpdateCoordinator as ImgwPibDataUpdateCoordinator
+from _typeshed import Incomplete
+from dataclasses import dataclass
+from homeassistant.config_entries import ConfigEntry as ConfigEntry
+from homeassistant.const import Platform as Platform
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.exceptions import ConfigEntryNotReady as ConfigEntryNotReady
+from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
+
+PLATFORMS: list[Platform]
+_LOGGER: Incomplete
+ImgwPibConfigEntry = ConfigEntry[ImgwPibData]
+
+@dataclass
+class ImgwPibData:
+    coordinator: ImgwPibDataUpdateCoordinator
+    def __init__(self, coordinator) -> None: ...
+
+async def async_setup_entry(hass: HomeAssistant, entry: ImgwPibConfigEntry) -> bool: ...
+async def async_unload_entry(hass: HomeAssistant, entry: ImgwPibConfigEntry) -> bool: ...

homeassistant-stubs/components/imgw_pib/binary_sensor.pyi

@@ -0,0 +1,28 @@
+from . import ImgwPibConfigEntry as ImgwPibConfigEntry
+from .coordinator import ImgwPibDataUpdateCoordinator as ImgwPibDataUpdateCoordinator
+from .entity import ImgwPibEntity as ImgwPibEntity
+from _typeshed import Incomplete
+from collections.abc import Callable as Callable
+from dataclasses import dataclass
+from homeassistant.components.binary_sensor import BinarySensorDeviceClass as BinarySensorDeviceClass, BinarySensorEntity as BinarySensorEntity, BinarySensorEntityDescription as BinarySensorEntityDescription
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from imgw_pib.model import HydrologicalData as HydrologicalData
+
+PARALLEL_UPDATES: int
+
+@dataclass(frozen=True, kw_only=True)
+class ImgwPibBinarySensorEntityDescription(BinarySensorEntityDescription):
+    value: Callable[[HydrologicalData], bool | None]
+    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, value) -> None: ...
+
+BINARY_SENSOR_TYPES: tuple[ImgwPibBinarySensorEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, entry: ImgwPibConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class ImgwPibBinarySensorEntity(ImgwPibEntity, BinarySensorEntity):
+    entity_description: ImgwPibBinarySensorEntityDescription
+    _attr_unique_id: Incomplete
+    def __init__(self, coordinator: ImgwPibDataUpdateCoordinator, description: ImgwPibBinarySensorEntityDescription) -> None: ...
+    @property
+    def is_on(self) -> bool | None: ...

homeassistant-stubs/components/imgw_pib/config_flow.pyi

@@ -0,0 +1,12 @@
+from .const import CONF_STATION_ID as CONF_STATION_ID, DOMAIN as DOMAIN
+from _typeshed import Incomplete
+from homeassistant.config_entries import ConfigFlow as ConfigFlow, ConfigFlowResult as ConfigFlowResult
+from homeassistant.helpers.aiohttp_client import async_get_clientsession as async_get_clientsession
+from homeassistant.helpers.selector import SelectOptionDict as SelectOptionDict, SelectSelector as SelectSelector, SelectSelectorConfig as SelectSelectorConfig, SelectSelectorMode as SelectSelectorMode
+from typing import Any
+
+_LOGGER: Incomplete
+
+class ImgwPibFlowHandler(ConfigFlow, domain=DOMAIN):
+    VERSION: int
+    async def async_step_user(self, user_input: dict[str, Any] | None = None) -> ConfigFlowResult: ...

homeassistant-stubs/components/imgw_pib/const.pyi

@@ -0,0 +1,6 @@
+from _typeshed import Incomplete
+
+DOMAIN: str
+ATTRIBUTION: str
+CONF_STATION_ID: str
+UPDATE_INTERVAL: Incomplete

homeassistant-stubs/components/imgw_pib/coordinator.pyi

@@ -0,0 +1,15 @@
+from .const import DOMAIN as DOMAIN, UPDATE_INTERVAL as UPDATE_INTERVAL
+from _typeshed import Incomplete
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.device_registry import DeviceEntryType as DeviceEntryType, DeviceInfo as DeviceInfo
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator as DataUpdateCoordinator, UpdateFailed as UpdateFailed
+from imgw_pib import HydrologicalData, ImgwPib as ImgwPib
+
+_LOGGER: Incomplete
+
+class ImgwPibDataUpdateCoordinator(DataUpdateCoordinator[HydrologicalData]):
+    imgwpib: Incomplete
+    station_id: Incomplete
+    device_info: Incomplete
+    def __init__(self, hass: HomeAssistant, imgwpib: ImgwPib, station_id: str) -> None: ...
+    async def _async_update_data(self) -> HydrologicalData: ...

homeassistant-stubs/components/imgw_pib/diagnostics.pyi

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

homeassistant-stubs/components/imgw_pib/entity.pyi

@@ -0,0 +1,10 @@
+from .const import ATTRIBUTION as ATTRIBUTION
+from .coordinator import ImgwPibDataUpdateCoordinator as ImgwPibDataUpdateCoordinator
+from _typeshed import Incomplete
+from homeassistant.helpers.update_coordinator import CoordinatorEntity as CoordinatorEntity
+
+class ImgwPibEntity(CoordinatorEntity[ImgwPibDataUpdateCoordinator]):
+    _attr_attribution = ATTRIBUTION
+    _attr_has_entity_name: bool
+    _attr_device_info: Incomplete
+    def __init__(self, coordinator: ImgwPibDataUpdateCoordinator) -> None: ...

homeassistant-stubs/components/imgw_pib/sensor.pyi

@@ -0,0 +1,30 @@
+from . import ImgwPibConfigEntry as ImgwPibConfigEntry
+from .coordinator import ImgwPibDataUpdateCoordinator as ImgwPibDataUpdateCoordinator
+from .entity import ImgwPibEntity as ImgwPibEntity
+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 EntityCategory as EntityCategory, UnitOfLength as UnitOfLength, UnitOfTemperature as UnitOfTemperature
+from homeassistant.core import HomeAssistant as HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback as AddEntitiesCallback
+from homeassistant.helpers.typing import StateType as StateType
+from imgw_pib.model import HydrologicalData as HydrologicalData
+
+PARALLEL_UPDATES: int
+
+@dataclass(frozen=True, kw_only=True)
+class ImgwPibSensorEntityDescription(SensorEntityDescription):
+    value: Callable[[HydrologicalData], StateType]
+    def __init__(self, *, key, device_class, entity_category, entity_registry_enabled_default, entity_registry_visible_default, force_update, icon, has_entity_name, name, translation_key, translation_placeholders, unit_of_measurement, last_reset, native_unit_of_measurement, options, state_class, suggested_display_precision, suggested_unit_of_measurement, value) -> None: ...
+
+SENSOR_TYPES: tuple[ImgwPibSensorEntityDescription, ...]
+
+async def async_setup_entry(hass: HomeAssistant, entry: ImgwPibConfigEntry, async_add_entities: AddEntitiesCallback) -> None: ...
+
+class ImgwPibSensorEntity(ImgwPibEntity, SensorEntity):
+    entity_description: ImgwPibSensorEntityDescription
+    _attr_unique_id: Incomplete
+    def __init__(self, coordinator: ImgwPibDataUpdateCoordinator, description: ImgwPibSensorEntityDescription) -> None: ...
+    @property
+    def native_value(self) -> StateType: ...