jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-20 19:57:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix husqvarna_automower typing (#116060)

Browse Source
This commit is contained in:
J. Nick Koston 2024-04-23 22:50:31 +02:00 committed by GitHub
parent f828eb2301
commit 0f60b404df
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
homeassistant/components/husqvarna_automower/number.py
View File

@ -3,7 +3,7 @@
from collections.abc import Awaitable, Callable
from dataclasses import dataclass
import logging
from typing import Any
from typing import TYPE_CHECKING, Any
from aioautomower.exceptions import ApiException
from aioautomower.model import MowerAttributes
@ -12,7 +12,7 @@ from aioautomower.session import AutomowerSession
from homeassistant.components.number import NumberEntity, NumberEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant
from homeassistant.core import HomeAssistant, callback
from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import AddEntitiesCallback
@ -32,6 +32,15 @@ class AutomowerNumberEntityDescription(NumberEntityDescription):
set_value_fn: Callable[[AutomowerSession, str, float], Awaitable[Any]]
@callback
def _async_get_cutting_height(data: MowerAttributes) -> int:
"""Return the cutting height."""
if TYPE_CHECKING:
# Sensor does not get created if it is None
assert data.cutting_height is not None
return data.cutting_height
NUMBER_TYPES: tuple[AutomowerNumberEntityDescription, ...] = (
AutomowerNumberEntityDescription(
key="cutting_height",
@ -41,7 +50,7 @@ NUMBER_TYPES: tuple[AutomowerNumberEntityDescription, ...] = (
native_min_value=1,
native_max_value=9,
exists_fn=lambda data: data.cutting_height is not None,
value_fn=lambda data: data.cutting_height,
value_fn=_async_get_cutting_height,
set_value_fn=lambda session, mower_id, cheight: session.set_cutting_height(
mower_id, int(cheight)
),