matthew/enhanced-discord.py
1
0
Fork 0
Code Issues 4 Pull Requests 21 Projects Releases Wiki Activity

Add proper literal support

Browse Source
This commit is contained in:
Gnome 2021-09-03 20:18:15 +01:00
parent 8a779ef595
commit 623fcc0ac0
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
discord/ext/commands/core.py
View File

@ -1231,14 +1231,19 @@ class Command(_BaseCommand, Generic[CogT, P, T]):
option["type"] = discord_type option["type"] = discord_type
break break
elif origin is Literal and len(origin.__args__) <= 25: # type: ignore elif origin is Literal:
option["choices"] = [{ literal_values = annotation.__args__
"name": literal_value, python_type = type(literal_values[0])
"value": literal_value if (all(type(value) == python_type for value in literal_values)
} for literal_value in origin.__args__] # type: ignore and python_type in application_option_type_lookup.keys()):
else:
option["type"] = 3 # STRING
option["type"] = application_option_type_lookup[python_type]
option["choices"] = [{
"name": literal_value,
"value": literal_value
} for literal_value in annotation.__args__]
option.setdefault("type", 3) # STRING
payload["options"].append(option) payload["options"].append(option)
# Now we have all options, make sure required is before optional. # Now we have all options, make sure required is before optional.