From b229a2c7e26af0289c628081797fc450dd4de027 Mon Sep 17 00:00:00 2001 From: mihalin Date: Fri, 24 Jun 2022 23:18:23 +0300 Subject: [PATCH] promo minor changes --- olgram/commands/info.py | 6 +++++- olgram/commands/promo.py | 6 ++++++ 2 files changed, 11 insertions(+), 1 deletion(-) diff --git a/olgram/commands/info.py b/olgram/commands/info.py index c6513f0..fdce6ee 100644 --- a/olgram/commands/info.py +++ b/olgram/commands/info.py @@ -25,6 +25,8 @@ async def info(message: types.Message, state: FSMContext): bots_count = len(bots) user_count = len(await models.User.all()) templates_count = len(await models.DefaultAnswer.all()) + promo_count = len(await models.Promo.all()) + olgram_text_disabled = len(await models.Bot.filter(enable_olgram_text=False)) income_messages = sum([bot.incoming_messages_count for bot in bots]) outgoing_messages = sum([bot.outgoing_messages_count for bot in bots]) @@ -33,4 +35,6 @@ async def info(message: types.Message, state: FSMContext): _("Количество пользователей (у конструктора): {0}\n").format(user_count) + _("Шаблонов ответов: {0}\n").format(templates_count) + _("Входящих сообщений у всех ботов: {0}\n").format(income_messages) + - _("Исходящих сообщений у всех ботов: {0}\n").format(outgoing_messages)) + _("Исходящих сообщений у всех ботов: {0}\n").format(outgoing_messages) + + _("Промо-кодов выдано: {0}\n").format(promo_count) + + _("Рекламную плашку выключили: {0}\n".format(olgram_text_disabled))) diff --git a/olgram/commands/promo.py b/olgram/commands/promo.py index 1e6dd6a..486bdb8 100644 --- a/olgram/commands/promo.py +++ b/olgram/commands/promo.py @@ -48,6 +48,12 @@ async def del_promo(message: types.Message, state: FSMContext): if not promo: return await message.answer(_("Такого кода не существует")) + user = await models.User.filter(promo=promo) + bots = await user.bots() + for bot in bots: + bot.enable_olgram_text = True + await bot.save(update_fields=["enable_olgram_text"]) + await promo.delete() await message.answer(_("Промокод отозван"))