Add set_channel_types function (#99)

brazier-dev · Feb 9, 2023 · 8cdcfe3 · 8cdcfe3
1 parent 85bd616
commit 8cdcfe3
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 8 deletions.
diff --git a/flare/components/base.py b/flare/components/base.py
@@ -158,17 +158,24 @@ async def from_partial(component: hikari.PartialComponent) -> CallbackComponent:
                     component_inst, TextSelect | RoleSelect | UserSelect | MentionableSelect | ChannelSelect
                 )
 
-            component_inst.set_max_values(component.max_values).set_placeholder(
-                component.placeholder or hikari.UNDEFINED
-            ).set_disabled(component.is_disabled)
+            (
+                component_inst.set_max_values(component.max_values)
+                .set_min_values(component.min_values)
+                .set_placeholder(component.placeholder or hikari.UNDEFINED)
+                .set_disabled(component.is_disabled)
+            )
 
             if isinstance(component, hikari.components.TextSelectMenuComponent):
                 if t.TYPE_CHECKING:
                     assert isinstance(component_inst, TextSelect)
 
-                component_inst.set_options(
-                    *((option.label, option.value) for option in component.options)
-                ).set_min_values(component.min_values)
+                component_inst.set_options(*((option.label, option.value) for option in component.options))
+
+            if isinstance(component, hikari.components.ChannelSelectMenuComponent):
+                if t.TYPE_CHECKING:
+                    assert isinstance(component_inst, ChannelSelect)
+
+                component_inst.set_channel_types(*(hikari.ChannelType(c) for c in component.channel_types))
 
         return component_inst
 

diff --git a/flare/components/select.py b/flare/components/select.py
@@ -311,8 +311,8 @@ def __post_init__(self):
         super().__post_init__()
         self.channel_types = self.__channel_types
 
-    def set_options(self, *options: tuple[str, str] | str | hikari.SelectMenuOption) -> Self:
-        self.options = options
+    def set_channel_types(self, *channel_types: hikari.ChannelType) -> Self:
+        self.channel_types = channel_types
         return self
 
     def build(