chatterbot/test_bot_loop.py at main · sunnuls/chatterbot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
"""
Test script for bot_loop with simulate mode
Usage: python test_bot_loop.py
"""

import sys
import threading
import time

# Настройка кодировки для Windows
if sys.platform == 'win32':
    sys.stdout.reconfigure(encoding='utf-8')

from scraper import bot_loop, stop_bot_loop
from auth import FanslyAuth

def test_bot_loop_simulate():
    """Тест bot_loop в simulate mode"""
    print("=" * 60)
    print("Testing bot_loop in simulate mode")
    print("=" * 60)

    # Создаем mock auth instance (не требуется для simulate mode)
    auth_instance = None

    # Style для генерации ответов
    style_desc = "flirty with 😘"

    # Stop event для управления циклом
    stop_event = threading.Event()

    print("\n🚀 Запуск bot_loop в simulate mode...")
    print("💡 Введите тестовые сообщения в формате: chat_id|message_text")
    print("   Пример: chat_123|hey gorgeous")
    print("   Введите 'stop' для остановки\n")

    # Запускаем bot_loop в отдельном потоке
    bot_thread = threading.Thread(
        target=bot_loop,
        args=(None, style_desc, auth_instance, None, True, None, stop_event),
        daemon=True
    )

    bot_thread.start()

    try:
        # Ждем завершения потока
        bot_thread.join()
    except KeyboardInterrupt:
        print("\n⚠️ Прерывание теста...")
        stop_event.set()
        stop_bot_loop()

    print("\n✅ Тест завершен!")

if __name__ == "__main__":
    test_bot_loop_simulate()