I add command /id and /dowmp3 for dowload video with Youtube and i improve code.

It's version 0.2.0
2025-10-04 18:56:50 +03:00
parent 7702c9a85b
commit 5197518029
10 changed files with 400 additions and 85 deletions
@@ -0,0 +1,81 @@
+from aiogram import types, Dispatcher, Bot
+from aiogram.filters import Command
+from models.state import BotState
+from utils.antispam import admin_required
+from logging import getLogger
+from .dowloadmp3_to_youtube import *
+from os import path, unlink
+
+logger = getLogger(__name__)
+
+def register_handlers(dp: Dispatcher, state: BotState, bot: Bot):
+    @dp.message(Command("dowmp3"))
+    @admin_required(5)
+    async def cmd_dowmp3(message: types.Message):
+        args = message.text.split(maxsplit=1)
+        if len(args) < 2:
+            await message.reply("❌ Укажи ссылку: /dowmp3 <youtube_url>")
+            return
+
+        url = args[1]
+        logger.info(f"Получена команда /dowmp3 от user_id={message.from_user.id}, url={url}")
+
+        status_msg = await message.reply("⏳ Скачиваю аудио... Это займет 1-2 минуты")
+
+        try:
+            filename, metadata = await download_mp3_isolated(url)
+            file_size = path.getsize(filename)
+
+            if file_size < 1000:
+                raise Exception("Файл слишком маленький")
+
+            await status_msg.edit_text(f"✅ Аудио готово! Отправляю...")
+
+            # Подготавливаем аудио файл
+            audio_input = types.FSInputFile(filename)
+
+            # Базовые параметры
+            send_params = {
+                'audio': audio_input,
+                'title': metadata['title'][:64],
+                'performer': metadata['performer'][:64],
+                'duration': int(metadata['duration']) if metadata['duration'] else None,
+                'caption': f"🎵 {metadata['title']}\n👤 {metadata['performer']}"
+            }
+
+            # Добавляем обложку если есть
+            if metadata['thumbnail_data']:
+                try:
+                    # Создаем временный файл для обложки
+                    with tempfile.NamedTemporaryFile(suffix='.jpg', delete=False) as thumb_file:
+                        thumb_filename = thumb_file.name
+                        thumb_file.write(metadata['thumbnail_data'])
+
+                    # Используем FSInputFile для обложки
+                    send_params['thumbnail'] = types.FSInputFile(thumb_filename)
+                    logger.info("Обложка добавлена к сообщению")
+
+                except Exception as e:
+                    logger.warning(f"Не удалось добавить обложку: {e}")
+
+            # Отправляем аудио
+            await message.answer_audio(**send_params)
+
+            # Удаляем временный файл обложки если создавали
+            if 'thumb_filename' in locals() and path.exists(thumb_filename):
+                unlink(thumb_filename)
+
+            await status_msg.delete()
+            logger.info(f"Аудио отправлено пользователю {message.from_user.id}")
+
+        except asyncio.TimeoutError:
+            await status_msg.edit_text("❌ Превышено время ожидания (5 минут)")
+        except Exception as e:
+            await status_msg.edit_text(f"❌ Ошибка: {str(e)}")
+            logger.error(f"Ошибка при скачивании: {e}")
+        finally:
+            if 'filename' in locals() and path.exists(filename):
+                try:
+                    unlink(filename)
+                except:
+                    pass