feat: add methods to interact with http methods

Author: AstreaTSS

interactions/models/discord/message.py

@@ -1,6 +1,7 @@
 import asyncio
 import base64
 import re
+from collections import namedtuple
 from dataclasses import dataclass
 from typing import (
     TYPE_CHECKING,
@@ -29,6 +30,7 @@
 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 interactions.models.misc.iterator import AsyncIterator
 
 from .base import DiscordObject
 from .enums import (
@@ -68,6 +70,32 @@
 
 channel_mention = re.compile(r"<#(?P<id>[0-9]{17,})>")
 
+class PollAnswerVotersIterator(AsyncIterator):
+    def __init__(self, message: "Message", answer_id: int, limit: int = 25, after: Snowflake_Type | None = None) -> None:
+        self.message: "Message" = message
+        self.answer_id = answer_id
+        self.after: Snowflake_Type | None = after
+        self._more: bool = True
+        super().__init__(limit)
+
+    async def fetch(self) -> list["models.User"]:
+        if not self.last:
+            self.last = namedtuple("temp", "id")
+            self.last.id = self.after
+
+        rcv = await self.message._client.http.get_answer_voters(
+            self.message._channel_id,
+            self.message.id,
+            self.answer_id,
+            limit=self.get_limit,
+            after=to_snowflake(self.last.id) if self.last.id else None,
+        )
+        if not rcv:
+            raise asyncio.QueueEmpty
+
+        users = [self.message._client.cache.place_user_data(user_data) for user_data in rcv["users"]]
+        return users
+
 
 @attrs.define(eq=False, order=False, hash=False, kw_only=True)
 class Attachment(DiscordObject):
@@ -361,6 +389,8 @@ class Message(BaseMessage):
     """Data showing the source of a crosspost, channel follow add, pin, or reply message"""
     flags: MessageFlags = attrs.field(repr=False, default=MessageFlags.NONE, converter=MessageFlags)
     """Message flags combined as a bitfield"""
+    poll: Optional[Poll] = attrs.field(repr=False, default=None, converter=optional_c(Poll.from_dict))
+    """A poll."""
     interaction: Optional["MessageInteraction"] = attrs.field(repr=False, default=None)
     """Sent if the message is a response to an Interaction"""
     components: Optional[List["models.ActionRow"]] = attrs.field(repr=False, default=None)
@@ -593,6 +623,18 @@ def proto_url(self) -> str:
         """A URL like `jump_url` that uses protocols."""
         return f"discord://-/channels/{self._guild_id or '@me'}/{self._channel_id}/{self.id}"
 
+    def answer_voters(self, answer_id: int, limit: int = 0, before: Snowflake_Type | None = None) -> PollAnswerVotersIterator:
+        """
+        An async iterator for getting the voters for an answer in the poll this message has.
+
+        Args:
+            answer_id: The answer to get voters for
+            after: Get messages after this user ID
+            limit: The max number of users to return (default 25, max 100)
+
+        """
+        return PollAnswerVotersIterator(self, answer_id, limit, before)
+
     async def edit(
         self,
         *,
@@ -848,6 +890,14 @@ async def publish(self) -> None:
         """
         await self._client.http.crosspost_message(self._channel_id, self.id)
 
+    async def end_poll(self) -> "Message":
+        """
+        Ends the poll in this message.
+        """
+        message_data = await self._client.http.end_poll(self._channel_id, self.id)
+        if message_data:
+            return self._client.cache.place_message_data(message_data)
+
 
 def process_allowed_mentions(allowed_mentions: Optional[Union[AllowedMentions, dict]]) -> Optional[dict]:
     """