#!/usr/bin/env python3
import json

def test_data_import():
    print("🔍 测试数据导入情况...")
    
    # 检查主数据文件
    try:
        with open('/root/book-manager/data/chinese_books.json', 'r', encoding='utf-8') as f:
            main_books = json.load(f)
        print(f"✅ 主数据库: {len(main_books)} 本书籍")
    except Exception as e:
        print(f"❌ 主数据库读取失败: {e}")
        return
    
    # 检查部分恢复数据文件
    try:
        partial_count = 0
        with open('/root/book-manager/data/partial_recovery.jsonl', 'r', encoding='utf-8') as f:
            for line in f:
                if line.strip():
                    partial_count += 1
        print(f"✅ 部分恢复数据: {partial_count} 条记录")
    except Exception as e:
        print(f"❌ 部分恢复数据读取失败: {e}")
        return
    
    print(f"📊 总计可用数据: {len(main_books)} + {partial_count} = {len(main_books) + partial_count}")
    print("✅ 数据导入成功！")

if __name__ == "__main__":
    test_data_import()