#!/usr/bin/env python
"""
COCO 导出测试脚本
用于测试 COCO 导出功能并打印详细信息
"""
import os
import sys

# 设置环境变量（与 start.sh 保持一致）
os.environ.setdefault('LABEL_STUDIO_BASE_DATA_DIR', '/data/label-studio-data')
os.environ.setdefault('TZ', 'Asia/Shanghai')

# 添加项目路径 - 需要添加 label_studio 目录
sys.path.insert(0, '/data/label-studio/label_studio')
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'core.settings.label_studio')

import django
django.setup()

from django.conf import settings
from projects.models import Project
from data_export.models import DataExport
from data_export.serializers import ExportDataSerializer

def test_coco_export():
    """测试 COCO 导出功能"""
    print("=" * 80)
    print("COCO 导出测试脚本")
    print("=" * 80)
    
    # 1. 检查配置
    print("\n[1] 检查配置:")
    print(f"MEDIA_ROOT: {settings.MEDIA_ROOT}")
    print(f"UPLOAD_DIR: {settings.UPLOAD_DIR}")
    print(f"完整上传路径: {os.path.join(settings.MEDIA_ROOT, settings.UPLOAD_DIR)}")
    print(f"路径是否存在: {os.path.exists(os.path.join(settings.MEDIA_ROOT, settings.UPLOAD_DIR))}")
    print(f"CONVERTER_DOWNLOAD_RESOURCES 默认值: {settings.CONVERTER_DOWNLOAD_RESOURCES}")
    
    upload_dir = os.path.join(settings.MEDIA_ROOT, settings.UPLOAD_DIR)
    if os.path.exists(upload_dir):
        print(f"上传目录内容: {os.listdir(upload_dir)}")
        # 检查项目1的文件夹
        project1_dir = os.path.join(upload_dir, '1')
        if os.path.exists(project1_dir):
            files = os.listdir(project1_dir)
            print(f"项目1的文件数量: {len(files)}")
            if files:
                print(f"示例文件: {files[0]}")
    
    # 2. 检查项目
    print("\n[2] 检查项目:")
    try:
        project = Project.objects.get(id=1)
        print(f"项目 ID: {project.id}")
        print(f"项目标题: {project.title}")
        print(f"任务数量: {project.tasks.count()}")
        print(f"标注数量: {sum(t.annotations.count() for t in project.tasks.all())}")
    except Project.DoesNotExist:
        print("错误：项目1不存在")
        return
    
    # 3. 获取任务数据
    print("\n[3] 获取任务数据:")
    tasks = project.tasks.all()
    if tasks.exists():
        first_task = tasks.first()
        print(f"第一个任务 ID: {first_task.id}")
        print(f"任务数据: {first_task.data}")
        
        # 检查图片路径
        if 'image' in first_task.data:
            image_path = first_task.data['image']
            print(f"图片路径（在数据中）: {image_path}")
            
            # 尝试解析路径
            if image_path.startswith('/data/'):
                # 移除 /data/ 前缀
                relative_path = image_path.replace('/data/', '')
                full_path = os.path.join(settings.MEDIA_ROOT, relative_path)
                print(f"完整文件路径: {full_path}")
                print(f"文件是否存在: {os.path.exists(full_path)}")
                if os.path.exists(full_path):
                    print(f"文件大小: {os.path.getsize(full_path)} 字节")
    else:
        print("警告：项目中没有任务")
        return
    
    # 4. 测试导出
    print("\n[4] 测试导出:")
    print("正在序列化任务数据...")
    task_data = ExportDataSerializer(tasks, many=True).data
    print(f"序列化的任务数量: {len(task_data)}")
    
    if task_data:
        print("\n第一个任务的数据结构:")
        first_task_keys = list(task_data[0].keys())
        print(f"字段: {', '.join(first_task_keys)}")
        
        if 'data' in task_data[0]:
            print(f"data 字段内容: {task_data[0]['data']}")
    
    # 5. 检查 label-studio-sdk 版本
    print("\n[5] 检查依赖:")
    try:
        import label_studio_sdk
        print(f"label-studio-sdk 版本: {label_studio_sdk.__version__}")
    except:
        print("警告：无法导入 label_studio_sdk")
    
    print("\n" + "=" * 80)
    print("测试完成")
    print("=" * 80)

if __name__ == '__main__':
    test_coco_export()