Как сократить код?

Question

[bjfssd757]

Можно ли сократить количество условий в один/несколько циклов? В таком виде код получается довольно громоздким и каким-то неудобным для чтения.
И вообще стоит ли делать такую проверку в функции on_message или будет лучше написать отдельную функцию для этого?

@client.event
async def on_message(message: discord.Message, *args):
    await client.process_commands(message)

    if message.author.bot == True:
        pass
    else:
        if len(args) >= 10:
            cursor.execute(f"UPDATE users SET xp = xp + 100 WHERE id = {message.author.id}")
            connection.commit()

            exp = cursor.execute(f"SELECT xp FROM users WHERE id = {message.author.id}").fetchone()
            lvl = cursor.execute(f"SELECT lvl FROM users WHERE id = {message.author.id}").fetchone()

            if lvl < 10:
                if exp % 10000 == 0:
                    cursor.execute(f"UPDATE users SET lvl = lvl + 1 WHERE id = {message.author.id}")
                    connection.commit()
                else:
                    pass
            elif lvl < 20 and lvl >= 10:
                if exp % 20000 == 0:
                    cursor.execute(f"UPDATE users SET lvl = lvl + 1 WHERE id = {message.author.id}")
                    connection.commit()
                else:
                    pass
            elif lvl < 100 and lvl >= 20:
                if exp % 40000 == 0:
                    cursor.execute(f"UPDATE users SET lvl = lvl + 1 WHERE id = {message.author.id}")
                    connection.commit()
                else:
                    pass
            elif lvl < 300 and lvl >= 100:
                if exp % 80000 == 0:
                    cursor.execute(f"UPDATE users SET lvl = lvl + 1 WHERE id = {message.author.id}")
                    connection.commit()
                else:
                    pass
            elif lvl < 700 and lvl >= 300:
                if exp % 160000 == 0:
                    cursor.execute(f"UPDATE users SET lvl = lvl + 1 WHERE id = {message.author.id}")
                    connection.commit()
                else:
                    pass
            elif lvl < 1000 and lvl >= 700:
                if exp % 320000 == 0:
                    cursor.execute(f"UPDATE users SET lvl = lvl + 1 WHERE id = {message.author.id}")
                    connection.commit()
                else:
                    pass

Answer 1

[Wispik]

Можно, как вариант, все условия вынести в отдельный список и по нему проходиться в цикле (в таком случае новые условия будет проще добавлять):

conditions = [
    {
        'min_lvl': 0,
        'max_lvl': 10,
        'exp': 10000
    },
    {
        'min_lvl': 10,
        'max_lvl': 20,
        'exp': 20000
    },
]

for _cond in conditions:
    if lvl < _cond['max_lvl'] and lvl >= _cond['min_lvl'] and exp % _cond['exp'] == 0:
        cursor.execute(f"UPDATE users SET lvl = lvl + 1 WHERE id = {message.author.id}")
        connection.commit()
        break

Comments

[bjfssd757]

Спасибо за ответ!
Воспользуюсь вашим вариантом)

Answer 2

[Zailox]

Можно все это написать в одну строку, но это будет ещё более неудобным для чтения.

Пример:

Вместо

if a == 5:
    do_smth()
else:
    if a > 10:
        func()
    else:
       func2()

Можно

do_smth() if a == 5 else func() if a > 10 else func2()

Не очень понятно, про какие вы говорили циклы, для чего они?

Насчёт "нужно ли выводить в функцию" - если вы используете этот код только здесь - нет, если в нескольких местах - да, желательно.

Comments

[bjfssd757]

Спасибо за ответ!
Насчёт циклов: я спрашивал про возможность изменить все эти if'ы на циклы (или не все...:). Но у меня даже предположения нет по реализации через циклы)

Answer 3

[fenrir]

Такое количество запросов к БД с коммитами это кошмар.

on_message это событие которые вызывается на каждое отправленное сообщение и в нём не рекомендуется использовать никаких запросов к БД.

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

Comments

[bjfssd757]

Спасибо за замечание, учту)
В данный момент это сильной проблемы не создаст, но лучше не буду дожидаться появления проблем)

[Zailox]

bjfssd757, проблемы возникнут, если очень много человек вместе будут пользоваться ботом, насчёт решения сверху - можно сделать так:

from threading import Thread

def db():
    #что то делаем

t = Thread(target=db)
t.start()

Этот код (из функции db) будет выполняться на фоне

[bjfssd757]

Zailox, я не очень знаком с потоками, поэтому мой вопрос может быть довольно глупым, но всё же) эту функцию (db) надо будет запускать в on_message или как? Я просто не очень понимаю, как ещё она будет выполняться при выполнение условий)

[Zailox]

bjfssd757, вне on_message, он по идее (поток) должен работать ещё до бота, чтобы читать словарь, т.е.

def db():
    *Читаем словарь и сохраняем в переменную, файл или куда там надо*
    time.sleep(60) #спим 1 минуту

А потом, когда данные из базы будут нужны, они уже будут в переменной/файле, запросы нужно будет делать только при записи.

Кулдаун можно поставить меньше, хоть 5 секунд.