add select overloads

Author: NeloBlivion

discord/ui/action_row.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
 
 from functools import partial
+from collections.abc import Sequence
 from typing import TYPE_CHECKING, ClassVar, Iterator, TypeVar
 
 from ..components import ActionRow as ActionRowComponent
-from ..components import SelectOption, _component_factory
+from ..components import SelectOption, SelectDefaultValue, _component_factory
 from ..enums import ButtonStyle, ChannelType, ComponentType
 from ..utils import find, get
 from .button import Button
@@ -210,6 +211,53 @@ def add_button(
 
         return self.add_item(button)
 
+    @overload
+    def add_select(
+        self,
+        select_type: Literal[ComponentType.string_select] = ...,
+        *,
+        custom_id: str | None = ...,
+        placeholder: str | None = ...,
+        min_values: int = ...,
+        max_values: int = ...,
+        options: list[SelectOption] | None = ...,
+        disabled: bool = ...,
+        id: int | None = ...,
+    ) -> None: ...
+
+    @overload
+    def add_select(
+        self,
+        select_type: Literal[ComponentType.channel_select] = ...,
+        *,
+        custom_id: str | None = ...,
+        placeholder: str | None = ...,
+        min_values: int = ...,
+        max_values: int = ...,
+        channel_types: list[ChannelType] | None = ...,
+        disabled: bool = ...,
+        id: int | None = ...,
+        default_values: Sequence[SelectDefaultValue] | None = ...,
+    ) -> None: ...
+
+    @overload
+    def add_select(
+        self,
+        select_type: Literal[
+            ComponentType.user_select,
+            ComponentType.role_select,
+            ComponentType.mentionable_select,
+        ] = ...,
+        *,
+        custom_id: str | None = ...,
+        placeholder: str | None = ...,
+        min_values: int = ...,
+        max_values: int = ...,
+        disabled: bool = ...,
+        id: int | None = ...,
+        default_values: Sequence[SelectDefaultValue] | None = ...,
+    ) -> None: ...
+
     def add_select(
         self,
         select_type: ComponentType = ComponentType.string_select,
@@ -222,6 +270,7 @@ def add_select(
         channel_types: list[ChannelType] | None = None,
         disabled: bool = False,
         id: int | None = None,
+        default_values: Sequence[SelectDefaultValue] | None = None,
     ) -> Self:
         """Adds a :class:`Select` to the container.
 
@@ -256,6 +305,11 @@ def add_select(
             Whether the select is disabled or not. Defaults to ``False``.
         id: Optional[:class:`int`]
             The select menu's ID.
+        default_values: Optional[Sequence[Union[:class:`discord.SelectDefaultValue`, :class:`discord.abc.Snowflake`]]]
+            The default values of this select. Only applicable if :attr:`.select_type` is not :attr:`discord.ComponentType.string_select`.
+
+            These can be either :class:`discord.SelectDefaultValue` instances or models, which will be converted into :class:`discord.SelectDefaultValue`
+            instances.
         """
 
         select = Select(
@@ -268,6 +322,7 @@ def add_select(
             channel_types=channel_types or [],
             disabled=disabled,
             id=id,
+            default_values=default_values
         )
 
         return self.add_item(select)

discord/ui/label.py

@@ -1,9 +1,10 @@
 from __future__ import annotations
 
 from typing import TYPE_CHECKING, Iterator, TypeVar
+from collections.abc import Sequence
 
 from ..components import Label as LabelComponent
-from ..components import SelectOption, _component_factory
+from ..components import SelectOption, SelectDefaultValue, _component_factory
 from ..enums import ButtonStyle, ChannelType, ComponentType, InputTextStyle
 from ..utils import find, get
 from .button import Button
@@ -184,6 +185,53 @@ def set_input_text(
 
         return self.set_item(text)
 
+    @overload
+    def set_select(
+        self,
+        select_type: Literal[ComponentType.string_select] = ...,
+        *,
+        custom_id: str | None = ...,
+        placeholder: str | None = ...,
+        min_values: int = ...,
+        max_values: int = ...,
+        options: list[SelectOption] | None = ...,
+        required: bool = ...,
+        id: int | None = ...,
+    ) -> None: ...
+
+    @overload
+    def set_select(
+        self,
+        select_type: Literal[ComponentType.channel_select] = ...,
+        *,
+        custom_id: str | None = ...,
+        placeholder: str | None = ...,
+        min_values: int = ...,
+        max_values: int = ...,
+        channel_types: list[ChannelType] | None = ...,
+        required: bool = ...,
+        id: int | None = ...,
+        default_values: Sequence[SelectDefaultValue] | None = ...,
+    ) -> None: ...
+
+    @overload
+    def set_select(
+        self,
+        select_type: Literal[
+            ComponentType.user_select,
+            ComponentType.role_select,
+            ComponentType.mentionable_select,
+        ] = ...,
+        *,
+        custom_id: str | None = ...,
+        placeholder: str | None = ...,
+        min_values: int = ...,
+        max_values: int = ...,
+        required: bool = ...,
+        id: int | None = ...,
+        default_values: Sequence[SelectDefaultValue] | None = ...,
+    ) -> None: ...
+
     def set_select(
         self,
         select_type: ComponentType = ComponentType.string_select,
@@ -196,6 +244,7 @@ def set_select(
         channel_types: list[ChannelType] | None = None,
         required: bool = True,
         id: int | None = None,
+        default_values: Sequence[SelectDefaultValue] | None = ...,
     ) -> Self:
         """Set this label's item to a select menu.
 
@@ -227,6 +276,11 @@ def set_select(
             Whether the select is required or not. Defaults to ``True``.
         id: Optional[:class:`int`]
             The select menu's ID.
+        default_values: Optional[Sequence[Union[:class:`discord.SelectDefaultValue`, :class:`discord.abc.Snowflake`]]]
+            The default values of this select. Only applicable if :attr:`.select_type` is not :attr:`discord.ComponentType.string_select`.
+
+            These can be either :class:`discord.SelectDefaultValue` instances or models, which will be converted into :class:`discord.SelectDefaultValue`
+            instances.
         """
 
         select = Select(
@@ -239,6 +293,7 @@ def set_select(
             channel_types=channel_types or [],
             required=required,
             id=id,
+            default_values=default_values
         )
 
         return self.set_item(select)

discord/ui/select.py

@@ -134,11 +134,11 @@ class Select(Generic[V, ST], Item[V]):
         rows. By default, items are arranged automatically into those 5 rows. If you'd
         like to control the relative positioning of the row then passing an index is advised.
         For example, row=1 will show up before row=2. Defaults to ``None``, which is automatic
-        ordering. The row number must be between 0 and 4 (i.e. zero indexed).
+        ordering. The row number must be between 0 and 4 (i.e. zero indexed). Does not work in :class:`ActionRow` or :class:`Label`.
     id: Optional[:class:`int`]
         The select menu's ID.
     required: Optional[:class:`bool`]
-        Whether the select is required or not. Only useable in modals. Defaults to ``True`` in modals.
+        Whether the select is required or not. Only useable when added to :class:`Label` for modals. Defaults to ``True`` in modals.
 
         .. versionadded:: 2.7
     default_values: Optional[Sequence[Union[:class:`discord.SelectDefaultValue`, :class:`discord.abc.Snowflake`]]]