From a3eb985d28332f102746e30704ee2472ed477e8a Mon Sep 17 00:00:00 2001
From: mihalin <stopa3.10.1996@mail.ru>
Date: Fri, 11 Feb 2022 21:19:30 +0300
Subject: [PATCH] minor fixes

---
 olgram/commands/menu.py | 18 ++++++++++++++----
 olgram/utils/mix.py     |  7 +++++--
 2 files changed, 19 insertions(+), 6 deletions(-)

diff --git a/olgram/commands/menu.py b/olgram/commands/menu.py
index 7473ce5..60c2ad3 100644
--- a/olgram/commands/menu.py
+++ b/olgram/commands/menu.py
@@ -5,7 +5,7 @@ from olgram.models.models import Bot, User, DefaultAnswer
 from aiogram.dispatcher import FSMContext
 from aiogram.utils.callback_data import CallbackData
 from textwrap import dedent
-from olgram.utils.mix import edit_or_create, button_text_limit
+from olgram.utils.mix import edit_or_create, button_text_limit, wrap
 from olgram.commands import bot_actions
 
 import typing as ty
@@ -237,7 +237,13 @@ async def send_bot_templates_menu(bot: Bot, call: ty.Optional[types.CallbackQuer
     """)
 
     templates = await bot.answers
-    templates_text = "\n".join(f"{n}. {template.text}" for n, template in enumerate(templates))
+
+    total_text_len = sum(len(t.text) for t in templates) + len(text)  # примерная длина текста
+    max_len = 1000
+    if total_text_len > 4000:
+        max_len = 100
+
+    templates_text = "\n".join(f"{n}. {wrap(template.text, max_len)}" for n, template in enumerate(templates))
     if not templates_text:
         templates_text = "(нет шаблонов)"
     text = text.format(bot.name, templates_text)
@@ -283,8 +289,12 @@ async def template_received(message: types.Message, state: FSMContext):
         await templates[number].delete()
     else:
         # Add template
-        template = DefaultAnswer(text=message.text, bot=bot)
-        await template.save()
+        total_templates = len(await bot.answers)
+        if total_templates > 30:
+            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)
 
diff --git a/olgram/utils/mix.py b/olgram/utils/mix.py
index 0130a56..a5aa382 100644
--- a/olgram/utils/mix.py
+++ b/olgram/utils/mix.py
@@ -22,8 +22,11 @@ async def edit_or_create(call: CallbackQuery, message: str,
                                     parse_mode=parse_mode)
 
 
-def button_text_limit(data: str) -> str:
-    max_len = 30
+def wrap(data: str, max_len: int) -> str:
     if len(data) > max_len:
         data = data[:max_len-4] + "..."
     return data
+
+
+def button_text_limit(data: str) -> str:
+    return wrap(data, 30)