add new markdown

1 files changed, 49 insertions, 7 deletions

diff --git a/commands.py b/commands.py
index f8f5678..9944314 100644
--- a/commands.py
+++ b/commands.py
@@ -2,10 +2,11 @@ from io import BytesIO
 from typing import Optional, Union
 
 from aiogram import Bot, Dispatcher, F, Router
+from aiogram.methods import SendRichMessage
 from aiogram.client.default import DefaultBotProperties
 from aiogram.enums import ChatAction, ChatType
 from aiogram.exceptions import TelegramForbiddenError
-from aiogram.filters import Command
+from aiogram.filters import Command, CommandStart, CommandObject
 from aiogram.types import (
     BufferedInputFile,
     CallbackQuery,
@@ -14,6 +15,7 @@ from aiogram.types import (
     InlineKeyboardMarkup,
     InlineQuery,
     InlineQueryResultArticle,
+    InputRichMessage,
     InputTextMessageContent,
     Message,
     User,
@@ -29,7 +31,7 @@ from supergenerator import *
 
 router = Router()
 
-commands = ["/stats", "/broadcast", "/ben", "/reporn", "/start", "/prompt", "/config", "/settings", "/clear", "/stop"]
+commands = ["/stats", "/broadcast", "/ben", "/reporn", "/start", "/prompt", "/config", "/settings", "/clear", "/stop", "/markpidor"]
 
 @router.message(Command("stats"))
 async def stats_handler(message: Message):
@@ -113,11 +115,19 @@ async def stop_handler(message: Message):
         user_id = message.from_user.id
         
     config = get_user_config(user_id)
-    set_user_config(user_id, config.get("model"), config.get("stream"), config.get("call"), config.get("prompt"), "net")
+    set_user_config(
+        user_id, 
+        config.get("model"), 
+        config.get("stream"), 
+        config.get("call"), 
+        config.get("prompt"), 
+        "net", 
+        config.get("markdown")
+    )
     await message.reply("✅ *рассылка успешно отключена!* чтобы включить её обратно:\n/settings --> рассылка")
 
 @router.message(Command("ben"))
-async def start_handler(message: Message):
+async def ben_handler(message: Message):
     if message.chat.type in [ChatType.GROUP, ChatType.SUPERGROUP]:
         if message.chat.id in BEN_IDS:
             replied = message.reply_to_message or None
@@ -138,7 +148,7 @@ async def start_handler(message: Message):
 
 
 @router.message(Command("reporn"))
-async def start_handler(message: Message):
+async def reporn_handler(message: Message):
     user_name = message.from_user.first_name or message.sender_chat.title or "вы"
     if message.chat.type in [ChatType.GROUP, ChatType.SUPERGROUP]:
         if message.chat.id in REPORN_IDS:
@@ -157,8 +167,8 @@ async def start_handler(message: Message):
     return
         
 
-@router.message(Command("start"))
-async def start_handler(message: Message):
+@router.message(CommandStart())
+async def start_handler(message: Message, command: CommandObject):
     if message.chat.type in [ChatType.GROUP, ChatType.SUPERGROUP]:
         await message.reply(
             "привет, запусти эту команду в ЛиС (личные сообщения) бота✅"
@@ -169,6 +179,32 @@ async def start_handler(message: Message):
         user_id = message.sender_chat.id
     else:
         user_id = message.from_user.id
+    
+    config = get_user_config(user_id)
+    
+    payload = command.args
+    if payload == "new_md":
+        set_user_config(
+            user_id, 
+            config.get("model"), 
+            config.get("stream"), 
+            config.get("call"), 
+            config.get("prompt"), 
+            config.get("notify"), 
+            "new"
+        )
+        await message.answer("✅ установлен *новый* стиль выделения сообщений!")
+    elif payload == "old_md":
+        set_user_config(
+            user_id, 
+            config.get("model"), 
+            config.get("stream"), 
+            config.get("call"), 
+            config.get("prompt"), 
+            config.get("notify"), 
+            "old"
+        )
+        await message.answer("✅ установлен *старый* стиль выделения сообщений!")
 
     invite_url = f"https://t.me/{BOT_USERNAME}?startgroup=start"
 
@@ -345,3 +381,9 @@ async def clear_handler(message: Message):
         user_contexts[user_id] = []
     except Exception as e:
         print(f"{RED} -- {e}{RESET}")
+
+        
+@router.message(Command("markpidor"))
+async def alo(message: Message):
+    await message.reply_rich(InputRichMessage(markdown="# ало"))
+    return