#!/usr/bin/env python3
"""
测试Word COM操作，定位报告生成卡住的具体位置
"""

import json
import sys
import os
import time
from pathlib import Path

def test_basic_word_operations():
    """测试基本的Word COM操作"""
    print("测试基本Word COM操作:")
    print("=" * 40)
    
    try:
        import win32com.client
        from win32com.client import constants
        
        print("✓ 导入win32com成功")
        
        # 创建Word应用程序
        print("创建Word应用程序...")
        word_app = win32com.client.Dispatch("Word.Application")
        word_app.Visible = False  # 不显示Word窗口
        print("✓ Word应用程序创建成功")
        
        # 创建新文档
        print("创建新文档...")
        doc = word_app.Documents.Add()
        print("✓ 新文档创建成功")
        
        # 插入表格
        print("插入测试表格...")
        table = doc.Tables.Add(doc.Range(0, 0), 25, 10)  # 25行10列，比实际需要的大
        print("✓ 表格插入成功")
        
        # 测试单元格写入
        print("测试单元格写入...")
        test_cells = [
            (1, 1, "测试1"),
            (5, 3, "23.2"),
            (10, 5, "2025-11-27 11:30:00"),
            (20, 8, "478.0")
        ]
        
        for row, col, value in test_cells:
            try:
                cell = table.Cell(row, col)
                cell.Range.Text = str(value)
                print(f"  ✓ 写入单元格 ({row}, {col}): {value}")
            except Exception as e:
                print(f"  ✗ 写入单元格 ({row}, {col}) 失败: {e}")
        
        print("✓ 基本操作测试完成")
        
        # 清理
        doc.Close(SaveChanges=False)
        word_app.Quit()
        print("✓ Word应用程序已关闭")
        
        return True
        
    except Exception as e:
        print(f"✗ Word COM操作失败: {e}")
        return False

def test_template_operations():
    """测试模板操作"""
    print("\n测试Word模板操作:")
    print("=" * 40)
    
    # 查找模板文件
    template_files = []
    search_paths = [
        Path("."),
        Path("E:/docx_creatorV2"),
        Path("f:/PyPro/docx_creator")
    ]
    
    for search_path in search_paths:
        if search_path.exists():
            templates = list(search_path.glob("*.docx"))
            template_files.extend(templates)
    
    if not template_files:
        print("⚠️  未找到Word模板文件")
        return False
    
    print(f"找到模板文件: {len(template_files)} 个")
    for template in template_files[:3]:  # 只显示前3个
        print(f"  - {template}")
    
    # 测试打开模板
    try:
        import win32com.client
        
        word_app = win32com.client.Dispatch("Word.Application")
        word_app.Visible = False
        
        template_path = template_files[0]
        print(f"\n尝试打开模板: {template_path}")
        
        # 确保使用绝对路径
        abs_template_path = template_path.resolve()
        print(f"绝对路径: {abs_template_path}")
        
        if not abs_template_path.exists():
            print(f"✗ 模板文件不存在: {abs_template_path}")
            word_app.Quit()
            return False
        
        doc = word_app.Documents.Open(str(abs_template_path))
        print("✓ 模板打开成功")
        
        # 检查模板中的表格
        tables_count = doc.Tables.Count
        print(f"模板中的表格数量: {tables_count}")
        
        if tables_count > 0:
            table = doc.Tables(1)  # Word中索引从1开始
            rows_count = table.Rows.Count
            cols_count = table.Columns.Count
            print(f"第一个表格尺寸: {rows_count} 行 x {cols_count} 列")
            
            # 查找scriptTable1标记
            found_token = False
            try:
                for i in range(1, min(rows_count + 1, 10)):  # 只检查前10行
                    for j in range(1, min(cols_count + 1, 10)):  # 只检查前10列
                        try:
                            cell_text = table.Cell(i, j).Range.Text
                            if "scriptTable1" in cell_text:
                                print(f"✓ 找到scriptTable1标记在 ({i}, {j})")
                                found_token = True
                                break
                        except:
                            continue
                    if found_token:
                        break
            except Exception as e:
                print(f"搜索标记时出错: {e}")
            
            if not found_token:
                print("⚠️  未找到scriptTable1标记")
        
        doc.Close(SaveChanges=False)
        word_app.Quit()
        print("✓ 模板测试完成")
        
        return True
        
    except Exception as e:
        print(f"✗ 模板操作失败: {e}")
        try:
            word_app.Quit()
        except:
            pass
        return False

def test_large_data_processing():
    """测试大量数据处理"""
    print("\n测试大量数据处理:")
    print("=" * 40)
    
    # 模拟你的实际数据量
    test_data = []
    for row in range(21):  # 0-20行
        for col in range(7):  # 0-6列
            if (row, col) in [(1, 1), (1, 3), (0, 1)]:  # 时间和环境温度
                test_data.append({"row": row, "col": col, "value": "2025-11-27 11:30:00"})
            elif row >= 4 and row <= 20:  # 温度数据
                test_data.append({"row": row, "col": col, "value": f"{20 + row + col * 0.5:.1f}"})
    
    print(f"测试数据量: {len(test_data)} 个单元格")
    
    try:
        import win32com.client
        
        word_app = win32com.client.Dispatch("Word.Application")
        word_app.Visible = False
        
        doc = word_app.Documents.Add()
        table = doc.Tables.Add(doc.Range(0, 0), 25, 10)
        
        print("开始批量写入数据...")
        start_time = time.time()
        
        processed = 0
        for i, cell_data in enumerate(test_data):
            try:
                row = cell_data["row"] + 1  # Word索引从1开始
                col = cell_data["col"] + 1
                value = str(cell_data["value"])
                
                cell = table.Cell(row, col)
                cell.Range.Text = value
                processed += 1
                
                # 每50个单元格报告一次进度
                if (i + 1) % 50 == 0:
                    elapsed = time.time() - start_time
                    print(f"  进度: {i + 1}/{len(test_data)} ({elapsed:.1f}秒)")
                    
                    # 如果处理时间过长，提前退出
                    if elapsed > 30:  # 30秒超时
                        print("⚠️  处理时间过长，提前退出")
                        break
                        
            except Exception as e:
                print(f"  ✗ 处理单元格 ({row}, {col}) 失败: {e}")
                break
        
        total_time = time.time() - start_time
        print(f"✓ 处理完成: {processed}/{len(test_data)} 个单元格，耗时 {total_time:.1f} 秒")
        
        if total_time > 10:
            print("⚠️  处理时间较长，可能存在性能问题")
        
        doc.Close(SaveChanges=False)
        word_app.Quit()
        
        return processed == len(test_data)
        
    except Exception as e:
        print(f"✗ 大量数据处理失败: {e}")
        try:
            word_app.Quit()
        except:
            pass
        return False

def main():
    print("Word COM操作测试")
    print("=" * 50)
    
    # 测试基本操作
    basic_ok = test_basic_word_operations()
    
    # 测试模板操作
    template_ok = test_template_operations()
    
    # 测试大量数据处理
    data_ok = test_large_data_processing()
    
    print("\n" + "=" * 50)
    print("测试结果总结:")
    print(f"基本Word操作: {'✓ 通过' if basic_ok else '✗ 失败'}")
    print(f"模板操作: {'✓ 通过' if template_ok else '✗ 失败'}")
    print(f"大量数据处理: {'✓ 通过' if data_ok else '✗ 失败'}")
    
    if not all([basic_ok, template_ok, data_ok]):
        print("\n建议:")
        if not basic_ok:
            print("- 检查Word安装和COM组件")
        if not template_ok:
            print("- 检查Word模板文件是否存在和可访问")
        if not data_ok:
            print("- 可能是大量数据处理导致的性能问题")
            print("- 建议分批处理数据或优化算法")

if __name__ == "__main__":
    main()