Shared via mypy Playground

main.py

from typing import overload, Self
from enum import StrEnum as _StrEnum


class StrEnum(_StrEnum):

    # mimic str.__new__'s overloads
    @overload
    def __new__(cls, object: object = ...) -> Self: ...
    @overload
    def __new__(cls, object: object, encoding: str = ..., errors: str = ...) -> Self: ...

    def __new__(cls, *values):
        # when we import enum, _StrEnum.__new__ gets moved to _new_member_ when
        # the "final" _StrEnum class is created via EnumType
        return _StrEnum._new_member_(cls, *values)


class Choices(StrEnum):
    LOREM = "lorem"
    IPSUM = "ipsum"


def my_func(obj: Choices) -> None:
    assert isinstance(obj, Choices)

for element in list(Choices):
    my_func(element)