310 lines
11 KiB
Python
310 lines
11 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
导出布局脚本
|
||
此脚本可以独立运行，用于导出布局，不依赖于PyQt6线程
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
import json
|
||
import arcpy
|
||
import argparse
|
||
|
||
|
||
def log(message):
|
||
"""日志输出函数"""
|
||
print(message)
|
||
sys.stdout.flush() # 确保立即输出
|
||
|
||
def parse_arguments():
|
||
"""解析命令行参数"""
|
||
parser = argparse.ArgumentParser(description='导出布局')
|
||
parser.add_argument('--mode', choices=['single', 'batch'], default='single', help='导出模式')
|
||
parser.add_argument('--aprx_file_list', help='ArcGIS Pro工程文件路径')
|
||
parser.add_argument('--input_aprx_folder', help='批量模式下的工程文件夹路径')
|
||
parser.add_argument('--output_image_path', required=True, help='输出路径')
|
||
parser.add_argument('--export_format', default='PDF', help='导出格式')
|
||
parser.add_argument('--resolution', type=int, default=300, help='分辨率(DPI)')
|
||
parser.add_argument('--use_multiprocessing', action='store_true', help='是否使用多进程')
|
||
parser.add_argument('--process_count', type=int, default=2, help='进程数')
|
||
parser.add_argument('--image_force_regenerate', help='输出文件名')
|
||
|
||
args = parser.parse_args()
|
||
|
||
# 处理图层列表参数（从JSON字符串转换为列表）
|
||
try:
|
||
# 尝试将字符串解析为JSON
|
||
if args.aprx_file_list.startswith('[') and args.aprx_file_list.endswith(']'):
|
||
args.aprx_file_list = json.loads(args.aprx_file_list)
|
||
else:
|
||
# 如果不是JSON格式，则假定是单个图层或逗号分隔的列表
|
||
if ',' in args.aprx_file_list:
|
||
cleaned = args.aprx_file_list.strip("[]")
|
||
args.aprx_file_list = [aprx_file_list.strip() for aprx_file_list in cleaned.split(',')]
|
||
else:
|
||
args.aprx_file_list = [args.aprx_file_list]
|
||
except json.JSONDecodeError:
|
||
args.aprx_file_list = [args.aprx_file_list]
|
||
|
||
return args
|
||
|
||
def get_file_extension(format_name):
|
||
"""根据格式名称获取文件扩展名"""
|
||
format_dict = {
|
||
"PDF": ".pdf",
|
||
"PNG": ".png",
|
||
"JPG": ".jpg",
|
||
"JPEG": ".jpg",
|
||
"TIFF": ".tif",
|
||
"EPS": ".eps",
|
||
"SVG": ".svg",
|
||
"AI": ".ai"
|
||
}
|
||
|
||
return format_dict.get(format_name.upper(), ".pdf")
|
||
|
||
|
||
def export_layout(params):
|
||
"""导出布局"""
|
||
aprx = None
|
||
try:
|
||
# 获取参数
|
||
log(f"开始导出布局...")
|
||
aprx_path = params['aprx_path']
|
||
output_folder = params['output_path']
|
||
export_format = params.get('export_format', 'PDF')
|
||
resolution = params.get('resolution', 300)
|
||
output_name = params.get('output_name', '')
|
||
|
||
# 确保输出文件夹存在
|
||
if not os.path.exists(output_folder):
|
||
os.makedirs(output_folder)
|
||
|
||
# 打开地图文档
|
||
try:
|
||
log(f"打开地图文档: {aprx_path}")
|
||
aprx = arcpy.mp.ArcGISProject(aprx_path) # type: ignore
|
||
except Exception as e:
|
||
raise Exception(f"无法打开地图文档: {str(e)}")
|
||
|
||
# 获取布局
|
||
layouts = aprx.listLayouts()
|
||
if not layouts:
|
||
raise Exception("地图文档中没有布局")
|
||
|
||
# 如果未指定输出名称，则使用地图文档名称
|
||
if not output_name:
|
||
output_name = os.path.splitext(os.path.basename(aprx_path))[0]
|
||
|
||
# 获取文件扩展名
|
||
file_ext = get_file_extension(export_format)
|
||
|
||
# 导出每个布局
|
||
exported_files = []
|
||
for layout in layouts:
|
||
layout_name = layout.name
|
||
output_file = os.path.join(output_folder, f"{output_name}{file_ext}")
|
||
log(f"导出布局 {layout_name} 到 {output_file}")
|
||
|
||
try:
|
||
if export_format.upper() == "PDF":
|
||
# 导出PDF
|
||
layout.exportToPDF(output_file, resolution=resolution)
|
||
elif export_format.upper() in ["PNG", "JPG", "JPEG", "TIFF"]:
|
||
# 导出栅格图像
|
||
layout.exportToJPEG(output_file, resolution=resolution,jpeg_quality=85) if export_format.upper() in ["JPG", "JPEG"] else None
|
||
layout.exportToPNG(output_file, resolution=resolution) if export_format.upper() == "PNG" else None
|
||
layout.exportToTIFF(output_file, resolution=resolution) if export_format.upper() == "TIFF" else None
|
||
elif export_format.upper() in ["EPS", "SVG"]:
|
||
# 导出矢量图像
|
||
layout.exportToEPS(output_file, resolution=resolution) if export_format.upper() == "EPS" else None
|
||
layout.exportToSVG(output_file, resolution=resolution) if export_format.upper() == "SVG" else None
|
||
else:
|
||
# 默认导出PDF
|
||
layout.exportToPDF(output_file, resolution=resolution)
|
||
|
||
exported_files.append(output_file)
|
||
log(f"成功导出布局 {layout_name} 到 {output_file}")
|
||
except Exception as e:
|
||
log(f"导出布局 {layout_name} 失败: {str(e)}")
|
||
|
||
return {
|
||
'exported_files': exported_files,
|
||
'count': len(exported_files)
|
||
}
|
||
|
||
except Exception as e:
|
||
log(f"导出布局失败: {str(e)}")
|
||
raise
|
||
finally:
|
||
# 释放资源
|
||
if aprx:
|
||
del aprx
|
||
arcpy.management.ClearWorkspaceCache()
|
||
|
||
|
||
def batch_export_layout(params):
|
||
"""批量导出布局"""
|
||
try:
|
||
# 获取参数
|
||
log(f"开始批量导出布局...")
|
||
aprx_files = params['aprx_files']
|
||
output_folder = params['output_path']
|
||
export_format = params.get('export_format', 'PDF')
|
||
resolution = params.get('resolution', 300)
|
||
|
||
# 确保输出文件夹存在
|
||
if not os.path.exists(output_folder):
|
||
os.makedirs(output_folder)
|
||
|
||
# 记录导出结果
|
||
all_exported_files = []
|
||
success_count = 0
|
||
failed_count = 0
|
||
|
||
# 逐个处理aprx文件
|
||
for aprx_path in aprx_files:
|
||
try:
|
||
file_name = os.path.splitext(os.path.basename(aprx_path))[0]
|
||
log(f"\n处理文件: {file_name}")
|
||
|
||
# 准备参数
|
||
export_params = {
|
||
'aprx_path': aprx_path,
|
||
'output_path': output_folder,
|
||
'export_format': export_format,
|
||
'resolution': resolution,
|
||
'output_name': file_name
|
||
}
|
||
|
||
# 调用导出布局函数
|
||
result = export_layout(export_params)
|
||
all_exported_files.extend(result['exported_files'])
|
||
success_count += result['count']
|
||
|
||
log(f"文件 {file_name} 处理完成")
|
||
except Exception as e:
|
||
log(f"处理文件 {os.path.basename(aprx_path)} 失败: {str(e)}")
|
||
failed_count += 1
|
||
|
||
# 返回结果
|
||
log(f"\n批量导出完成")
|
||
log(f"成功: {success_count} 个布局")
|
||
log(f"失败: {failed_count} 个文件")
|
||
|
||
return {
|
||
'exported_files': all_exported_files,
|
||
'count': success_count,
|
||
'success_count': success_count,
|
||
'failed_count': failed_count
|
||
}
|
||
|
||
except Exception as e:
|
||
log(f"批量导出布局失败: {str(e)}")
|
||
raise
|
||
|
||
|
||
def export_worker(aprx_path, output_path, export_format, resolution):
|
||
"""子进程专用工作函数（保持最小化参数）"""
|
||
try:
|
||
# 每个子进程独立初始化ArcPy环境
|
||
import arcpy
|
||
arcpy.env.overwriteOutput = True
|
||
|
||
result = export_layout({
|
||
'aprx_path': aprx_path,
|
||
'output_path': output_path,
|
||
'export_format': export_format,
|
||
'resolution': resolution
|
||
})
|
||
return (True, aprx_path, result)
|
||
except Exception as e:
|
||
return (False, aprx_path, str(e))
|
||
|
||
|
||
def main():
|
||
"""主函数"""
|
||
|
||
try:
|
||
args = parse_arguments()
|
||
|
||
if len(args.aprx_file_list) == 1:
|
||
aprx_file = args.aprx_file_list[0]
|
||
if not os.path.exists(aprx_file):
|
||
log(f"所选文件{aprx_file}不存在，请确认")
|
||
return 1
|
||
|
||
# 准备参数
|
||
params = {
|
||
'aprx_path': aprx_file,
|
||
'output_path': args.output_image_path,
|
||
'export_format': args.export_format,
|
||
'resolution': args.resolution
|
||
}
|
||
|
||
# 调用导出函数
|
||
result = export_layout(params)
|
||
log(f"导出完成，成功导出 {result['count']} 个布局")
|
||
|
||
elif len(args.aprx_file_list) >1:
|
||
if not args.input_aprx_folder:
|
||
log("批量导出模式需要指定aprx_folder参数")
|
||
return 1
|
||
|
||
# 查找所有aprx文件
|
||
aprx_files = []
|
||
valied_files = []
|
||
failed_files = []
|
||
|
||
for file in args.aprx_file_list:
|
||
if not os.path.exists(file) and file.lower().endswith('.aprx'):
|
||
failed_files.append(os.path.basename(file))
|
||
continue
|
||
aprx_files.append(file)
|
||
valied_files.append(os.path.basename(file))
|
||
|
||
if not aprx_files:
|
||
log(f"在指定文件夹中未找到aprx文件: {args.input_aprx_folder}")
|
||
return 1
|
||
|
||
log(f"找到 {len(valied_files)} 个有效aprx文件: {', '.join(valied_files)}\n")
|
||
log(f"{len(failed_files)} 个无效文件: {', '.join(failed_files)}")
|
||
|
||
if args.use_multiprocessing and args.process_count > 1 and len(aprx_files)>1:
|
||
from multiprocessing import Pool
|
||
tasks = [(aprx_file, args.output_image_path, args.export_format, args.resolution) for aprx_file in aprx_files]
|
||
with Pool(min(int(args.process_count), len(tasks))) as p:
|
||
results = p.starmap(export_worker, tasks)
|
||
|
||
for success, aprx_path, result in results:
|
||
if success:
|
||
log(f"成功导出布局 {aprx_path} 到 {result['exported_files']}") # type: ignore
|
||
else:
|
||
log(f"导出布局 {aprx_path} 失败: {result}")
|
||
return 0
|
||
|
||
else:
|
||
# 准备参数
|
||
params = {
|
||
'aprx_files': aprx_files,
|
||
'output_path': args.output_image_path,
|
||
'export_format': args.export_format,
|
||
'resolution': args.resolution
|
||
}
|
||
|
||
# 调用批量导出函数
|
||
result = batch_export_layout(params)
|
||
log(f"批量导出完成，成功导出 {result['count']} 个布局")
|
||
return 0
|
||
else:
|
||
log("请选择要处理的aprx文件")
|
||
return 0
|
||
except Exception as e:
|
||
log(f"导出失败: {str(e)}")
|
||
return 1
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main()) |