#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
测试脚本
用于测试ArcGIS Pro图层属性和标注功能
"""

import os
import sys
import argparse
import traceback
import arcpy


def log(message):
    """日志输出函数"""
    print(message)


def test_arcgis_environment():
    """测试ArcGIS Pro环境"""
    log("=== ArcGIS Pro环境信息 ===")
    log(f"Python版本: {sys.version}")
    log(f"Python路径: {sys.executable}")
    log(f"当前工作目录: {os.getcwd()}")
    log(f"ArcPy版本: {arcpy.GetInstallInfo()['Version']}")
    log(f"ArcPy产品: {arcpy.GetInstallInfo()['ProductName']}")
    
    # 当前环境设置
    log("\n=== ArcGIS环境设置 ===")
    log(f"工作空间: {arcpy.env.workspace}")
    log(f"输出坐标系统: {arcpy.env.outputCoordinateSystem}")
    log(f"覆盖输出: {arcpy.env.overwriteOutput}")


def test_layer_properties(layer_path):
    """测试图层属性"""
    log(f"\n=== 图层属性测试: {layer_path} ===")
    
    if not arcpy.Exists(layer_path):
        log(f"错误: 图层不存在 - {layer_path}")
        return
    
    try:
        # 创建图层对象
        log("尝试创建图层对象...")
        layer = arcpy.mp.Layer(layer_path)
        log(f"成功创建图层: {layer.name}")
        
        # 图层基本属性
        log("\n图层基本属性:")
        log(f"名称: {layer.name}")
        log(f"数据源类型: {type(layer.dataSource).__name__ if hasattr(layer, 'dataSource') else 'N/A'}")
        log(f"长名称: {layer.longName if hasattr(layer, 'longName') else 'N/A'}")
        
        # 图层类型
        log("\n图层类型判断:")
        log(f"是要素图层: {layer.isFeatureLayer if hasattr(layer, 'isFeatureLayer') else 'N/A'}")
        log(f"是栅格图层: {layer.isRasterLayer if hasattr(layer, 'isRasterLayer') else 'N/A'}")
        log(f"是图形图层: {layer.isGroupLayer if hasattr(layer, 'isGroupLayer') else 'N/A'}")
        
        # 支持的属性
        log("\n支持的属性:")
        properties = [
            'LABELCLASSES', 'SHOWLABELS', 'NAME', 'DATASOURCE', 'DEFINITIONQUERY',
            'VISIBLE', 'TRANSPARENCY', 'BRIGHTNESS', 'CONTRAST', 'SYMBOLOGY'
        ]
        
        for prop in properties:
            try:
                support = hasattr(layer, prop.lower()) or (hasattr(layer, 'supports') and layer.supports(prop))
                log(f"{prop}: {'支持' if support else '不支持'}")
            except Exception as e:
                log(f"{prop}: 检查失败 - {str(e)}")
        
        # 尝试获取标注类
        log("\n标注类测试:")
        try:
            if hasattr(layer, 'listLabelClasses'):
                label_classes = layer.listLabelClasses()
                log(f"找到 {len(label_classes)} 个标注类")
                
                # 显示每个标注类的信息
                for i, lc in enumerate(label_classes):
                    log(f"标注类 #{i+1}: {lc.name if hasattr(lc, 'name') else 'N/A'}")
                    log(f"  表达式: {lc.expression if hasattr(lc, 'expression') else 'N/A'}")
                    log(f"  SQL查询: {lc.SQLQuery if hasattr(lc, 'SQLQuery') else 'N/A'}")
                    log(f"  可见: {lc.showClassLabels if hasattr(lc, 'showClassLabels') else 'N/A'}")
            else:
                log("图层不支持listLabelClasses方法")
                
            # 检查标注状态
            if hasattr(layer, 'showLabels'):
                log(f"标注显示状态: {layer.showLabels}")
            else:
                log("图层没有showLabels属性")
        except Exception as e:
            log(f"获取标注类时出错: {str(e)}")
            
        # 使用arcpy.da.Describe获取详细信息
        log("\narcpy.da.Describe描述信息:")
        try:
            desc = arcpy.da.Describe(layer)
            for key in desc:
                # 跳过复杂对象
                if isinstance(desc[key], (dict, list, tuple)):
                    log(f"{key}: [复杂类型]")
                else:
                    log(f"{key}: {desc[key]}")
        except Exception as e:
            log(f"获取Describe信息时出错: {str(e)}")
            
    except Exception as e:
        log(f"测试图层属性时出错: {str(e)}")
        log(traceback.format_exc())


def test_annotation_conversion(map_path, layer_name, output_folder):
    """测试标注转注记功能"""
    log(f"\n=== 标注转注记测试: {map_path}, 图层: {layer_name} ===")
    
    if not arcpy.Exists(map_path):
        log(f"错误: 地图文档不存在 - {map_path}")
        return
        
    # 确保输出文件夹存在
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)
    
    try:
        # 打开地图文档
        log(f"打开地图文档...")
        aprx = arcpy.mp.ArcGISProject(map_path)
        
        # 获取活动地图
        log(f"获取地图...")
        maps = aprx.listMaps()
        if not maps:
            log("错误: 地图文档中没有地图")
            return
            
        target_map = maps[0]  # 默认使用第一个地图
        log(f"使用地图: {target_map.name}")
        
        # 查找指定图层
        log(f"查找图层: {layer_name}")
        layers = target_map.listLayers(layer_name)
        if not layers:
            log(f"错误: 找不到图层 {layer_name}")
            return
            
        label_layer = layers[0]
        log(f"找到图层: {label_layer.name}")
        
        # 测试图层属性
        test_layer_properties(label_layer)
        
        # 尝试开启标注
        try:
            if hasattr(label_layer, 'showLabels'):
                log(f"当前标注状态: {label_layer.showLabels}")
                label_layer.showLabels = True
                log(f"已开启标注，新状态: {label_layer.showLabels}")
        except Exception as e:
            log(f"设置标注状态时出错: {str(e)}")
        
        # 尝试执行标注转注记
        log("\n执行标注转注记...")
        try:
            anno_name = f"{label_layer.name}_Anno"
            output_anno = os.path.join(output_folder, anno_name)
            
            # 检查是否已存在
            if arcpy.Exists(output_anno):
                log(f"注记已存在: {output_anno}，将尝试删除")
                arcpy.management.Delete(output_anno)
                
            # 执行转换
            log("使用ConvertLabelsToAnnotation...")
            arcpy.cartography.ConvertLabelsToAnnotation(
                target_map, 
                [label_layer], 
                output_folder,
                "FEATURE_LINKED",
                "STANDARD",
                None,
                None,
                None
            )
            log("ConvertLabelsToAnnotation执行成功")
            
        except Exception as e:
            log(f"执行标注转注记时出错: {str(e)}")
            log(traceback.format_exc())
            
            # 尝试替代方案
            try:
                log("\n尝试使用LabelFeatures替代方案...")
                arcpy.cartography.LabelFeatures(
                    in_features=label_layer,
                    out_geodatabase=output_folder
                )
                log("LabelFeatures执行成功")
            except Exception as e2:
                log(f"LabelFeatures也失败: {str(e2)}")
    
    except Exception as e:
        log(f"测试标注转注记时出错: {str(e)}")
        log(traceback.format_exc())
    finally:
        if 'aprx' in locals():
            del aprx


def main():
    """主函数"""
    parser = argparse.ArgumentParser(description='ArcGIS Pro图层和标注功能测试')
    parser.add_argument('--message', default='测试消息', help='测试消息')
    parser.add_argument('--count', type=int, default=1, help='重复次数')
    parser.add_argument('--map', help='地图文档路径(.aprx)')
    parser.add_argument('--layer', help='图层名称')
    parser.add_argument('--layer_path', help='图层文件路径(.lyrx)')
    parser.add_argument('--output', help='输出文件夹')
    
    args = parser.parse_args()
    
    # 输出系统信息
    test_arcgis_environment()
    
    # 输出测试消息
    for i in range(args.count):
        log(f"\n{i+1}. {args.message}")
    
    # 如果提供了图层文件路径，测试图层属性
    if args.layer_path:
        test_layer_properties(args.layer_path)
    
    # 如果提供了地图文档、图层名称和输出文件夹，测试标注转注记
    if args.map and args.layer and args.output:
        test_annotation_conversion(args.map, args.layer, args.output)
    
    # 退出前打印完成消息
    log("\n测试脚本执行完成")
    return 0


if __name__ == "__main__":
    sys.exit(main())