feat: add polls to send functions

Author: AstreaTSS

interactions/client/mixins/send.py

@@ -10,6 +10,7 @@
     from interactions.models.discord.components import BaseComponent
     from interactions.models.discord.embed import Embed
     from interactions.models.discord.message import AllowedMentions, Message, MessageReference
+    from interactions.models.discord.poll import Poll
     from interactions.models.discord.sticker import Sticker
     from interactions.models.discord.snowflake import Snowflake_Type
 
@@ -49,6 +50,7 @@ async def send(
         delete_after: Optional[float] = None,
         nonce: Optional[str | int] = None,
         enforce_nonce: bool = False,
+        poll: "Optional[Poll | dict]" = None,
         **kwargs: Any,
     ) -> "Message":
         """
@@ -73,6 +75,7 @@ async def send(
             enforce_nonce: If enabled and nonce is present, it will be checked for uniqueness in the past few minutes. \
                 If another message was created by the same author with the same nonce, that message will be returned \
                 and no new message will be created.
+            poll: A poll.
 
         Returns:
             New message object that was sent.
@@ -115,6 +118,7 @@ async def send(
             flags=flags,
             nonce=nonce,
             enforce_nonce=enforce_nonce,
+            poll=poll,
             **kwargs,
         )
 

interactions/ext/hybrid_commands/context.py

@@ -23,6 +23,7 @@
     Attachment,
     process_message_payload,
     TYPE_MESSAGEABLE_CHANNEL,
+    Poll,
 )
 from interactions.models.discord.enums import ContextType
 from interactions.client.mixins.send import SendMixin
@@ -309,6 +310,7 @@ async def send(
         suppress_embeds: bool = False,
         silent: bool = False,
         flags: Optional[Union[int, "MessageFlags"]] = None,
+        poll: "Optional[Poll | dict]" = None,
         delete_after: Optional[float] = None,
         ephemeral: bool = False,
         **kwargs: Any,
@@ -330,6 +332,7 @@ async def send(
             suppress_embeds: Should embeds be suppressed on this send
             silent: Should this message be sent without triggering a notification.
             flags: Message flags to apply.
+            poll: A poll.
             delete_after: Delete message after this many seconds.
             ephemeral: Should this message be sent as ephemeral (hidden) - only works with interactions
 
@@ -358,6 +361,7 @@ async def send(
             file=file,
             tts=tts,
             flags=flags,
+            poll=poll,
             delete_after=delete_after,
             pass_self_into_delete=bool(self._slash_ctx),
             **kwargs,

interactions/models/discord/message.py

@@ -28,6 +28,7 @@
 from interactions.models.discord.embed import process_embeds
 from interactions.models.discord.emoji import process_emoji_req_format
 from interactions.models.discord.file import UPLOADABLE_TYPE
+from interactions.models.discord.poll import Poll
 
 from .base import DiscordObject
 from .enums import (
@@ -981,6 +982,7 @@ def process_message_payload(
     flags: Optional[Union[int, MessageFlags]] = None,
     nonce: Optional[str | int] = None,
     enforce_nonce: bool = False,
+    poll: Optional[Poll | dict] = None,
     **kwargs,
 ) -> dict:
     """
@@ -1000,6 +1002,7 @@ def process_message_payload(
         enforce_nonce: If enabled and nonce is present, it will be checked for uniqueness in the past few minutes. \
             If another message was created by the same author with the same nonce, that message will be returned \
             and no new message will be created.
+        poll: A poll.
 
     Returns:
         Dictionary
@@ -1017,6 +1020,9 @@ def process_message_payload(
     if attachments:
         attachments = [attachment.to_dict() for attachment in attachments]
 
+    if isinstance(poll, Poll):
+        poll = poll.to_dict()
+
     return dict_filter_none(
         {
             "content": content,
@@ -1030,6 +1036,7 @@ def process_message_payload(
             "flags": flags,
             "nonce": nonce,
             "enforce_nonce": enforce_nonce,
+            "poll": poll,
             **kwargs,
         }
     )

interactions/models/discord/webhooks.py

@@ -27,6 +27,7 @@
         Message,
         MessageReference,
     )
+    from interactions.models.discord.poll import Poll
     from interactions.models.discord.sticker import Sticker
 
 __all__ = ("WebhookTypes", "Webhook")
@@ -190,6 +191,7 @@ async def send(
         tts: bool = False,
         suppress_embeds: bool = False,
         flags: Optional[Union[int, "MessageFlags"]] = None,
+        poll: "Optional[Poll | dict]" = None,
         username: str | None = None,
         avatar_url: str | None = None,
         wait: bool = False,
@@ -212,6 +214,7 @@ async def send(
             tts: Should this message use Text To Speech.
             suppress_embeds: Should embeds be suppressed on this send
             flags: Message flags to apply.
+            poll: A poll.
             username: The username to use
             avatar_url: The url of an image to use as the avatar
             wait: Waits for confirmation of delivery. Set this to True if you intend to edit the message
@@ -241,6 +244,7 @@ async def send(
             reply_to=reply_to,
             tts=tts,
             flags=flags,
+            poll=poll,
             username=username,
             avatar_url=avatar_url,
             **kwargs,

interactions/models/internal/context.py

@@ -12,6 +12,7 @@
 from interactions.client.const import get_logger, MISSING
 from interactions.models.discord.components import BaseComponent
 from interactions.models.discord.file import UPLOADABLE_TYPE
+from interactions.models.discord.poll import Poll
 from interactions.models.discord.sticker import Sticker
 from interactions.models.discord.user import Member, User
 
@@ -540,6 +541,7 @@ async def send(
         suppress_embeds: bool = False,
         silent: bool = False,
         flags: typing.Optional[typing.Union[int, "MessageFlags"]] = None,
+        poll: "typing.Optional[Poll | dict]" = None,
         delete_after: typing.Optional[float] = None,
         ephemeral: bool = False,
         **kwargs: typing.Any,
@@ -561,6 +563,7 @@ async def send(
             suppress_embeds: Should embeds be suppressed on this send
             silent: Should this message be sent without triggering a notification.
             flags: Message flags to apply.
+            poll: A poll.
             delete_after: Delete message after this many seconds.
             ephemeral: Whether the response should be ephemeral
 
@@ -589,6 +592,7 @@ async def send(
             file=file,
             tts=tts,
             flags=flags,
+            poll=poll,
             delete_after=delete_after,
             pass_self_into_delete=True,
             **kwargs,