Unable to stubify Enum so it can be used with Literals, without using type ignore
Nadie ha tomado este issue todavía.
- Lenguaje dominante
- Python
- Estrellas
- 20.6k
- Forks
- 3.3k
- Métricas de merge de PR
- Métricas de PR pendientes
Descripción
Bug Report
Extending the example for #3434 I want to define an enum Flag, combine some of the members, and then also reference those members as a TypeAlias in a stub file.
class Color1(Flag):
RED = auto()
BLUE = auto()
GREEN = auto()
WHITE = RED | BLUE | GREEN # <-- This line works at runtime but not in a stub (see the gist)
# Invalid type: try using Literal[Color1.RED] instead
Favorites: TypeAlias = Literal[Color3.RED | Color3.BLUE]
(A clear and concise description of what the bug is.)
To Reproduce
https://mypy-play.net/?mypy=latest&python=3.11&gist=d270d7541c61b2aea87e99c9afc77534
Expected Behavior
I expect mypy to suggest working code, or to accept the | operator for Flag members.
Actual Behavior
See the gist, but depending on how I write the stub I get these errors for the member definition
# foo.pyi:9: error: Invalid type: try using Literal[Color1.RED] instead? [valid-type]
# foo.pyi:9: error: Invalid type: try using Literal[Color1.BLUE] instead? [valid-type]
# foo.pyi:9: error: Invalid type: try using Literal[Color1.GREEN] instead? [valid-type]
or these for the TypeAlias
# foo.pyi:12: error: Parameter 1 of Literal[...] cannot be of type "Any" [valid-type]
Your Environment
- Mypy version used: 1.5.1
- Mypy command-line flags:
--strict - Mypy configuration options from
mypy.ini(and other config files): N/A - Python version used: 3.10.13
I was curious what stubgen came up with:
from _typeshed import Incomplete
from enum import Flag
from typing import TypeAlias
class Color(Flag):
RED: Incomplete
BLUE: Incomplete
GREEN: Incomplete
WHITE: Incomplete
Favorites: TypeAlias
but that loses information about the Literal Favorite Colors.
Guía de contribución
Primeros pasos
- Lee el issue completo y luego la guía de contribución del proyecto.
- Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
- Haz un fork del repositorio y trabaja en una rama.
- Abre un pull request que haga referencia al número del issue.
Línea de trabajo
Comienza con la reproducción en mypy-play y los ejemplos de foo.pyi del informe, centrándote en las definiciones de miembros de Flag y en el Literal TypeAlias. Compara los errores valid-type y Any notificados con el tratamiento esperado de los miembros combinados de Flag; se considera completado cuando el stub previsto conserva la información de color literal sin requerir type ignores.
Escrito por el modelo de indexación a partir del texto del issue.
Evaluación
- Stack tecnológico
- python
- Área
- compilers
- Tipo de issue
- Error
- Dificultad
- 4/5
- Tiempo estimado
- 3-5 días
- Estado de actividad
- Estancado
- Claridad
- Bastante claro
- Aptitud para principiantes
- 35/100