Fix deprecated enums (#134824)

This commit is contained in:
Artur Pragacz 2025-01-14 10:04:48 +01:00 committed by GitHub
parent 440cd5bee0
commit 1de4d0efda
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions

View File

@ -4,7 +4,7 @@ from __future__ import annotations
from collections.abc import Callable
from contextlib import suppress
from enum import Enum, EnumType, _EnumDict
from enum import EnumType, IntEnum, IntFlag, StrEnum, _EnumDict
import functools
import inspect
import logging
@ -255,7 +255,7 @@ class DeprecatedConstant(NamedTuple):
class DeprecatedConstantEnum(NamedTuple):
"""Deprecated constant."""
enum: Enum
enum: StrEnum | IntEnum | IntFlag
breaks_in_ha_version: str | None
@ -306,7 +306,7 @@ def check_if_deprecated_constant(name: str, module_globals: dict[str, Any]) -> A
replacement = deprecated_const.replacement
breaks_in_ha_version = deprecated_const.breaks_in_ha_version
elif isinstance(deprecated_const, DeprecatedConstantEnum):
value = deprecated_const.enum.value
value = deprecated_const.enum
replacement = (
f"{deprecated_const.enum.__class__.__name__}.{deprecated_const.enum.name}"
)

View File

@ -295,7 +295,7 @@ def _get_value(
return obj.value
if isinstance(obj, DeprecatedConstantEnum):
return obj.enum.value
return obj.enum
if isinstance(obj, DeprecatedAlias):
return obj.value