implement select default values

Author: DA-344

discord/components.py

@@ -34,6 +34,7 @@
     ChannelType,
     ComponentType,
     InputTextStyle,
+    SelectDefaultValueType,
     SeparatorSpacingSize,
     try_enum,
 )
@@ -44,7 +45,6 @@
 if TYPE_CHECKING:
     from .emoji import AppEmoji, GuildEmoji
     from .types.components import ActionRow as ActionRowPayload
-    from .types.components import BaseComponent as BaseComponentPayload
     from .types.components import ButtonComponent as ButtonComponentPayload
     from .types.components import Component as ComponentPayload
     from .types.components import ContainerComponent as ContainerComponentPayload
@@ -60,6 +60,7 @@
     from .types.components import TextDisplayComponent as TextDisplayComponentPayload
     from .types.components import ThumbnailComponent as ThumbnailComponentPayload
     from .types.components import UnfurledMediaItem as UnfurledMediaItemPayload
+    from .types.components import SelectDefaultValue as SelectDefaultValuePayload
 
 __all__ = (
     "Component",
@@ -437,6 +438,7 @@ class SelectMenu(Component):
         "channel_types",
         "disabled",
         "required",
+        "default_values",
     )
 
     __repr_info__: ClassVar[tuple[str, ...]] = __slots__
@@ -457,6 +459,7 @@ def __init__(self, data: SelectMenuPayload):
             try_enum(ChannelType, ct) for ct in data.get("channel_types", [])
         ]
         self.required: bool | None = data.get("required")
+        self.default_values: list[SelectDefaultValue] = SelectDefaultValue._from_data(data.get("default_values"))
 
     def to_dict(self) -> SelectMenuPayload:
         payload: SelectMenuPayload = {
@@ -476,10 +479,50 @@ def to_dict(self) -> SelectMenuPayload:
             payload["placeholder"] = self.placeholder
         if self.required is not None:
             payload["required"] = self.required
+        if self.type is not ComponentType.string_select:
+            payload["default_values"] = [dv.to_dict() for dv in self.default_values]
 
         return payload
 
 
+class SelectDefaultValue:
+    r"""Represents a :class:`discord.SelectMenu`\s default value.
+
+    This is only applicable to selects of type other than :attr:`ComponentType.string_select`.
+
+    .. versionadded:: 2.7
+
+    Parameters
+    ----------
+    id: :class:`int`
+        The ID of the default value.
+    type: :class:`SelectDefaultValueType`
+        The default value type.
+    """
+
+    __slots__ = ("id", "type")
+
+    def __init__(
+        self,
+        id: int,
+        type: SelectDefaultValueType,
+    ) -> None:
+        self.id: int = id
+        self.type: SelectDefaultValueType = type
+
+    @classmethod
+    def _from_data(cls, default_values: list[SelectDefaultValuePayload] | None) -> list[SelectDefaultValue]:
+        if not default_values:
+            return []
+        return [cls(id=int(d['id']), type=try_enum(SelectDefaultValueType, d['type'])) for d in default_values]
+
+    def to_dict(self) -> SelectDefaultValuePayload:
+        return {
+            "id": self.id,
+            "type": self.type.value,
+        }
+
+
 class SelectOption:
     """Represents a :class:`discord.SelectMenu`'s option.
 

discord/enums.py

@@ -83,6 +83,7 @@
     "ThreadArchiveDuration",
     "SubscriptionStatus",
     "SeparatorSpacingSize",
+    "SelectDefaultValueType",
 )
 
 
@@ -1120,6 +1121,14 @@ def __int__(self):
         return self.value
 
 
+class SelectDefaultValueType(Enum):
+    """Represents the default value type of a select menu."""
+
+    channel = "channel"
+    role = "role"
+    user = "user"
+
+
 T = TypeVar("T")
 
 

discord/object.py

@@ -28,6 +28,7 @@
 from typing import TYPE_CHECKING, SupportsInt, Union
 
 from . import utils
+from .abc import Snowflake
 from .mixins import Hashable
 
 if TYPE_CHECKING:
@@ -70,9 +71,11 @@ class Object(Hashable):
     ----------
     id: :class:`int`
         The ID of the object.
+    type: type[:class:`abc.Snowflake`]
+        The model this object's ID is based off.
     """
 
-    def __init__(self, id: SupportsIntCast):
+    def __init__(self, id: SupportsIntCast, type: type[Snowflake] = utils.MISSING):
         try:
             id = int(id)
         except ValueError:
@@ -81,6 +84,7 @@ def __init__(self, id: SupportsIntCast):
             ) from None
         else:
             self.id = id
+        self.type: type[Snowflake] = type or self.__class__
 
     def __repr__(self) -> str:
         return f"<Object id={self.id!r}>"

discord/types/components.py

@@ -37,6 +37,7 @@
 ButtonStyle = Literal[1, 2, 3, 4, 5, 6]
 InputTextStyle = Literal[1, 2]
 SeparatorSpacingSize = Literal[1, 2]
+SelectDefaultValueType = Literal['channel', 'role', 'user']
 
 
 class BaseComponent(TypedDict):
@@ -46,7 +47,7 @@ class BaseComponent(TypedDict):
 
 class ActionRow(BaseComponent):
     type: Literal[1]
-    components: list[ButtonComponent, InputText, SelectMenu]
+    components: list[ButtonComponent | InputText | SelectMenu]
 
 
 class ButtonComponent(BaseComponent):
@@ -80,6 +81,11 @@ class SelectOption(TypedDict):
     default: bool
 
 
+class SelectDefaultValue(TypedDict):
+    id: Snowflake
+    type: SelectDefaultValueType
+
+
 class SelectMenu(BaseComponent):
     placeholder: NotRequired[str]
     min_values: NotRequired[int]
@@ -90,6 +96,7 @@ class SelectMenu(BaseComponent):
     type: Literal[3, 5, 6, 7, 8]
     custom_id: str
     required: NotRequired[bool]
+    default_values: NotRequired[list[SelectDefaultValue]]
 
 
 class TextDisplayComponent(BaseComponent):
@@ -100,7 +107,7 @@ class TextDisplayComponent(BaseComponent):
 class SectionComponent(BaseComponent):
     type: Literal[9]
     components: list[TextDisplayComponent]
-    accessory: NotRequired[ThumbnailComponent, ButtonComponent]
+    accessory: NotRequired[ThumbnailComponent | ButtonComponent]
 
 
 class UnfurledMediaItem(TypedDict):

discord/ui/button.py

@@ -294,7 +294,7 @@ def button(
     emoji: str | GuildEmoji | AppEmoji | PartialEmoji | None = None,
     row: int | None = None,
     id: int | None = None,
-) -> Callable[[ItemCallbackType], ItemCallbackType]:
+) -> Callable[[ItemCallbackType[Button[V]]], Button[V]]:
     """A decorator that attaches a button to a component.
 
     The function being decorated should have three parameters, ``self`` representing
@@ -347,4 +347,4 @@ def decorator(func: ItemCallbackType) -> ItemCallbackType:
         }
         return func
 
-    return decorator
+    return decorator  # type: ignore # lie to the type checkers, because after a View is instated, the button callback is converted into a Button instance