Add support for default values in non string selects

Author: doluk

discord/components.py

@@ -39,13 +39,16 @@
     from .types.components import InputText as InputTextComponentPayload
     from .types.components import SelectMenu as SelectMenuPayload
     from .types.components import SelectOption as SelectOptionPayload
+    from .types.components import SelectDefaultValue as SelectDefaultValuePayload
+    from .types.components import SelectMenuDefaultValueType
 
 __all__ = (
     "Component",
     "ActionRow",
     "Button",
     "SelectMenu",
     "SelectOption",
+    "SelectDefaultValue",
     "InputText",
 )
 
@@ -328,6 +331,9 @@ class SelectMenu(Component):
         except for :attr:`ComponentType.channel_select`.
     disabled: :class:`bool`
         Whether the select is disabled or not.
+    default_values: List[:class:`SelectDefaultValue`]
+        A list of options that are selected by default for select menus of type user, role, channel and mentionable.
+        Will be an empty list for component type :attr:`ComponentType.string_select`.
     """
 
     __slots__: tuple[str, ...] = (
@@ -338,6 +344,7 @@ class SelectMenu(Component):
         "options",
         "channel_types",
         "disabled",
+        "default_values",
     )
 
     __repr_info__: ClassVar[tuple[str, ...]] = __slots__
@@ -355,6 +362,7 @@ def __init__(self, data: SelectMenuPayload):
         self.channel_types: list[ChannelType] = [
             try_enum(ChannelType, ct) for ct in data.get("channel_types", [])
         ]
+        self.default_values: list[SelectDefaultValue] = []
 
     def to_dict(self) -> SelectMenuPayload:
         payload: SelectMenuPayload = {
@@ -371,6 +379,8 @@ def to_dict(self) -> SelectMenuPayload:
             payload["channel_types"] = [ct.value for ct in self.channel_types]
         if self.placeholder:
             payload["placeholder"] = self.placeholder
+        if self.type is ComponentType.role_select and self.default_values:
+            payload["default_values"] = []
 
         return payload
 
@@ -494,6 +504,63 @@ def to_dict(self) -> SelectOptionPayload:
         return payload
 
 
+class SelectDefaultValue:
+    """Represents a :class:`discord.SelectMenu`'s default option for user, role, channel and mentionable select menu types.
+
+    These can be created by users.
+
+    .. versionadded:: 2.7
+
+    Attributes
+    ----------
+    id: :class:`int`
+        The snowflake id of the default option.
+    type: :class:`SelectMenuDefaultValueType`
+        The type of the default value. This is not displayed to users.
+    """
+
+    __slots__: tuple[str, ...] = (
+        "id",
+        "type",
+    )
+
+    def __init__(
+        self,
+        *,
+        id: int,
+        type: SelectMenuDefaultValueType
+    ) -> None:
+        self.id = id
+        self.type = type
+
+
+
+    def __repr__(self) -> str:
+        return (
+            "<SelectDefaultValue"
+            f" id={self.id!r} type={self.type!r}>"
+        )
+
+    def __str__(self) -> str:
+        return f"{self.id} {self.type}"
+
+    @classmethod
+    def from_dict(cls, data: SelectDefaultValuePayload) -> SelectDefaultValue:
+
+        return cls(
+                id=data["id"],
+                type=data["type"],
+        )
+
+    def to_dict(self) -> SelectDefaultValuePayload:
+        payload: SelectDefaultValuePayload = {
+            "id": self.id,
+            "type": self.type,
+        }
+
+        return payload
+
+
 def _component_factory(data: ComponentPayload) -> Component:
     component_type = data["type"]
     if component_type == 1:

discord/enums.py

@@ -78,6 +78,7 @@
     "InteractionContextType",
     "PollLayoutType",
     "MessageReferenceType",
+    "SelectMenuDefaultValueType",
 )
 
 
@@ -1078,6 +1079,17 @@ class SubscriptionStatus(Enum):
     inactive = 2
 
 
+class SelectMenuDefaultValueType(Enum):
+    """The type of a select menu's default value."""
+
+    Channel = 'channel'
+    Role = 'role'
+    User = 'user'
+
+    def __str__(self):
+        return self.name
+
+
 T = TypeVar("T")
 
 

discord/types/components.py

@@ -36,6 +36,7 @@
 ComponentType = Literal[1, 2, 3, 4]
 ButtonStyle = Literal[1, 2, 3, 4, 5, 6]
 InputTextStyle = Literal[1, 2]
+SelectMenuDefaultValueType = Literal["channel", "role", "user"]
 
 
 class ActionRow(TypedDict):
@@ -74,6 +75,11 @@ class SelectOption(TypedDict):
     default: bool
 
 
+class SelectDefaultValue(TypedDict):
+    id: Snowflake
+    type: SelectMenuDefaultValueType
+
+
 class SelectMenu(TypedDict):
     placeholder: NotRequired[str]
     min_values: NotRequired[int]
@@ -83,6 +89,7 @@ class SelectMenu(TypedDict):
     options: NotRequired[list[SelectOption]]
     type: Literal[3, 5, 6, 7, 8]
     custom_id: str
+    default_values: NotRequired[list[SelectDefaultValue]]
 
 
 Component = Union[ActionRow, ButtonComponent, SelectMenu, InputText]