добавлен скрипт отправки сообщений

broadcast.py

@@ -0,0 +1,207 @@
+#!/usr/bin/env python3
+"""
+Скрипт для рассылки сообщений всем пользователям бота.
+
+Использование:
+    python broadcast.py "Текст сообщения"
+    python broadcast.py --html "<b>Жирный</b> текст"
+    python broadcast.py --file message.txt
+"""
+import os
+import sys
+import sqlite3
+import argparse
+from pathlib import Path
+
+# Загружаем переменные окружения из .env если есть
+def load_env_file():
+    env_file = Path(__file__).resolve().parent / '.env'
+    if env_file.exists():
+        with open(env_file, 'r') as f:
+            for line in f:
+                line = line.strip()
+                if line and not line.startswith('#') and '=' in line:
+                    key, value = line.split('=', 1)
+                    os.environ.setdefault(key.strip(), value.strip())
+
+load_env_file()
+
+import requests
+import time
+
+# Настройки
+BASE_DIR = Path(__file__).resolve().parent
+DB_FILE = BASE_DIR / 'data' / 'bot.db'
+TELEGRAM_BOT_TOKEN = os.getenv('TELEGRAM_BOT_TOKEN')
+
+
+def get_telegram_api_url():
+    return f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}"
+
+
+def get_all_users() -> list[tuple[int, str, str]]:
+    """Получает всех пользователей из базы данных"""
+    if not DB_FILE.exists():
+        print(f"❌ База данных не найдена: {DB_FILE}")
+        sys.exit(1)
+    
+    conn = sqlite3.connect(str(DB_FILE))
+    cursor = conn.cursor()
+    cursor.execute('SELECT chat_id, username, locale FROM users')
+    users = cursor.fetchall()
+    conn.close()
+    return users
+
+
+def send_message(chat_id: int, text: str, parse_mode: str = None) -> bool:
+    """Отправляет сообщение пользователю"""
+    payload = {
+        "chat_id": chat_id,
+        "text": text,
+    }
+    if parse_mode:
+        payload["parse_mode"] = parse_mode
+    
+    try:
+        response = requests.post(
+            f"{get_telegram_api_url()}/sendMessage",
+            json=payload,
+            timeout=30
+        )
+        
+        if response.status_code == 200:
+            return True
+        else:
+            result = response.json()
+            error = result.get('description', 'Unknown error')
+            # Пользователь заблокировал бота или удалил чат
+            if 'blocked' in error.lower() or 'chat not found' in error.lower():
+                return False
+            print(f"  ⚠️  {chat_id}: {error}")
+            return False
+    except Exception as e:
+        print(f"  ❌ {chat_id}: {e}")
+        return False
+
+
+def broadcast(text: str, parse_mode: str = None):
+    """Рассылает сообщение всем пользователям"""
+    users = get_all_users()
+    
+    if not users:
+        print("❌ Нет пользователей в базе данных")
+        return
+    
+    print(f"📤 Начинаю рассылку для {len(users)} пользователей...")
+    print(f"📝 Сообщение:\n{text[:100]}{'...' if len(text) > 100 else ''}\n")
+    
+    success = 0
+    failed = 0
+    
+    for chat_id, username, locale in users:
+        result = send_message(chat_id, text, parse_mode)
+        if result:
+            success += 1
+            print(f"  ✅ {chat_id} (@{username or 'no_username'})")
+        else:
+            failed += 1
+        
+        # Небольшая задержка чтобы не превысить лимиты Telegram
+        time.sleep(0.05)
+    
+    print(f"\n📊 Результат:")
+    print(f"  ✅ Успешно: {success}")
+    print(f"  ❌ Ошибок: {failed}")
+    print(f"  📨 Всего: {len(users)}")
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description='Рассылка сообщений всем пользователям бота',
+        formatter_class=argparse.RawDescriptionHelpFormatter,
+        epilog="""
+Примеры:
+  python broadcast.py "Привет! Это важное сообщение."
+  python broadcast.py --html "<b>Важно!</b> Новая функция добавлена."
+  python broadcast.py --file announcement.txt
+  python broadcast.py --file announcement.txt --html
+        """
+    )
+    
+    parser.add_argument(
+        'message',
+        nargs='?',
+        help='Текст сообщения для рассылки'
+    )
+    parser.add_argument(
+        '--html',
+        action='store_true',
+        help='Использовать HTML-разметку в сообщении'
+    )
+    parser.add_argument(
+        '--file', '-f',
+        type=str,
+        help='Файл с текстом сообщения'
+    )
+    parser.add_argument(
+        '--list', '-l',
+        action='store_true',
+        help='Показать список пользователей без отправки'
+    )
+    parser.add_argument(
+        '--yes', '-y',
+        action='store_true',
+        help='Пропустить подтверждение'
+    )
+    
+    args = parser.parse_args()
+    
+    # Проверяем токен
+    if not TELEGRAM_BOT_TOKEN:
+        print("❌ TELEGRAM_BOT_TOKEN не установлен!")
+        print("   Установите переменную окружения или создайте .env файл")
+        sys.exit(1)
+    
+    # Показать список пользователей
+    if args.list:
+        users = get_all_users()
+        print(f"👥 Всего пользователей: {len(users)}\n")
+        for chat_id, username, locale in users:
+            print(f"  {chat_id:>12} | @{username or 'no_username':<20} | {locale}")
+        sys.exit(0)
+    
+    # Получаем текст сообщения
+    if args.file:
+        file_path = Path(args.file)
+        if not file_path.exists():
+            print(f"❌ Файл не найден: {args.file}")
+            sys.exit(1)
+        text = file_path.read_text(encoding='utf-8')
+    elif args.message:
+        text = args.message
+    else:
+        parser.print_help()
+        sys.exit(1)
+    
+    if not text.strip():
+        print("❌ Сообщение не может быть пустым")
+        sys.exit(1)
+    
+    # Подтверждение
+    users = get_all_users()
+    print(f"⚠️  Вы собираетесь отправить сообщение {len(users)} пользователям.")
+    
+    if not args.yes:
+        confirm = input("Продолжить? (y/N): ").strip().lower()
+        if confirm != 'y':
+            print("❌ Отменено")
+            sys.exit(0)
+    
+    # Отправляем
+    parse_mode = 'HTML' if args.html else None
+    broadcast(text, parse_mode)
+
+
+if __name__ == '__main__':
+    main()
+