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

Add return type annotation to StrEnum (#60624)

Browse Source
This commit is contained in:
Marc Mueller 2021-11-30 16:26:02 +01:00 committed by GitHub
parent ad75c217ce
commit a90ef488a1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
homeassistant/util/enum.py
View File

@ -2,13 +2,15 @@
from __future__ import annotations from __future__ import annotations
from enum import Enum from enum import Enum
from typing import Any from typing import Any, TypeVar
T = TypeVar("T", bound="StrEnum")
class StrEnum(str, Enum): class StrEnum(str, Enum):
"""Partial backport of Python 3.11's StrEnum for our basic use cases.""" """Partial backport of Python 3.11's StrEnum for our basic use cases."""
def __new__(cls, value: str, *args: Any, **kwargs: Any): # type: ignore[no-untyped-def] def __new__(cls: type[T], value: str, *args: Any, **kwargs: Any) -> T:
"""Create a new StrEnum instance.""" """Create a new StrEnum instance."""
if not isinstance(value, str): if not isinstance(value, str):
raise TypeError(f"{value!r} is not a string") raise TypeError(f"{value!r} is not a string")