Dependencies/olgram
1
0
Fork 1
mirror of https://github.com/civsocit/olgram.git synced 2023-07-22 01:29:12 +03:00
Code Issues Packages Projects Releases Wiki Activity

templates minor improvements

Browse Source
This commit is contained in:
mihalin 2022-02-12 01:28:10 +03:00
parent a3eb985d28
commit 2164ee6f2c
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
olgram/commands/menu.py
View File

@ -283,6 +283,8 @@ async def template_received(message: types.Message, state: FSMContext):
# Delete template # Delete template
number = int(message.text) number = int(message.text)
templates = await bot.answers templates = await bot.answers
if not templates:
await message.answer("У вас нет шаблонов, чтобы их удалять")
if number < 0 or number >= len(templates): if number < 0 or number >= len(templates):
await message.answer(f"Неправильное число. Чтобы удалить шаблон, введите число от 0 до {len(templates)}") await message.answer(f"Неправильное число. Чтобы удалить шаблон, введите число от 0 до {len(templates)}")
return return
@ -291,10 +293,14 @@ async def template_received(message: types.Message, state: FSMContext):
# Add template # Add template
total_templates = len(await bot.answers) total_templates = len(await bot.answers)
if total_templates > 30: if total_templates > 30:
await message.answer("У вашего бота уже слишком много щаблонов") await message.answer("У вашего бота уже слишком много шаблонов")
else: else:
template = DefaultAnswer(text=message.text, bot=bot) answers = await bot.answers.filter(text=message.text)
await template.save() if answers:
await message.answer("Такой текст уже есть в списке шаблонов")
else:
template = DefaultAnswer(text=message.text, bot=bot)
await template.save()
await send_bot_templates_menu(bot, chat_id=message.chat.id) await send_bot_templates_menu(bot, chat_id=message.chat.id)