Skip to content

Commit 4c6015c

Browse files
committed
feat: a couple of create changes
1 parent 21f31fa commit 4c6015c

File tree

1 file changed

+14
-2
lines changed

1 file changed

+14
-2
lines changed

interactions/models/discord/poll.py

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,12 @@ class Poll(DictSerializationMixin):
9595

9696
@classmethod
9797
def create(
98-
cls, question: str, duration: int, *, allow_multiselect: bool = False, answers: Optional[list[PollMedia]] = None
98+
cls,
99+
question: str,
100+
*,
101+
duration: int,
102+
allow_multiselect: bool = False,
103+
answers: Optional[list[PollMedia | str]] = None,
99104
) -> Self:
100105
"""
101106
Create a Poll object for sending.
@@ -111,7 +116,14 @@ def create(
111116
112117
"""
113118
if answers:
114-
media_to_answers = [PollAnswer(poll_media=answer) for answer in answers]
119+
media_to_answers = [
120+
(
121+
PollAnswer(poll_media=answer)
122+
if isinstance(answer, PollMedia)
123+
else PollAnswer(poll_media=PollMedia.create(text=answer))
124+
)
125+
for answer in answers
126+
]
115127
else:
116128
media_to_answers = []
117129

0 commit comments

Comments
 (0)