my_converterbot/bot.py

import asyncio
import logging
import os
import shutil
import zipfile
from pathlib import Path
from tempfile import TemporaryDirectory

from aiogram import Bot, Dispatcher, F, Router
from aiogram.types import Message, BufferedInputFile
from aiogram.filters import Command
from aiogram.enums.parse_mode import ParseMode
from aiogram.fsm.storage.memory import MemoryStorage
from aiogram.client.bot import DefaultBotProperties

# Импорт наших конвертеров
from md_to_docx import MarkdownToDocxConverter, DocumentSettings
from rep_to_txt import generate_complete_project_structure

# Конфигурация
BOT_TOKEN = "**************************"  # @my_convbot

# Инициализация бота
bot = Bot(token=BOT_TOKEN, default=DefaultBotProperties(parse_mode=ParseMode.HTML))
dp = Dispatcher(storage=MemoryStorage())
router = Router()

logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

@router.message(Command("start"))
async def start_handler(msg: Message) -> None:
    """Приветственное сообщение"""
    await msg.answer(
        "<b>Приветствую в своём Конвертере!</b>\n\n"
        "📋 <b>Возможности:</b>\n"
        "• Отправьте .md файл → получите .docx\n"
        "• Отправьте .zip архив → получите структуру проекта в .txt\n\n"
        "📝 <b>/help</b> - Для подробной информации"
    )

@router.message(Command("help"))
async def help_handler(msg: Message) -> None:
    """Подробная справка"""
    help_text = (
        "<b>📚 Подробное руководство</b>\n\n"
        "<b>1. Конвертация Markdown → DOCX:</b>\n"
        "• Отправьте .md файл\n"
        "• Получите DOCX с форматированием по ГОСТ\n\n"
        "<b>2. Анализ архива → TXT:</b>\n"
        "• Отправьте .zip архив\n"
        "• Получите полную структуру проекта в текстовом файле\n\n"
        "<b>⚡ Ограничения:</b>\n"
        "• Размер файла: до 20 МБ\n"
        "• Поддерживаемые форматы: .md, .zip"
    )
    await msg.answer(help_text)

@router.message(F.document)
async def handle_document(msg: Message) -> None:
    """Обработка загруженных документов"""
    document = msg.document
    file_name = document.file_name
    file_size = document.file_size
    
    if file_size > 20 * 1024 * 1024:
        await msg.answer("❌ Файл слишком большой! Максимум 20 МБ")
        return
    
    file_ext = Path(file_name).suffix.lower()
    
    status_msg = await msg.answer("⏳ Обрабатываю файл...")
    
    try:
        with TemporaryDirectory() as temp_dir:
            # Скачиваем файл
            file_info = await bot.get_file(document.file_id)
            input_path = os.path.join(temp_dir, file_name)
            await bot.download_file(file_info.file_path, input_path)
            
            if file_ext == '.md':
                # Конвертация MD → DOCX
                output_path = await convert_md_to_docx(input_path, temp_dir)
                output_name = Path(file_name).stem + '.docx'
                
            elif file_ext in ['.zip']:
                # Анализ архива → TXT
                output_path = await analyze_archive(input_path, temp_dir, file_ext)
                output_name = Path(file_name).stem + '_structure.txt'
                
            else:
                await status_msg.edit_text("❌ Неподдерживаемый формат файла!")
                return
            
            # Отправка результата
            with open(output_path, 'rb') as output_file:
                result_file = BufferedInputFile(
                    output_file.read(),
                    filename=output_name
                )
                await msg.answer_document(result_file)
            
            await status_msg.edit_text("✅ Конвертация завершена!")
            
    except Exception as e:
        logger.error(f"Ошибка обработки файла: {e}")
        await status_msg.edit_text(f"❌ Ошибка обработки: {str(e)}")

async def convert_md_to_docx(md_path: str, temp_dir: str) -> str:
    """Конвертация Markdown в DOCX"""
    output_path = os.path.join(temp_dir, "output.docx")
    
    # Настройки
    settings = DocumentSettings()
    settings.font_name = "Times New Roman"
    settings.font_size = 14
    settings.line_spacing = 1.5
    settings.margin_left = 3.0
    settings.auto_numbering_headings = True
    
    converter = MarkdownToDocxConverter(settings)
    converter.convert(md_path, output_path)
    
    return output_path

async def analyze_archive(archive_path: str, temp_dir: str, file_ext: str) -> str:
    """Анализ архива и создание структуры проекта"""
    extract_dir = os.path.join(temp_dir, "extracted")
    os.makedirs(extract_dir, exist_ok=True)
    
    # Извлечение архива
    with zipfile.ZipFile(archive_path, 'r') as zip_ref:
        zip_ref.extractall(extract_dir)
    
    # Поиск основной папки проекта
    extracted_items = os.listdir(extract_dir)
    if len(extracted_items) == 1 and os.path.isdir(os.path.join(extract_dir, extracted_items[0])):
        project_root = os.path.join(extract_dir, extracted_items[0])
    else:
        project_root = extract_dir
    
    # Генерация структуры
    structure = generate_complete_project_structure(project_root)
    
    # Сохранение в файл
    output_path = os.path.join(temp_dir, "project_structure.txt")
    with open(output_path, 'w', encoding='utf-8') as f:
        f.write(structure)
    
    return output_path

@router.message()
async def handle_other_messages(msg: Message) -> None:
    """Обработка остальных сообщений"""
    await msg.answer(
        "<b>Отправьте файл для конвертации:</b>\n"
        "• .md файл для конвертации в DOCX\n"
        "• .zip архив для анализа структуры\n\n"
        "Используйте /help для подробной справки!"
    )

async def main() -> None:
    """Запуск бота"""
    try:
        logger.info("Запуск File Converter Bot...")
        dp.include_router(router)
        await bot.delete_webhook(drop_pending_updates=True)
        logger.info("Bot started successfully")

        await dp.start_polling(bot)
        
    except Exception as e:
        logger.critical(f"Критическая ошибка: {e}")
    finally:
        await bot.session.close()

if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        logger.info("Bot stopped by user")
    except Exception as e:
        logger.critical(f"Unhandled exception: {e}")