#!/usr/bin/python
# -*- coding: UTF-8 -*-
import arcpy
import sys
import os  # 新增：用于文件路径操作和断点记录
import datetime
from arcpy import env
import math

# -------------------------- 1. 断点工具函数（新增）--------------------------
def init_breakpoint_file(breakpoint_path):
    """初始化断点文件：若不存在则创建空文件"""
    if not os.path.exists(breakpoint_path):
        with open(breakpoint_path, 'w', encoding='utf-8') as f:
            f.write("")  # 创建空文件
        print(f"初始化断点文件：{breakpoint_path}")

def get_completed_dkbms(breakpoint_path):
    """读取断点文件，返回已完成的DKBM列表"""
    completed = []
    if os.path.exists(breakpoint_path):
        with open(breakpoint_path, 'r', encoding='utf-8') as f:
            lines = f.readlines()
            completed = [line.strip() for line in lines if line.strip()]  # 去空行和空格
    print(f"已完成导出的DKBM数量：{len(completed)}")
    return completed

def add_completed_dkbm(breakpoint_path, dkbm):
    """将已完成的DKBM写入断点文件（避免重复）"""
    completed = get_completed_dkbms(breakpoint_path)
    if dkbm not in completed:
        with open(breakpoint_path, 'a', encoding='utf-8') as f:
            f.write(f"{dkbm}\n")  # 每行存一个DKBM，便于后续读取
        print(f"记录完成：DKBM={dkbm}")

# -------------------------- 2. 原功能函数保留（修复部分问题）--------------------------
def mkdir(path):             
    path = path.strip().rstrip("\\")
    if not os.path.exists(path):
        os.makedirs(path) 
        print(f"{path} 创建成功")
        return True
    else:
        print(f"{path} 已存在")
        return False

# -------------------------- 3. 主逻辑（新增断点续跑）--------------------------
if __name__ == "__main__":
    # -------------------------- 基础参数配置 --------------------------
    resolution = 10
    scale = 96 * 1000 * resolution / 25.4
    imgWidth = 300    # 导出图片宽度
    imgHeight = 200   # 导出图片高度
    selectfield = "DKBM"  # 用于筛选和命名的字段
    label_field = "ZW"    # 标注字段

    # 文件路径配置（根据实际情况修改）
    mxdfile = r"E:\gzct\gzct.mxd"       # MXD文件路径
    outdir = r"E:\gzct"                 # 导出JPG的输出目录
    shpname = r"E:\gzct\dk6209222021_241106_2.shp"  # 要素类路径
    breakpoint_file = os.path.join(outdir, "breakpoint.txt")  # 断点记录文件（放在输出目录下）

    # -------------------------- 初始化：创建输出目录和断点文件 --------------------------
    mkdir(outdir)  # 确保输出目录存在
    init_breakpoint_file(breakpoint_file)  # 初始化断点文件
    completed_dkbms = get_completed_dkbms(breakpoint_file)  # 读取已完成的DKBM

    # -------------------------- 加载MXD和图层 --------------------------
    try:
        mxd = arcpy.mapping.MapDocument(mxdfile)
        df = arcpy.mapping.ListDataFrames(mxd)[0]  # 获取第一个数据框
        input_layer = arcpy.mapping.ListLayers(mxd, "", df)[0]  # 获取目标图层
        target_layer = input_layer  # 标注用图层（与目标图层一致）
        print(f"成功加载MXD：{mxdfile}")
        print(f"目标图层：{input_layer.name}")
    except Exception as e:
        print(f"加载MXD/图层失败：{str(e)}")
        sys.exit(1)  # 加载失败则退出

    # -------------------------- 断点续跑：循环处理要素 --------------------------
    total_count = 0  # 总要素数
    processed_count = 0  # 本次处理的要素数
    skipped_count = 0  # 跳过的已完成要素数

    # 先统计总要素数（可选，用于进度显示）
    with arcpy.da.SearchCursor(shpname, [selectfield]) as count_cursor:
        total_count = sum(1 for _ in count_cursor)
    print(f"要素类总要素数：{total_count}")

    # 核心：循环处理每个要素（跳过已完成项）
    with arcpy.da.SearchCursor(shpname, [selectfield, label_field]) as cursor:
        for row in cursor:
            current_dkbm = str(row[0]).strip()  # 当前要素的DKBM
            current_zw = str(row[1]).strip()     # 当前要素的标注文本

            # 1. 断点判断：若已完成，直接跳过
            if current_dkbm in completed_dkbms:
                skipped_count += 1
                print(f"跳过已完成：DKBM={current_dkbm}（累计跳过：{skipped_count}）")
                continue

            # 2. 处理未完成的要素：筛选、标注、导出
            try:
                # 2.1 按DKBM筛选要素
                where_clause = f'"{selectfield}" = \'{current_dkbm}\''  # SQL筛选条件
                arcpy.SelectLayerByAttribute_management(
                    in_layer_or_view=input_layer,
                    selection_type="NEW_SELECTION",
                    where_clause=where_clause
                )

                # 2.2 检查是否有选中要素
                selected_count = int(arcpy.GetCount_management(input_layer).getOutput(0))
                if selected_count == 0:
                    print(f"警告：DKBM={current_dkbm} 无匹配要素，跳过")
                    continue

                # 2.3 缩放至选中要素（可调整缩放比例）
                df.zoomToSelectedFeatures()
                df.scale = df.scale * 1.1  # 扩大10%视野（避免要素贴边）
                arcpy.RefreshActiveView()  # 刷新视图

                # 2.4 设置标注（仅显示当前选中要素的标注）
                if target_layer.supports("LABELCLASSES"):
                    lbl_class = target_layer.labelClasses[0]
                    lbl_class.expression = f"[{label_field}]"  # 标注字段
                    lbl_class.SQLQuery = where_clause  # 仅标注当前选中要素
                    target_layer.showLabels = True
                    arcpy.RefreshActiveView()  # 刷新标注

                # 2.5 导出JPG
                out_file = os.path.join(outdir, f"{current_dkbm}.jpg")
                arcpy.mapping.ExportToJPEG(
                    map_document=mxd,
                    out_jpeg=out_file,
                    data_frame=df,
                    width=imgWidth,
                    height=imgHeight,
                    resolution=90  # 图片分辨率
                )
                print(f"导出成功：{out_file}")

                # 2.6 记录断点：只有导出成功才写入记录
                add_completed_dkbm(breakpoint_file, current_dkbm)
                processed_count += 1

            except Exception as e:
                # 异常处理：记录错误但不终止脚本（继续处理下一个）
                error_msg = f"处理DKBM={current_dkbm}时出错：{str(e)}"
                print(error_msg)
                # 可选：将错误记录到日志文件
                with open(os.path.join(outdir, "error_log.txt"), 'a', encoding='utf-8') as err_f:
                    err_f.write(f"{datetime.datetime.now()} - {error_msg}\n")
                continue

    # -------------------------- 结束：清理资源和显示统计 --------------------------
    # 清理ArcGIS资源
    arcpy.SelectLayerByAttribute_management(input_layer, "CLEAR_SELECTION")  # 清除选择
    del mxd, df, input_layer, target_layer
    arcpy.RefreshActiveView()
    arcpy.RefreshTOC()

    # 显示最终统计
    print("\n" + "="*50)
    print("断点续跑任务完成！")
    print(f"总要素数：{total_count}")
    print(f"已完成总数（含历史）：{len(get_completed_dkbms(breakpoint_file))}")
    print(f"本次处理数：{processed_count}")
    print(f"本次跳过数：{skipped_count}")
    print(f"断点文件路径：{breakpoint_file}")
    print("="*50)

#!/usr/bin/python
# -*- coding: UTF-8 -*-
import arcpy
import sys
import os
import datetime
from arcpy import env
import math

# -------------------------- 1. 断点工具函数（核心）--------------------------
def init_breakpoint_file(breakpoint_path):
    """初始化断点文件：若不存在则创建空文件，避免后续读取报错"""
    if not os.path.exists(breakpoint_path):
        with open(breakpoint_path, 'w', encoding='utf-8') as f:
            f.write("")
        print(f"✅ 初始化断点文件：{breakpoint_path}")

def get_completed_dkbms(breakpoint_path):
    """读取断点文件，返回已完成的DKBM列表（去重、去空）"""
    completed = []
    if os.path.exists(breakpoint_path):
        with open(breakpoint_path, 'r', encoding='utf-8') as f:
            lines = f.readlines()
            completed = [line.strip() for line in lines if line.strip()]
    print(f"📊 已完成导出的DKBM数量：{len(completed)}")
    return completed

def add_completed_dkbm(breakpoint_path, dkbm):
    """将已完成的DKBM写入断点文件（避免重复写入）"""
    completed = get_completed_dkbms(breakpoint_path)
    if dkbm not in completed:
        with open(breakpoint_path, 'a', encoding='utf-8') as f:
            f.write(f"{dkbm}\n")
        print(f"📝 记录完成：DKBM={dkbm}")

# -------------------------- 2. 辅助工具函数 --------------------------
def mkdir(path):
    """创建目录（若不存在）"""
    path = path.strip().rstrip("\\")
    if not os.path.exists(path):
        os.makedirs(path)
        print(f"📁 目录创建成功：{path}")
        return True
    else:
        print(f"ℹ️  目录已存在：{path}")
        return False

# -------------------------- 3. 主逻辑（断点续跑核心）--------------------------
if __name__ == "__main__":
    # ====================== 1. 基础参数配置（需根据实际情况修改）======================
    img_width = 300       # 导出JPG宽度（像素）
    img_height = 200      # 导出JPG高度（像素）
    img_resolution = 90   # 导出JPG分辨率（DPI）
    scale_expand = 1.1    # 缩放系数（缩放至要素后扩大10%，避免要素贴边）
    select_field = "DKBM" # 用于筛选要素和命名文件的字段
    label_field = "ZW"    # 用于地图标注的字段

    # 文件路径配置（Windows路径需加r转义，避免反斜杠问题）
    mxd_path = r"E:\gzct\gzct.mxd"                 # MXD地图文档路径
    out_dir = r"E:\gzct\export_jpg"                # JPG导出目录
    shp_path = r"E:\gzct\dk6209222021_241106_2.shp"# 矢量要素类路径
    breakpoint_path = os.path.join(out_dir, "breakpoint.txt") # 断点文件路径
    error_log_path = os.path.join(out_dir, "error_log.txt")   # 错误日志路径

    # ====================== 2. 初始化：目录与断点文件 ======================
    mkdir(out_dir)                # 确保导出目录存在
    init_breakpoint_file(breakpoint_path)  # 初始化断点文件
    completed_dkbms = get_completed_dkbms(breakpoint_path)  # 读取已完成列表

    # ====================== 3. 加载ArcGIS资源（MXD、图层、数据框）======================
    try:
        # 加载MXD文档
        mxd = arcpy.mapping.MapDocument(mxd_path)
        # 获取第一个数据框（若有多个，需指定名称，如ListDataFrames(mxd, "数据框1")）
        df = arcpy.mapping.ListDataFrames(mxd)[0]
        # 获取目标图层（若有多个，需指定名称，如ListLayers(mxd, "地块图层", df)[0]）
        target_layer = arcpy.mapping.ListLayers(mxd, "", df)[0]
        
        print(f"✅ 成功加载资源：")
        print(f"   - MXD文档：{mxd_path}")
        print(f"   - 数据框：{df.name}")
        print(f"   - 目标图层：{target_layer.name}")
    except Exception as e:
        print(f"❌ 加载ArcGIS资源失败：{str(e)}")
        sys.exit(1)  # 资源加载失败，直接退出脚本

    # ====================== 4. 统计总要素数（可选，用于进度展示）======================
    try:
        with arcpy.da.SearchCursor(shp_path, [select_field]) as count_cursor:
            total_features = sum(1 for _ in count_cursor)  # 统计总要素数
        print(f"📊 要素类统计：总要素数={total_features}，已完成数={len(completed_dkbms)}")
    except Exception as e:
        print(f"⚠️  统计总要素数失败：{str(e)}，将跳过统计")
        total_features = "未知"

    # ====================== 5. 断点续跑：批量处理要素 ======================
    processed_count = 0  # 本次处理的未完成要素数
    skipped_count = 0    # 本次跳过的已完成要素数

    # 使用da.SearchCursor高效遍历要素（比旧版Cursor更快、更稳定）
    with arcpy.da.SearchCursor(shp_path, [select_field, label_field]) as cursor:
        for row in cursor:
            current_dkbm = str(row[0]).strip()  # 当前要素的DKBM（去空格，避免匹配问题）
            current_zw = str(row[1]).strip()     # 当前要素的标注文本

            # -------------------------- 5.1 断点判断：跳过已完成要素 --------------------------
            if current_dkbm in completed_dkbms:
                skipped_count += 1
                print(f"ℹ️  跳过已完成：DKBM={current_dkbm}（累计跳过：{skipped_count}）")
                continue

            # -------------------------- 5.2 处理未完成要素：筛选→标注→导出 --------------------------
            try:
                # 5.2.1 按DKBM筛选要素（构建SQL条件，注意字段名加双引号，值加单引号）
                where_clause = f'"{select_field}" = \'{current_dkbm}\''
                arcpy.SelectLayerByAttribute_management(
                    in_layer_or_view=target_layer,
                    selection_type="NEW_SELECTION",  # 新建选择集
                    where_clause=where_clause
                )

                # 5.2.2 检查是否有匹配的选中要素
                selected_count = int(arcpy.GetCount_management(target_layer).getOutput(0))
                if selected_count == 0:
                    print(f"⚠️  无匹配要素：DKBM={current_dkbm}，跳过")
                    continue

                # 5.2.3 缩放至选中要素（并扩大视野，避免要素贴边）
                df.zoomToSelectedFeatures()  # 缩放至选中要素范围
                df.scale = df.scale * scale_expand  # 扩大视野（可根据需求调整系数）
                arcpy.RefreshActiveView()  # 刷新视图，确保缩放生效

                # 5.2.4 配置标注（仅显示当前选中要素的标注，避免干扰）
                if target_layer.supports("LABELCLASSES"):  # 检查图层是否支持标注类
                    lbl_class = target_layer.labelClasses[0]  # 获取第一个标注类
                    lbl_class.expression = f"[{label_field}]"  # 设置标注字段（如[ZW]）
                    lbl_class.SQLQuery = where_clause  # 仅标注当前选中要素（关键！）
                    target_layer.showLabels = True  # 启用标注
                    arcpy.RefreshActiveView()  # 刷新标注，确保生效
                else:
                    print(f"⚠️  图层{target_layer.name}不支持标注类，跳过标注设置")

                # 5.2.5 导出JPG地图
                jpg_name = f"{current_dkbm}.jpg"  # 以DKBM命名JPG文件
                jpg_path = os.path.join(out_dir, jpg_name)  # 完整JPG路径

                # 调用ExportToJPEG函数导出
                arcpy.mapping.ExportToJPEG(
                    map_document=mxd,
                    out_jpeg=jpg_path,
                    data_frame=df,          # 导出的数据框
                    width=img_width,        # 图片宽度
                    height=img_height,      # 图片高度
                    resolution=img_resolution,  # 图片分辨率
                    world_file=False        # 是否生成世界文件（.jgw），按需开启
                )
                print(f"✅ 导出成功：{jpg_path}（本次处理：{processed_count + 1}）")

                # 5.2.6 记录断点：仅当导出成功时，才将DKBM写入断点文件
                add_completed_dkbm(breakpoint_path, current_dkbm)
                processed_count += 1

            # -------------------------- 5.3 异常处理：单个要素失败不影响整体 --------------------------
            except Exception as e:
                error_msg = f"处理DKBM={current_dkbm}时出错：{str(e)}"
                print(f"❌ {error_msg}")
                # 将错误信息写入日志文件，便于后续排查
                with open(error_log_path, 'a', encoding='utf-8') as err_f:
                    err_f.write(f"[{datetime.datetime.now()}] {error_msg}\n")
                continue  # 跳过当前错误要素，继续处理下一个

    # ====================== 6. 清理资源与输出最终统计 ======================
    # 清理ArcGIS资源（避免内存泄漏）
    arcpy.SelectLayerByAttribute_management(target_layer, "CLEAR_SELECTION")  # 清除选择集
    del mxd, df, target_layer  # 删除MXD、数据框、图层对象
    arcpy.RefreshActiveView()  # 刷新视图
    arcpy.RefreshTOC()         # 刷新目录表

    # 输出最终统计结果
    print("\n" + "="*60)
    print("🎯 断点续跑任务完成！")
    print(f"📊 统计汇总：")
    print(f"   - 总要素数：{total_features}")
    print(f"   - 已完成总数（含历史）：{len(get_completed_dkbms(breakpoint_path))}")
    print(f"   - 本次处理数：{processed_count}")
    print(f"   - 本次跳过数：{skipped_count}")
    print(f"📁 关键文件路径：")
    print(f"   - 断点文件：{breakpoint_path}")
    print(f"   - 错误日志：{error_log_path}")
    print("="*60)