mirrors/discord.py
1
0
Fork 0
mirror of https://github.com/Rapptz/discord.py.git synced 2025-04-16 06:03:11 +00:00
Code Issues Projects Releases Wiki Activity

Add support for enum values that aren't int, str, or float

Browse Source 
This coerces them into string options to allow them to be transformed
in the actual code.
This commit is contained in:
Rapptz 2022-04-18 20:15:45 -04:00
parent 88b22611bc
commit 2e02f618f0
1 changed files with 22 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
discord/app_commands/transformers.py
View File

@ -378,6 +378,24 @@ def _make_enum_transformer(enum) -> Type[Transformer]:
return type(f'{enum.__name__}EnumTransformer', (Transformer,), ns)
def _make_complex_enum_transformer(enum) -> Type[Transformer]:
values = list(enum)
if len(values) < 2:
raise TypeError(f'enum.Enum requires at least two values.')
async def transform(cls, interaction: Interaction, value: Any) -> Any:
return enum[value]
ns = {
'type': classmethod(lambda _: AppCommandOptionType.string),
'transform': classmethod(transform),
'__discord_app_commands_transformer_enum__': enum,
'__discord_app_commands_transformer_choices__': [Choice(name=v.name, value=v.name) for v in values],
}
return type(f'{enum.__name__}ComplexEnumTransformer', (Transformer,), ns)
if TYPE_CHECKING:
from typing_extensions import Annotated as Transform
from typing_extensions import Annotated as Range
@ -611,7 +629,10 @@ def get_supported_annotation(
if issubclass(annotation, Transformer):
return (annotation, MISSING)
if issubclass(annotation, (Enum, InternalEnum)):
return (_make_enum_transformer(annotation), MISSING)
if all(isinstance(v.value, (str, int, float)) for v in annotation):
return (_make_enum_transformer(annotation), MISSING)
else:
return (_make_complex_enum_transformer(annotation), MISSING)
if annotation is Choice:
raise TypeError(f'Choice requires a type argument of int, str, or float')