Strict typing Sensibo (#72454)

This commit is contained in:
G Johansson 2022-05-29 01:26:50 +02:00 committed by GitHub
parent a4f678e7c9
commit 24c34c0ef0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
15 changed files with 136 additions and 40 deletions

View file

@ -1,7 +1,7 @@
"""Base entity for Sensibo integration."""
from __future__ import annotations
from typing import Any
from typing import TYPE_CHECKING, Any
import async_timeout
from pysensibo.model import MotionSensor, SensiboDevice
@ -119,6 +119,8 @@ class SensiboMotionBaseEntity(SensiboBaseEntity):
)
@property
def sensor_data(self) -> MotionSensor:
def sensor_data(self) -> MotionSensor | None:
"""Return data for device."""
if TYPE_CHECKING:
assert self.device_data.motion_sensors
return self.device_data.motion_sensors[self._sensor_id]