Home-Assistant/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2026-02-15 07:36:16 +00:00
Code Activity

Remove str from sensor state class (#161686)

Browse Source
This commit is contained in:
epenet
2026-02-06 12:01:23 +01:00
committed by GitHub
parent 47672614df
commit fbc91d3d3d
3 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/components/sensor/__init__.py
View File

@@ -112,7 +112,7 @@ class SensorEntityDescription(EntityDescription, frozen_or_thawed=True):
last_reset: datetime | None = None last_reset: datetime | None = None
native_unit_of_measurement: str | None = None native_unit_of_measurement: str | None = None
options: list[str] | None = None options: list[str] | None = None
state_class: SensorStateClass | str | None = None state_class: SensorStateClass | None = None
suggested_display_precision: int | None = None suggested_display_precision: int | None = None
suggested_unit_of_measurement: str | None = None suggested_unit_of_measurement: str | None = None
unit_of_measurement: None = None # Type override, use native_unit_of_measurement unit_of_measurement: None = None # Type override, use native_unit_of_measurement
@@ -120,7 +120,7 @@ class SensorEntityDescription(EntityDescription, frozen_or_thawed=True):
def _numeric_state_expected( def _numeric_state_expected(
device_class: SensorDeviceClass | None, device_class: SensorDeviceClass | None,
state_class: SensorStateClass | str | None, state_class: SensorStateClass | None,
native_unit_of_measurement: str | None, native_unit_of_measurement: str | None,
suggested_display_precision: int | None, suggested_display_precision: int | None,
) -> bool: ) -> bool:
@@ -188,7 +188,7 @@ class SensorEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
_attr_native_unit_of_measurement: str | None _attr_native_unit_of_measurement: str | None
_attr_native_value: StateType | date | datetime | Decimal = None _attr_native_value: StateType | date | datetime | Decimal = None
_attr_options: list[str] | None _attr_options: list[str] | None
_attr_state_class: SensorStateClass | str | None _attr_state_class: SensorStateClass | None
_attr_state: None = None # Subclasses of SensorEntity should not set this _attr_state: None = None # Subclasses of SensorEntity should not set this
_attr_suggested_display_precision: int | None _attr_suggested_display_precision: int | None
_attr_suggested_unit_of_measurement: str | None _attr_suggested_unit_of_measurement: str | None
@@ -329,7 +329,7 @@ class SensorEntity(Entity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
return None return None
@cached_property @cached_property
def state_class(self) -> SensorStateClass | str | None: def state_class(self) -> SensorStateClass | None:
"""Return the state class of this entity, if any.""" """Return the state class of this entity, if any."""
if hasattr(self, "_attr_state_class"): if hasattr(self, "_attr_state_class"):
return self._attr_state_class return self._attr_state_class

2
homeassistant/components/sensor/const.py
View File

@@ -835,7 +835,7 @@ DEVICE_CLASS_STATE_CLASSES: dict[SensorDeviceClass, set[SensorStateClass]] = {
} }
STATE_CLASS_UNITS: dict[SensorStateClass | str, set[type[StrEnum] | str | None]] = { STATE_CLASS_UNITS: dict[SensorStateClass, set[type[StrEnum] | str | None]] = {
SensorStateClass.MEASUREMENT_ANGLE: {DEGREE}, SensorStateClass.MEASUREMENT_ANGLE: {DEGREE},
} }

2
pylint/plugins/hass_enforce_type_hints.py
View File

@@ -2526,7 +2526,7 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
), ),
TypeHintMatch( TypeHintMatch(
function_name="state_class", function_name="state_class",
return_type=["SensorStateClass", "str", None], return_type=["SensorStateClass", None],
mandatory=True, mandatory=True,
), ),
TypeHintMatch( TypeHintMatch(