Как считать время (кулдаун) для команды?

Question

[Mem1388]

Хочу сделать откат для команды, который будет хранится в общей базе данных, и при перезапуске бота продолжал считать время. Вот что написал я, но цикл для отсчёта не работает, в базу записывается время и более не обновляется.

conn = sqlite3.connect('testingBot.db')
cur = conn.cursor()

@Bot.event
async def on_ready():
    cur.execute(f"""CREATE TABLE IF NOT EXISTS users (
        id BIGINT,
        rep INT,
        rep_last_user BIGINT,
        rep_time BIGINT
    )""")
    for guild in Bot.guilds:
        for member in guild.members:
            if cur.execute(f"SELECT id FROM users WHERE id = {member.id}").fetchone() is None:
                cur.execute(f"INSERT INTO users VALUES (?, ?, ?, ?)", (member.id, 0, 0, 0))
            else:
                pass
    conn.commit()
    print(f"Bot logged as -> {Bot.user}")

# Цикл для отсчёта времени, который написан некорректно.
for guild in Bot.guilds:
    for member in guild.members:
        while cur.execute(f"SELECT rep_time FROM users WHERE id = {member.id}").fetchone()[0] > 0:
            cur.execute(f"UPDATE users SET rep_time = rep_time - 5 WHERE id = {member.id}")
            conn.commit()

@Bot.command()
async def rep(ctx, member: Optional[discord.Member]):
    if cur.execute(f"SELECT rep_time FROM users WHERE id = {ctx.message.author.id}").fetchone()[0] == 0:
        if member is None:
            return
        elif member.id == ctx.message.author.id:
            emb = discord.Embed(
                description = f'{ctx.message.author.mention} Вы не можете повысить репутацию сами себе.',
                color = 0xff0000
            )
            await ctx.send(embed = emb)
        else:
            cur.execute(f"UPDATE users SET rep = rep + 1 WHERE id = {member.id}")
            conn.commit()
            emb = discord.Embed(
                description = f'{ctx.message.author.mention} Вы повысили репутацию пользователя {member.mention} \nТеперь у него **+{cur.execute(f"SELECT rep FROM users WHERE id = {member.id}").fetchone()[0]}☆** репутации. \n\nПовысить снова можно через **2 часа.**',
                color = 0x00E584
            )
            await ctx.send(embed = emb)
            cur.execute(f"UPDATE users SET rep_time = 10800 WHERE id = {ctx.message.author.id}")
            conn.commit()
    elif cur.execute(f"SELECT rep_time FROM users WHERE id = {member.id}").fetchone()[0] > 0:
        rep_cooldown = strftime('%Hh %Mm %Ss', gmtime(cur.execute(f"SELECT rep_time FROM users WHERE id = {ctx.message.author.id}").fetchone()[0]))
        emb = discord.Embed(
            description = f'{ctx.message.author.mention} Вы уже использовали очки репутации. Снова повысить репутацию вы сможете через **{rep_cooldown}.**',
            color = 0xff0000
        )
        await ctx.send(embed = emb)

Answer 1

[Vindicar]

Неправильная архитектура.
Тебе незачем отсчитывать время постоянно - просто запомни, когда команда была отправлена данным пользователем, и при следующем запуске этой команды проверь запомненное предыдущее время отправки и сверь его с текущим. Если разница больше кулдауна, можно выполнять команду снова.

Comments

[Mem1388]

Тогда получается, что время будет идти и после того, как бот завершит работу.

[Vindicar]

Mem1388, да. Это плохо?

[Mem1388]

Vindicar, В данный момент интересует именно названый мною вариант.
В какой архитектуре нужно расположить циклы, чтобы при включении бот начинал отсчитывать время у пользователей?

[Vindicar]

Не используйте цикл вообще.
Вы используете discord.py? В документации есть пример периодического действия на фоне.

Скопирую сюда:

from discord.ext import tasks

    @tasks.loop(seconds=60) # асинхронная подпрограмма будет вызываться каждые 60 секунд
    # она НЕ ДОЛЖНА выполнять длительных действий, так как на это время бот блокируется!
    async def my_background_task(self):
        #в этом пример бо тпросто пишет в канал увеличивающийся счётчик
        channel = self.get_channel(1234567) # channel ID goes here
        self.counter += 1
        await channel.send(self.counter)

    @my_background_task.before_loop # а так можно выполнить подпрограмму ДО начала циклической задачи
    #обрати внимание на имя декоратора: my_background_task, как у функции выше
    async def before_my_task(self):
        await self.wait_until_ready() # начнём счёт, когда бот залогинится.

[Mem1388]

Vindicar,

она НЕ ДОЛЖНА выполнять длительных действий, так как на это время бот блокируется!

Т.е. каждый раз когда когда бот будет обновлять данные - он не будет выполнять других команд?

[Vindicar]

Mem1388, это завязано на принципы работы асинхронных приложений.
Если на пальцах:
1. Пока выполняемая команда (корутина) ждёт результата асинхронной операции ввода-вывода (например, пока отправится сообщение в чат), другие корутины могут выполняться. Иными словами, пока мы в await чтототам(), бот может заниматься другими делами.
2. Пока корутина выполняет синхронную операцию (т.е. что угодно кроме вышеупомянутого), другие корутины выполняться не могут. До них дойдёт очередь, когда бот закончит с текущей корутиной.

Встроенная sqlite3 - синхронная, поэтому она будет блокировать корутину, пока запрос не выполнится.
Отсюда вывод: если тебе нужен длительный запрос к БД, ты должен или а) использовать асинхронную библиотеку для работы (например, aiosqlite)
или б) разбивать большой запрос на меньшие части (например, по 1000 пользователей за раз), и вставлять между ними вызов await asyncio.sleep() с минимальной задержкой. Тогда бот будет время от времени "отвлекаться" на другие ожидающие задачи, если они есть.

[Mem1388]

Vindicar, А как мне без цикла сделать это для каждого пользователя из всех серверов?

[Vindicar]

Mem1388, используешь асинхронную библиотеку типа aiosqlite.
Я полагаю, у тебя получится код следующего вида (код условный, работоспособность не проверял!):

#подразумевается, что этот код - часть cog-класса для бота.
def __init__(self):
    #создаём подключение к БД сами...
    #возможно, его стоит передавать в параметрах конструктора нашего сog?
    self.db =  aiosqlite.connect('mydatabase.sqlite')

@tasks.loop(seconds=60) #выполняем эту функцию каждые 60 секунд
async def countdown_processor(self):
    #тупо уменьшаем значение счётчика для всех пользователей
    #но (для примера) с ограничением до нуля
    cursor = await self.db.execute("UPDATE user_counters SET counter = counter - 1 WHERE counter > 0")
    await cursor.close() #так как это не select-запрос, нам курсор без надобности.
    #всё, больше ничего не требуется! 
    #так как вызов асинхронный, бот может заниматься другими делами, пока запрос выполняется

Возможно, имеет смысл вытащить счётчики в отдельную таблицу user_counters, со ссылкой по FOREIGN KEY на основную таблицу пользователей. Но тогда понадобится отдельно заводить счётчик при вставке в таблицу пользователей новой строки.

В принципе в user_counters могут быть группы строк - по одной на пару "пользователь-команда". При необходимости проверить кулдаун - проверяешь, что пара "пользователь-команда" в этой таблице либо отсутствует, либо её счётчик равен 0. При необходимости задать кулдаун - вставляешь/обновляешь строку для пары "пользователь-команда" с желаемым значением счётчика.

Но вообще я бы очень советовал использовать мой первый совет. Гораздо проще и надёжнее.