Фоновая петля с реакциями Discord.Py - PullRequest
Новая
       9

Фоновая петля с реакциями Discord.Py

0 голосов
/ 12 сентября 2018

У меня есть фоновая петля, которая будет выплевывать эмодзи каждые X минут с прикрепленной к нему реакцией.Я хочу, чтобы, когда кто-то нажимал на реакцию сообщения, он удалял сообщение, а затем отправлял другое сообщение, гласящее: «автор сообщения захватил добычу», а затем добавлял сумму в файл json cash.

Прямо сейчасмой код заставляет работать фоновый цикл, но я не уверен, как получить message.author.id в отношении фонового цикла, чтобы я мог ссылаться на него в on_reaction_add.Текущий код заставляет бота реагировать один раз, когда он выплевывает фоновый цикл, а затем снова в on_reaction_add.Я пытаюсь заставить его ждать, пока пользователь отреагирует на сообщение фонового цикла тем же смайликом, а не ботом. 

client = discord.Client()
emoji_msg_grab = {}


try:
    with open("cash.json") as fp:
        cash = json.load(fp)
except Exception:
    cash = {}

def save_cash():
    with open("cash.json", "w+") as fp:
        json.dump(cash, fp, sort_keys=True, indent=4)

def add_dollars(user: discord.User, dollars: int):
    id = user.id
    if id not in cash:
        cash[id] = {}
    cash[id]["dollars"] = cash[id].get("dollars", 0) + dollars
    print("{} now has {} dollars".format(user.name, cash[id]["dollars"]))
    save_cash()


async def background_loop():
    await client.wait_until_ready()
    while not client.is_closed:
        channel = client.get_channel("479919577279758340")
        emojigrab = '?'
        emojimsgid = await client.send_message(channel, emojigrab)
        await client.add_reaction(emojimsgid, "?")
        user_id = emojimsgid.author.id
        emoji_msg_grab[user_id] = {"emoji_msg_id": emojimsgid.id, "emoji_user_id": user_id}
        await asyncio.sleep(600)

@client.event
async def on_reaction_add(reaction, user):
    msgid = reaction.message.id
    chat = reaction.message.channel

    if reaction.emoji == "?" and msgid == emoji_msg_grab[user.id]["emoji_msg_id"] and user.id == emoji_msg_grab[user.id]["emoji_user_id"]:
        emoji_msg_grab[user.id]["emoji_msg_id"] = None
        await client.send_message(chat, "{} has grabbed the loot!".format(user.mention))
        await client.delete_message(reaction.message)
        add_dollars(user, 250)

client.loop.create_task(background_loop())

1 Ответ

0 голосов
/ 12 сентября 2018

Я бы использовал Client.wait_for_reaction вместо on_reaction_add: 

async def background_loop():
    await client.wait_until_ready()
    channel = client.get_channel("479919577279758340")
    while not client.is_closed:
        emojigrab = '?'
        emojimsg = await client.send_message(channel, emojigrab)
        await client.add_reaction(emojimsg, "?")
        res = await client.wait_for_reaction(emoji="?", message=emojimsg, timeout=600, 
                                             check=lambda reaction, user: user != client.user)
        if res:  # not None
            await client.delete_message(emojimsg)
            await client.send_message(channel, "{} has grabbed the loot!".format(res.user.mention))
            await asyncio.sleep(1)
            add_dollars(res.user, 250)
...