#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Скрипт для миграции существующих пользователей из users.json в базу данных аналитики
"""

import json
import os
from bot_analytics import BotAnalytics

def migrate_users():
    """Миграция пользователей из users.json в базу данных аналитики"""
    
    # Путь к файлу пользователей
    users_file = "/var/www/u0236315/data/www/consultsolution.ru/users.json"
    
    # Проверяем существование файла
    if not os.path.exists(users_file):
        print(f"❌ Файл {users_file} не найден")
        return
    
    # Загружаем пользователей
    try:
        with open(users_file, 'r', encoding='utf-8') as f:
            users_data = json.load(f)
        
        users = users_data.get('users', [])
        print(f"📊 Найдено пользователей в users.json: {len(users)}")
        
        # Инициализируем аналитику
        analytics = BotAnalytics(db_path="/var/www/u0236315/data/www/consultsolution.ru/bot_illustrator.db")
        
        # Мигрируем каждого пользователя
        migrated = 0
        for user_id in users:
            try:
                analytics.register_user(user_id)
                migrated += 1
                print(f"✅ Пользователь {user_id} мигрирован")
            except Exception as e:
                print(f"❌ Ошибка миграции пользователя {user_id}: {e}")
        
        print(f"\n📈 Миграция завершена!")
        print(f"✅ Успешно мигрировано: {migrated}/{len(users)}")
        
    except Exception as e:
        print(f"❌ Ошибка при миграции: {e}")

if __name__ == "__main__":
    print("🔄 Начинаем миграцию пользователей в базу данных аналитики...")
    migrate_users()