Update app.py

app.py

@@ -1,313 +1,282 @@
-import asyncio
-import logging
-import sqlite3
 import os
-import io
+import asyncio
 import threading
-from pathlib import Path
-
+import uuid
+from flask import Flask, request, render_template_string, redirect, url_for
 from aiogram import Bot, Dispatcher, types
+from aiogram.types import InputFile
 from aiogram.utils import executor
-from aiogram.types import InputFile, ParseMode
-
-from flask import Flask, request, render_template_string, redirect, url_for, flash
+from aiogram.contrib.fsm_storage.memory import MemoryStorage
 
 # --- Configuration ---
-BOT_TOKEN = "7835463659:AAGNePbelZIAOeaglyQi1qulOqnjs4BGQn4"
+BOT_TOKEN = "7835463659:AAGNePbelZIAOeaglyQi1qulOqnjs4BGQn4"  # Ваш токен бота
 FLASK_HOST = "0.0.0.0"
 FLASK_PORT = 7860
-DB_NAME = "subscribers.db"
-UPLOAD_FOLDER = 'uploads' # For temporary file storage if needed, though we'll use streams
 
-# --- Logging ---
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
+# --- Shared State ---
+# Используем set для хранения уникальных chat_id
+subscribed_users = set()
+# Временная директория для загружаемых медиафайлов
+TEMP_UPLOAD_FOLDER = 'temp_uploads'
+os.makedirs(TEMP_UPLOAD_FOLDER, exist_ok=True)
 
-# --- Database Setup ---
-def init_db():
-    conn = sqlite3.connect(DB_NAME)
-    cursor = conn.cursor()
-    cursor.execute('''
-        CREATE TABLE IF NOT EXISTS users (
-            user_id INTEGER PRIMARY KEY
-        )
-    ''')
-    conn.commit()
-    conn.close()
-    logger.info("Database initialized.")
+# --- Aiogram Bot ---
+# Инициализируем объекты бота и диспетчера
+# Важно: создаем loop явно для использования в другом потоке
+loop = asyncio.get_event_loop()
+bot = Bot(token=BOT_TOKEN)
+storage = MemoryStorage()
+dp = Dispatcher(bot, storage=storage, loop=loop)
 
-def add_user(user_id: int):
-    conn = sqlite3.connect(DB_NAME)
-    cursor = conn.cursor()
+@dp.message_handler(commands=['start'])
+async def process_start_command(message: types.Message):
+    """
+    Handler для команды /start.
+    Добавляет пользователя в список подписанных и отправляет приветственное сообщение.
+    """
+    chat_id = message.chat.id
+    if chat_id not in subscribed_users:
+        subscribed_users.add(chat_id)
+        print(f"User {chat_id} subscribed.")
+        await message.reply("Вы подписались на уведомления!")
+    else:
+         await message.reply("Вы уже подписаны на уведомления!") # Опционально: сообщение для уже подписанных
+
+async def send_notification_to_user(chat_id, text, media_path=None):
+    """
+    Асинхронная функция для отправки уведомления одному пользователю.
+    """
     try:
-        cursor.execute("INSERT INTO users (user_id) VALUES (?)", (user_id,))
-        conn.commit()
-        logger.info(f"User {user_id} added to database.")
-        return True
-    except sqlite3.IntegrityError:
-        logger.info(f"User {user_id} already exists in database.")
-        return False # Already exists
-    finally:
-        conn.close()
+        if media_path and os.path.exists(media_path):
+            # Отправляем медиа с текстом в качестве подписи
+            with open(media_path, 'rb') as photo:
+                 await bot.send_photo(chat_id, photo=InputFile(photo), caption=text)
+        elif text:
+            # Отправляем только текст
+             await bot.send_message(chat_id, text=text)
+        else:
+            print(f"Skipping sending to {chat_id}: No text or media provided.")
 
-def get_all_users():
-    conn = sqlite3.connect(DB_NAME)
-    cursor = conn.cursor()
-    cursor.execute("SELECT user_id FROM users")
-    users = [row[0] for row in cursor.fetchall()]
-    conn.close()
-    return users
+    except Exception as e:
+        # Обрабатываем ошибки отправки (например, пользователь заблокировал бота)
+        print(f"Failed to send message to {chat_id}: {e}")
+        # Опционально: удалить пользователя из subscribed_users, если ошибка указывает на блокировку
+        # if "bot was blocked by the user" in str(e):
+        #     subscribed_users.discard(chat_id)
 
-# --- Aiogram Bot Setup ---
-bot = Bot(token=BOT_TOKEN)
-dp = Dispatcher(bot)
 
-# --- Flask App Setup ---
-flask_app = Flask(__name__)
-flask_app.secret_key = os.urandom(24) # For flash messages
-if not os.path.exists(UPLOAD_FOLDER):
-    os.makedirs(UPLOAD_FOLDER)
-flask_app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+async def send_notification_to_all(text, media_path=None):
+    """
+    Асинхронная функция для отправ��и уведомления всем подписанным пользователям.
+    """
+    print(f"Attempting to send notification to {len(subscribed_users)} users.")
+    # Создаем копию set, чтобы избежать проблем при изменении set во время итерации
+    users_to_notify = list(subscribed_users)
+    for chat_id in users_to_notify:
+        await send_notification_to_user(chat_id, text, media_path)
+        # Небольшая задержка, чтобы избежать превышения лимитов Telegram API
+        await asyncio.sleep(0.05) # 50ms delay per user
 
-# --- HTML Templates (Inline) ---
-ADMIN_PANEL_HTML = """
-<!DOCTYPE html>
+# --- Flask Admin Panel ---
+app = Flask(__name__)
+
+# HTML для админ панели
+ADMIN_HTML = """
+<!doctype html>
 <html lang="ru">
 <head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>Админ Панель Уведомлений</title>
+    <meta charset="utf-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1, shrink-to-fit=no">
+    <title>Админ панель бота</title>
     <style>
-        body { font-family: Arial, sans-serif; margin: 0; padding: 20px; background-color: #f4f4f4; color: #333; }
-        .container { background-color: #fff; padding: 20px; border-radius: 8px; box-shadow: 0 0 10px rgba(0,0,0,0.1); max-width: 600px; margin: auto; }
-        h1 { text-align: center; color: #333; }
-        label { display: block; margin-bottom: 8px; font-weight: bold; }
+        body {
+            font-family: sans-serif;
+            line-height: 1.6;
+            margin: 20px;
+            background-color: #f8f8f8;
+            color: #333;
+        }
+        .container {
+            max-width: 600px;
+            margin: 0 auto;
+            background-color: #fff;
+            padding: 20px;
+            border-radius: 8px;
+            box-shadow: 0 2px 5px rgba(0,0,0,0.1);
+        }
+        h1 {
+            text-align: center;
+            color: #0056b3;
+        }
+        form {
+            display: flex;
+            flex-direction: column;
+        }
+        label {
+            margin-bottom: 5px;
+            font-weight: bold;
+        }
         textarea, input[type="file"] {
-            width: calc(100% - 22px);
+            margin-bottom: 15px;
             padding: 10px;
-            margin-bottom: 20px;
-            border: 1px solid #ddd;
+            border: 1px solid #ccc;
             border-radius: 4px;
-            box-sizing: border-box;
+            font-size: 1rem;
+        }
+        textarea {
+            resize: vertical;
+            min-height: 100px;
         }
         button {
+            padding: 10px 15px;
             background-color: #007bff;
             color: white;
-            padding: 10px 20px;
             border: none;
             border-radius: 4px;
             cursor: pointer;
-            font-size: 16px;
-            display: block;
-            width: 100%;
+            font-size: 1rem;
+            transition: background-color 0.3s ease;
+        }
+        button:hover {
+            background-color: #0056b3;
         }
-        button:hover { background-color: #0056b3; }
-        .flash-messages { list-style: none; padding: 0; margin-bottom: 20px; }
-        .flash-messages li {
+        .flash-message {
             padding: 10px;
-            margin-bottom: 10px;
+            margin-bottom: 15px;
             border-radius: 4px;
+            background-color: #d4edda;
+            color: #155724;
+            border: 1px solid #c3e6cb;
         }
-        .flash-success { background-color: #d4edda; color: #155724; border: 1px solid #c3e6cb; }
-        .flash-error { background-color: #f8d7da; color: #721c24; border: 1px solid #f5c6cb; }
-        #preview-container { margin-top: 15px; text-align: center; }
-        #media-preview { max-width: 100%; max-height: 200px; border: 1px solid #ddd; margin-top: 5px; }
     </style>
 </head>
 <body>
     <div class="container">
-        <h1>Отправить Уведомление</h1>
-        {% with messages = get_flashed_messages(with_categories=true) %}
-          {% if messages %}
-            <ul class="flash-messages">
-            {% for category, message in messages %}
-              <li class="flash-{{ category }}">{{ message }}</li>
-            {% endfor %}
-            </ul>
-          {% endif %}
+        <h1>Отправить уведомление</h1>
+        {% with messages = get_flashed_messages() %}
+            {% if messages %}
+                {% for message in messages %}
+                    <div class="flash-message">{{ message }}</div>
+                {% endfor %}
+            {% endif %}
         {% endwith %}
-        <form method="POST" action="{{ url_for('send_notification_route') }}" enctype="multipart/form-data" onsubmit="return showLoader();">
-            <div>
-                <label for="message">Текст сообщения (HTML поддерживается):</label>
-                <textarea name="message" id="message" rows="5" placeholder="Введите ваше сообщение..."></textarea>
-            </div>
-            <div>
-                <label for="media">Медиафайл (фото, видео, документ):</label>
-                <input type="file" name="media" id="media" onchange="previewFile()">
-            </div>
-            <div id="preview-container">
-                <img id="media-preview" src="#" alt="Предпросмотр медиа" style="display:none;"/>
-                <video id="video-preview" controls style="display:none; max-width: 100%; max-height: 200px;"></video>
-                <p id="file-name-preview" style="display:none;"></p>
-            </div>
-            <button type="submit" id="submit-button">Отправить Всем</button>
-            <div id="loader" style="display:none; text-align:center; margin-top:15px;">Отправка... Пожалуйста, подождите.</div>
-        </form>
-    </div>
-    <script>
-        function previewFile() {
-            const preview = document.getElementById('media-preview');
-            const videoPreview = document.getElementById('video-preview');
-            const fileNamePreview = document.getElementById('file-name-preview');
-            const file = document.getElementById('media').files[0];
-            const reader = new FileReader();
+        <p>Подписано пользователей: {{ user_count }}</p>
+        <form action="{{ url_for('send_notification') }}" method="post" enctype="multipart/form-data">
+            <label for="message">Текст уведомления:</label>
+            <textarea id="message" name="message" placeholder="Введите текст уведомления"></textarea>
 
-            preview.style.display = 'none';
-            videoPreview.style.display = 'none';
-            fileNamePreview.style.display = 'none';
-            videoPreview.src = ''; // Reset video src
+            <label for="media">Медиафайл (фото):</label>
+            <input type="file" id="media" name="media" accept="image/*">
 
-            if (file) {
-                reader.onloadend = function() {
-                    if (file.type.startsWith('image/')) {
-                        preview.src = reader.result;
-                        preview.style.display = 'block';
-                    } else if (file.type.startsWith('video/')) {
-                        videoPreview.src = reader.result;
-                        videoPreview.style.display = 'block';
-                    } else {
-                        fileNamePreview.textContent = "Выбран файл: " + file.name;
-                        fileNamePreview.style.display = 'block';
-                    }
-                }
-                reader.readAsDataURL(file);
-            }
-        }
-        function showLoader() {
-            document.getElementById('submit-button').style.display = 'none';
-            document.getElementById('loader').style.display = 'block';
-            return true; // Proceed with form submission
-        }
-    </script>
+            <button type="submit">Отправить всем</button>
+        </form>
+    </div>
 </body>
 </html>
 """
 
-# --- Aiogram Handlers ---
-@dp.message_handler(commands=['start'])
-async def send_welcome(message: types.Message):
-    user_id = message.from_user.id
-    username = message.from_user.username or message.from_user.first_name
-    if add_user(user_id):
-        await message.reply("🎉 Вы успешно подписались на уведомления! 🎉")
-        logger.info(f"User {username} (ID: {user_id}) subscribed.")
-    else:
-        await message.reply("Вы уже подписаны на уведомления. 👍")
-        logger.info(f"User {username} (ID: {user_id}) tried to subscribe again.")
+from flask import flash # Импортируем flash для сообщений
 
-# --- Flask Routes ---
-@flask_app.route('/', methods=['GET'])
-def admin_panel_route():
-    return render_template_string(ADMIN_PANEL_HTML)
+@app.route('/admin')
+def admin_panel():
+    """
+    Отображает страницу админ панели.
+    """
+    return render_template_string(ADMIN_HTML, user_count=len(subscribed_users))
 
-@flask_app.route('/send_notification', methods=['POST'])
-def send_notification_route():
-    message_text = request.form.get('message', '')
+@app.route('/send_notification', methods=['POST'])
+def send_notification():
+    """
+    Обрабатывает отправку уведомления из админ панели.
+    """
+    message_text = request.form.get('message', '').strip()
     media_file = request.files.get('media')
+    media_path = None
 
-    if not message_text and not media_file:
-        flash("Сообщение не может быть пустым, если не прикреплен медиафайл.", "error")
-        return redirect(url_for('admin_panel_route'))
-
-    users = get_all_users()
-    if not users:
-        flash("Нет подписчиков для отправки уведомлений.", "error")
-        return redirect(url_for('admin_panel_route'))
-
-    sent_count = 0
-    error_count = 0
+    # Проверяем, загружен ли файл
+    if media_file and media_file.filename != '':
+        # Создаем уникальное имя файла и сохраняем его временно
+        filename = str(uuid.uuid4()) + os.path.splitext(media_file.filename)[1]
+        media_path = os.path.join(TEMP_UPLOAD_FOLDER, filename)
+        try:
+            media_file.save(media_path)
+            print(f"Saved media file to {media_path}")
+        except Exception as e:
+            print(f"Error saving file: {e}")
+            flash("Ошибка при сохранении медиафайла.", "error")
+            return redirect(url_for('admin_panel'))
+    
+    # Проверяем, есть ли текст или медиа
+    if not message_text and not media_path:
+         flash("Пожалуйста, введите текст или прикрепите медиафайл.", "warning")
+         if media_path and os.path.exists(media_path):
+             os.remove(media_path) # Удалить временный файл, если он был создан
+         return redirect(url_for('admin_panel'))
 
-    # We need to run async bot functions from this synchronous Flask thread
-    # Get the bot's event loop
-    loop = bot.loop # or dp.loop
+    # Вызываем асинхронную функцию отправки уведомлений в контексте asyncio loop
+    # run_coroutine_threadsafe отправляет корутину в loop, который выполняется в другом потоке
+    try:
+        future = asyncio.run_coroutine_threadsafe(
+            send_notification_to_all(message_text, media_path),
+            loop # Используем loop бота
+        )
+        # Ждем завершения выполнения корутины (опционально, можно запустить в фоне)
+        # .result() делает этот Flask запрос блокирующим до отправки всем пользователям
+        future.result()
+        flash(f"Уведомление отправлено {len(subscribed_users)} пользователям.", "success")
+        print("Notification sent successfully.")
 
-    for user_id in users:
-        try:
-            if media_file:
-                # Reset stream position for each send
-                media_file.seek(0)
-                file_content = io.BytesIO(media_file.read())
-                file_content.name = media_file.filename # Important for aiogram to determine type
-                
-                input_file = InputFile(file_content, filename=media_file.filename)
-                
-                content_type = media_file.content_type
-                logger.info(f"Attempting to send media with content type: {content_type} to {user_id}")
+    except Exception as e:
+        print(f"Error sending notification: {e}")
+        flash(f"Произошла ошибка при отправке: {e}", "error")
 
-                coro = None
-                if 'image' in content_type:
-                    coro = bot.send_photo(user_id, input_file, caption=message_text if message_text else None, parse_mode=ParseMode.HTML)
-                elif 'video' in content_type:
-                    coro = bot.send_video(user_id, input_file, caption=message_text if message_text else None, parse_mode=ParseMode.HTML)
-                elif 'audio' in content_type: # Added audio support
-                    coro = bot.send_audio(user_id, input_file, caption=message_text if message_text else None, parse_mode=ParseMode.HTML)
-                else: # Default to document
-                    coro = bot.send_document(user_id, input_file, caption=message_text if message_text else None, parse_mode=ParseMode.HTML)
-                
-                if coro:
-                    # Run the coroutine in the bot's event loop
-                    future = asyncio.run_coroutine_threadsafe(coro, loop)
-                    future.result(timeout=30) # Wait for completion with timeout
-                else: # If only text was provided with media, but media type was not sendable this way
-                    if message_text:
-                         coro_text = bot.send_message(user_id, message_text, parse_mode=ParseMode.HTML)
-                         future = asyncio.run_coroutine_threadsafe(coro_text, loop)
-                         future.result(timeout=10)
+    finally:
+        # Удаляем временный файл после отправки
+        if media_path and os.path.exists(media_path):
+            try:
+                os.remove(media_path)
+                print(f"Cleaned up temporary file {media_path}")
+            except Exception as e:
+                print(f"Error cleaning up file {media_path}: {e}")
+        # Опционально: удалить пустую временную директорию
+        # if not os.listdir(TEMP_UPLOAD_FOLDER):
+        #     os.rmdir(TEMP_UPLOAD_FOLDER)
 
-            elif message_text: # Only text
-                coro = bot.send_message(user_id, message_text, parse_mode=ParseMode.HTML)
-                future = asyncio.run_coroutine_threadsafe(coro, loop)
-                future.result(timeout=10) # Wait for completion with timeout
-            
-            sent_count += 1
-            logger.info(f"Notification sent to {user_id}")
-        
-        except Exception as e:
-            error_count += 1
-            logger.error(f"Failed to send notification to {user_id}: {e}")
-            # Common errors: bot blocked by user, chat not found, timeout
-            if "bot was blocked by the user" in str(e).lower():
-                logger.warning(f"User {user_id} blocked the bot. Consider removing them.")
-            # You might want to remove users who have blocked the bot from the DB here
-    
-    if media_file: # Close the original file stream if it was opened
-        media_file.close()
 
-    flash(f"Уведомление отправлено {sent_count} пользователям. Ошибок: {error_count}.", "success" if error_count == 0 else "error")
-    return redirect(url_for('admin_panel_route'))
+    return redirect(url_for('admin_panel')) # Перенаправляем обратно на админ панель
 
 
-# --- Functions to run Flask and Aiogram ---
-def run_flask():
-    # Use '0.0.0.0' to make it accessible externally, False for debug in "production"
-    flask_app.run(host=FLASK_HOST, port=FLASK_PORT, debug=False, use_reloader=False)
+# --- Running Both ---
 
-async def main_bot_loop():
-    # Start polling
-    logger.info("Starting bot polling...")
-    # No need to pass bot, dp will use the one it was initialized with
-    await dp.start_polling()
+def start_bot_polling():
+    """
+    Функция для запуска поллинга бота в отдельном потоке.
+    """
+    print("Starting bot polling...")
+    # Устанавливаем loop для этого потока, хотя aiogram 2.x уже делает это
+    # asyncio.set_event_loop(loop) # Возможно, не нужно с aiogram 2.x+ и explicit loop
+    # executor.start_polling блокирует выполнение
+    # Мы запускаем его в отдельном потоке, чтобы основной поток мог запустить Flask
+    loop.run_until_complete(dp.start_polling())
 
 
 if __name__ == '__main__':
-    init_db() # Initialize database on startup
+    # Flask требует SECRET_KEY для flash сообщений
+    app.secret_key = 'super secret key for flask notifications' # Замените на реальный секретный ключ
 
-    # Run Flask in a separate thread
-    # daemon=True ensures the thread exits when the main program exits
-    flask_thread = threading.Thread(target=run_flask, daemon=True)
-    flask_thread.start()
-    logger.info(f"Flask admin panel running on http://{FLASK_HOST}:{FLASK_PORT}")
+    # Создаем и запускаем поток для бота
+    bot_thread = threading.Thread(target=start_bot_polling, daemon=True) # daemon=True позволяет приложению завершиться, даже если поток бота еще работает
+    bot_thread.start()
 
-    # Run Aiogram in the main thread's asyncio loop
-    # executor.start_polling(dp, skip_updates=True) # This is a blocking call
-    # For more control or if mixing with other asyncio tasks:
-    loop = asyncio.get_event_loop()
-    try:
-        loop.run_until_complete(main_bot_loop())
-    except KeyboardInterrupt:
-        logger.info("Bot polling stopped by user.")
-    finally:
-        logger.info("Shutting down...")
-        # You might want to close bot session here if needed, though start_polling handles it
-        # await bot.session.close() # if you are managing session explicitly
-        loop.close()
+    # Запускаем Flask веб-сервер в основном потоке
+    print(f"Starting Flask admin panel on http://{FLASK_HOST}:{FLASK_PORT}/admin")
+    # use_reloader=False очень важен, чтобы избежать запуска Flask в двух процессах,
+    # что привело бы к двойному запуску бота и проблемам с потоками/логами.
+    app.run(host=FLASK_HOST, port=FLASK_PORT, use_reloader=False)
+
+    # Остановка бота при завершении Flask (опционально, т.к. daemon=True)
+    # print("Stopping bot...")
+    # loop.call_soon_threadsafe(dp.stop_polling)
+    # bot_thread.join(timeout=5) # Ждем завершения потока бота
+    # print("Bot stopped.")
+    # loop.close() # Закрываем loop асинхронных операций (важно для чистого завер��ения)