You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
33 lines
1.3 KiB
33 lines
1.3 KiB
4 years ago
|
import asyncio
|
||
|
|
from typing import List, Callable, Awaitable
|
||
|
|
|
||
|
|
from discord import Message, Embed
|
||
|
|
from discord.ext.commands import Context
|
||
|
|
|
||
|
|
|
||
|
|
async def make_tabbed_message(ctx: Context, message: Message, emote_ids: List[int], embeds: List[Embed], timeout=300):
|
||
|
|
async def callback(index, _ctx, _message):
|
||
|
|
await message.edit(embed=embeds[index])
|
||
|
|
|
||
|
|
await make_reaction_message(ctx, message, emote_ids, callback, timeout)
|
||
|
|
|
||
|
|
|
||
|
|
async def make_reaction_message(ctx: Context, message: Message, emote_ids: List[int],
|
||
|
|
callback: Callable[[int, Context, Message], Awaitable[None]], timeout = 300):
|
||
|
|
for emote_id in emote_ids:
|
||
|
|
await message.add_reaction(ctx.bot.get_emoji(emote_id))
|
||
|
|
|
||
|
|
def check(rxn, usr):
|
||
|
|
return usr == ctx.author and rxn.emoji.id in emote_ids and rxn.message.id == message.id
|
||
|
|
|
||
|
|
while True:
|
||
|
|
try:
|
||
|
|
reaction, user = await ctx.bot.wait_for('reaction_add', timeout=timeout, check=check)
|
||
|
|
emote_index = emote_ids.index(reaction.emoji.id)
|
||
|
|
await callback(emote_index, ctx, message)
|
||
|
|
await message.remove_reaction(reaction, user)
|
||
|
|
except asyncio.TimeoutError:
|
||
|
|
for emote_id in emote_ids:
|
||
|
|
await message.remove_reaction(ctx.bot.get_emoji(emote_id), ctx.bot.user)
|
||
|
|
break
|