migrate.py

import os
import json
import sqlite3
import time

CACHE_DIR = "cache"
USERS_FILE = "users.json"
CACHED_INDEX_FILE = "cached.json" # Старый индекс ссылок
DB_PATH = "gsmarbot.db"

def get_connection():
    return sqlite3.connect(DB_PATH)

def migrate_devices(conn):
    print(f"\n📱 --- Миграция Устройств ---")
    cursor = conn.cursor()

    # Создаем таблицу, если нет
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS devices (
            id TEXT PRIMARY KEY,
            url TEXT UNIQUE,
            title TEXT,
            data TEXT,
            updated_at INTEGER
        )
    ''')
    cursor.execute('CREATE INDEX IF NOT EXISTS idx_url ON devices (url)')
    cursor.execute('CREATE INDEX IF NOT EXISTS idx_title ON devices (title)')

    # Читаем файлы
    if not os.path.exists(CACHE_DIR):
        print("❌ Папка cache не найдена.")
        return

    files = [f for f in os.listdir(CACHE_DIR) if f.endswith('.json')]
    total = len(files)
    print(f"📂 Найдено файлов устройств: {total}")

    count = 0
    skipped = 0

    # Также загрузим старый cached.json для сверки (опционально)
    old_index = {}
    if os.path.exists(CACHED_INDEX_FILE):
        try:
            with open(CACHED_INDEX_FILE, 'r') as f:
                old_index = json.load(f)
            print(f"ℹ️ Загружен старый индекс cached.json: {len(old_index)} ссылок.")
        except: pass

    for filename in files:
        filepath = os.path.join(CACHE_DIR, filename)
        try:
            with open(filepath, 'r', encoding='utf-8') as f:
                content = f.read().strip()
                if not content: continue
                data = json.loads(content)

            if 'id' not in data or 'link' not in data:
                skipped += 1
                continue

            # Генерация descr, если нет (для старых записей)
            if 'descr' not in data:
                res = data.get('screen', '').split(',')[0].split('(')[0].strip()
                cpu = data.get('CPU', 'CPU not specified')
                data['descr'] = f"{res} | {cpu}"

            json_dump = json.dumps(data, ensure_ascii=False)
            updated_at = int(data.get('infodate', time.time()))

            cursor.execute('''
                INSERT OR REPLACE INTO devices (id, url, title, data, updated_at)
                VALUES (?, ?, ?, ?, ?)
            ''', (data['id'], data['link'], data.get('title', 'Unknown'), json_dump, updated_at))

            count += 1
            if count % 1000 == 0: print(f"   Processed {count} devices...")

        except Exception:
            skipped += 1

    conn.commit()
    print(f"✅ Устройства перенесены: {count}. Пропущено: {skipped}")

def migrate_users(conn):
    print(f"\n👤 --- Миграция Пользователей ---")
    cursor = conn.cursor()

    cursor.execute('''
        CREATE TABLE IF NOT EXISTS user_settings (
            user_id INTEGER PRIMARY KEY,
            settings TEXT
        )
    ''')

    if not os.path.exists(USERS_FILE):
        print("ℹ️ Файл users.json не найден, пропускаем.")
        return

    try:
        with open(USERS_FILE, 'r', encoding='utf-8') as f:
            users_data = json.load(f)

        total_users = len(users_data)
        print(f"📂 Найдено пользователей: {total_users}")

        count = 0
        for uid_str, settings_list in users_data.items():
            try:
                user_id = int(uid_str)
                # settings_list это массив [1, 0, None...]
                # Сохраняем его как JSON строку в базу
                settings_json = json.dumps(settings_list)

                cursor.execute('''
                    INSERT OR REPLACE INTO user_settings (user_id, settings)
                    VALUES (?, ?)
                ''', (user_id, settings_json))
                count += 1
            except:
                pass

        conn.commit()
        print(f"✅ Пользователи перенесены: {count}")

    except Exception as e:
        print(f"❌ Ошибка чтения users.json: {e}")

def main():
    conn = get_connection()
    migrate_devices(conn)
    migrate_users(conn)
    conn.close()
    print("\n🎉 ВСЯ МИГРАЦИЯ ЗАВЕРШЕНА!")

if __name__ == "__main__":
    main()