add_group, add_Channel process modification

handlers/add_channel.py

@@ -1,18 +1,62 @@
+
 from telegram import Update
-from telegram.ext import ContextTypes
+from telegram.ext import ContextTypes, ConversationHandler, CommandHandler, MessageHandler, filters
 from db import AsyncSessionLocal
 from models import Channel
 
-async def add_channel(update: Update, context: ContextTypes.DEFAULT_TYPE):
-    args = context.args or []
-    if update.message is None:
-        return
-    if len(args) < 2:
-        await update.message.reply_text('Используйте: /add_channel <название> <ссылка>')
-        return
-    name, link = args[0], args[1]
+INPUT_NAME, INPUT_LINK = range(2)
+
+async def add_channel_start(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    if context.user_data is None:
+        context.user_data = {}
+    if update.message:
+        await update.message.reply_text('Введите название канала (или "название ссылка" через пробел):')
+    return INPUT_NAME
+
+async def input_channel_name(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    if context.user_data is None:
+        context.user_data = {}
+    text = update.message.text.strip() if update.message and update.message.text else ''
+    if ' ' in text:
+        name, link = text.split(' ', 1)
+        context.user_data['channel_name'] = name
+        context.user_data['channel_link'] = link
+        return await save_channel(update, context)
+    else:
+        context.user_data['channel_name'] = text
+        if update.message:
+            await update.message.reply_text('Теперь отправьте ссылку на канал:')
+        return INPUT_LINK
+
+async def input_channel_link(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    if context.user_data is None:
+        context.user_data = {}
+    link = update.message.text.strip() if update.message and update.message.text else ''
+    context.user_data['channel_link'] = link
+    return await save_channel(update, context)
+
+async def save_channel(update: Update, context: ContextTypes.DEFAULT_TYPE):
+    if context.user_data is None:
+        context.user_data = {}
+    name = context.user_data.get('channel_name')
+    link = context.user_data.get('channel_link')
+    if not name or not link:
+        if update.message:
+            await update.message.reply_text('Ошибка: не указано название или ссылка.')
+        return ConversationHandler.END
     async with AsyncSessionLocal() as session:
         channel = Channel(name=name, link=link)
         session.add(channel)
         await session.commit()
-    await update.message.reply_text(f'Канал "{name}" добавлен.')
+    if update.message:
+        await update.message.reply_text(f'Канал "{name}" добавлен.')
+    return ConversationHandler.END
+
+add_channel_conv = ConversationHandler(
+    entry_points=[CommandHandler('add_channel', add_channel_start)],
+    states={
+        INPUT_NAME: [MessageHandler(filters.TEXT & ~filters.COMMAND, input_channel_name)],
+        INPUT_LINK: [MessageHandler(filters.TEXT & ~filters.COMMAND, input_channel_link)],
+    },
+    fallbacks=[]
+)