385 lines
15 KiB
Python
385 lines
15 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
导出地图选项卡模块
|
|
"""
|
|
|
|
import os
|
|
import arcpy
|
|
import traceback
|
|
from PyQt6.QtWidgets import (QWidget, QVBoxLayout, QHBoxLayout, QFormLayout, QLabel,
|
|
QLineEdit, QPushButton, QCheckBox, QFileDialog, QMessageBox)
|
|
from PyQt6.QtCore import pyqtSignal
|
|
from ..components.file_list_group import FileListGroup
|
|
from ..runners.script_runner import ScriptRunner
|
|
|
|
|
|
class ExportMapTab(QWidget):
|
|
"""导出地图选项卡"""
|
|
|
|
# 定义日志信号，用于将日志消息传递给主窗口
|
|
log_signal = pyqtSignal(str)
|
|
|
|
def __init__(self, parent=None):
|
|
super().__init__(parent)
|
|
self.main_window = parent
|
|
self.setupUi()
|
|
self.load_settings()
|
|
|
|
def setupUi(self):
|
|
"""设置界面"""
|
|
# 主布局
|
|
self.main_layout = QVBoxLayout(self)
|
|
|
|
# 参数区域
|
|
self.form_layout = QFormLayout()
|
|
|
|
# 区县名称
|
|
self.county_name_edit = QLineEdit()
|
|
self.form_layout.addRow("区县名称:", self.county_name_edit)
|
|
|
|
# 配置文件
|
|
config_layout = QHBoxLayout()
|
|
|
|
self.config_file_edit = QLineEdit()
|
|
self.config_file_edit.setEnabled(False)
|
|
config_layout.addWidget(self.config_file_edit)
|
|
|
|
# self.config_file_browse = QPushButton("浏览...")
|
|
# self.config_file_browse.clicked.connect(self.on_browse_config_file)
|
|
# config_layout.addWidget(self.config_file_browse)
|
|
|
|
self.form_layout.addRow("配置文件:", config_layout)
|
|
|
|
# 模板文件
|
|
self.template_aprx_file_edit = QLineEdit()
|
|
self.template_path_browse = QPushButton("浏览...")
|
|
self.template_path_browse.clicked.connect(self.on_browse_template_path)
|
|
|
|
template_layout = QHBoxLayout()
|
|
template_layout.addWidget(self.template_aprx_file_edit)
|
|
template_layout.addWidget(self.template_path_browse)
|
|
self.form_layout.addRow("模板文件:", template_layout)
|
|
|
|
# 数据源路径
|
|
self.data_source_path_edit = QLineEdit()
|
|
self.data_source_path_browse = QPushButton("浏览...")
|
|
self.data_source_path_browse.clicked.connect(self.on_browse_data_source_path)
|
|
|
|
data_source_layout = QHBoxLayout()
|
|
data_source_layout.addWidget(self.data_source_path_edit)
|
|
data_source_layout.addWidget(self.data_source_path_browse)
|
|
self.form_layout.addRow("数据源路径:", data_source_layout)
|
|
|
|
# 符号系统路径
|
|
self.symbol_path_edit = QLineEdit()
|
|
self.symbol_path_browse = QPushButton("浏览...")
|
|
self.symbol_path_browse.clicked.connect(self.on_browse_symbol_path)
|
|
|
|
symbol_layout = QHBoxLayout()
|
|
symbol_layout.addWidget(self.symbol_path_edit)
|
|
symbol_layout.addWidget(self.symbol_path_browse)
|
|
self.form_layout.addRow("符号系统路径:", symbol_layout)
|
|
|
|
# 统计表格图片路径
|
|
self.pic_path_edit = QLineEdit()
|
|
self.pic_path_browse = QPushButton("浏览...")
|
|
self.pic_path_browse.clicked.connect(self.on_browse_pic_path)
|
|
|
|
pic_layout = QHBoxLayout()
|
|
pic_layout.addWidget(self.pic_path_edit)
|
|
pic_layout.addWidget(self.pic_path_browse)
|
|
self.form_layout.addRow("面积统计图片路径:", pic_layout)
|
|
|
|
# 输出路径
|
|
self.output_path_edit = QLineEdit()
|
|
self.output_path_browse = QPushButton("浏览...")
|
|
self.output_path_browse.clicked.connect(self.on_browse_output_path)
|
|
|
|
output_layout = QHBoxLayout()
|
|
output_layout.addWidget(self.output_path_edit)
|
|
output_layout.addWidget(self.output_path_browse)
|
|
self.form_layout.addRow("输出路径:", output_layout)
|
|
|
|
# 文件列表布局
|
|
self.file_list_group = FileListGroup(self, "选择要导出的要素类：")
|
|
self.file_list_group.load_files.connect(self.on_load_polygon)
|
|
|
|
# 选项区域
|
|
options_layout = QVBoxLayout()
|
|
|
|
# 强制重新生成选项
|
|
self.force_regenerate_check = QCheckBox("强制重新生成工程文件")
|
|
options_layout.addWidget(self.force_regenerate_check)
|
|
|
|
# 添加到主布局
|
|
self.main_layout.addLayout(self.form_layout)
|
|
self.main_layout.addWidget(self.file_list_group)
|
|
self.main_layout.addLayout(options_layout)
|
|
|
|
# 导出按钮
|
|
self.export_button = QPushButton("导出地图")
|
|
self.export_button.clicked.connect(self.on_export_map)
|
|
self.main_layout.addWidget(self.export_button)
|
|
|
|
# 状态标签
|
|
self.status_label = QLabel("状态: 就绪")
|
|
self.main_layout.addWidget(self.status_label)
|
|
|
|
def on_browse_config_file(self):
|
|
"""浏览配置文件"""
|
|
initial_dir = ""
|
|
if os.path.exists(self.config_file_edit.text()):
|
|
initial_dir = self.config_file_edit.text()
|
|
|
|
file_path, _ = QFileDialog.getOpenFileName(self, "选择配置文件", initial_dir, "JSON文件 (*.json);;所有文件 (*.*)")
|
|
|
|
if file_path:
|
|
self.config_file_edit.setText(file_path)
|
|
|
|
def on_browse_template_path(self):
|
|
"""浏览模板文件"""
|
|
initial_dir = ""
|
|
if os.path.exists(self.template_aprx_file_edit.text()):
|
|
initial_dir = self.template_aprx_file_edit.text()
|
|
|
|
file_path, _ = QFileDialog.getOpenFileName(self, "选择模板文件", initial_dir, "ArcGIS Pro工程文件 (*.aprx)")
|
|
|
|
if file_path:
|
|
self.template_aprx_file_edit.setText(file_path)
|
|
|
|
def on_browse_data_source_path(self):
|
|
"""浏览数据源路径"""
|
|
initial_dir = ""
|
|
if os.path.exists(self.data_source_path_edit.text()):
|
|
initial_dir = os.path.dirname(self.data_source_path_edit.text())
|
|
|
|
folder_path = QFileDialog.getExistingDirectory(self, "选择数据源文件夹", initial_dir)
|
|
if folder_path and folder_path.endswith(".gdb"):
|
|
self.data_source_path_edit.setText(folder_path)
|
|
# 自动加载图层列表
|
|
self.on_load_polygon()
|
|
else:
|
|
QMessageBox.warning(self, "错误", "请选择GDB的数据源。")
|
|
|
|
def on_browse_symbol_path(self):
|
|
"""浏览符号系统路径"""
|
|
initial_dir = ""
|
|
if os.path.exists(self.symbol_path_edit.text()):
|
|
initial_dir = os.path.dirname(self.symbol_path_edit.text())
|
|
|
|
folder_path = QFileDialog.getExistingDirectory(self, "选择符号系统文件夹", initial_dir)
|
|
if folder_path:
|
|
self.symbol_path_edit.setText(folder_path)
|
|
|
|
def on_browse_pic_path(self):
|
|
"""浏览符号系统路径"""
|
|
initial_dir = ""
|
|
if os.path.exists(self.pic_path_edit.text()):
|
|
initial_dir = os.path.dirname(self.pic_path_edit.text())
|
|
|
|
folder_path = QFileDialog.getExistingDirectory(self, "选择符号系统文件夹", initial_dir)
|
|
if folder_path:
|
|
self.pic_path_edit.setText(folder_path)
|
|
|
|
def on_browse_output_path(self):
|
|
"""浏览输出路径"""
|
|
initial_dir = ""
|
|
if os.path.exists(self.output_path_edit.text()):
|
|
initial_dir = os.path.dirname(self.output_path_edit.text())
|
|
|
|
folder_path = QFileDialog.getExistingDirectory(self, "选择输出文件夹", initial_dir)
|
|
if folder_path.endswith(".gdb"):
|
|
QMessageBox.warning(self, "警告", "请选择有效的输出路径")
|
|
return
|
|
|
|
self.output_path_edit.setText(folder_path)
|
|
|
|
def on_load_polygon(self):
|
|
"""加载图层列表"""
|
|
try:
|
|
data_source_paths = self.data_source_path_edit.text()
|
|
|
|
# 清空列表
|
|
self.file_list_group.file_list.clear()
|
|
|
|
# 添加图层
|
|
if arcpy.Exists(data_source_paths):
|
|
arcpy.env.workspace = data_source_paths
|
|
for polygon in arcpy.ListFeatureClasses(feature_type="Polygon"):
|
|
self.file_list_group.file_list.addItem(polygon)
|
|
|
|
self.log_message(f"已加载 {self.file_list_group.file_list.count()} 个图层")
|
|
|
|
except Exception as e:
|
|
self.log_error(f"加载图层列表失败: {str(e)}")
|
|
traceback.print_exc()
|
|
|
|
def get_selected_polygon(self):
|
|
"""获取选中的图层列表"""
|
|
selected_items = self.file_list_group.file_list.selectedItems()
|
|
return [item.text() for item in selected_items]
|
|
|
|
def on_export_map(self):
|
|
"""导出地图按钮点击事件"""
|
|
try:
|
|
# 验证并获取参数
|
|
params = self.validate_and_params()
|
|
if not params:
|
|
return
|
|
|
|
polygon_list = params.get('polygon_list')
|
|
|
|
# 清空日志
|
|
self.log_message("开始导出工程文件...")
|
|
self.log_message(f"选中的图层: {polygon_list}")
|
|
|
|
# 创建导出线程
|
|
self.runner = ScriptRunner()
|
|
self.runner.task_log.connect(self.on_script_log)
|
|
self.runner.task_error.connect(self.log_error)
|
|
self.runner.task_finished.connect(self.on_script_finished)
|
|
|
|
# 禁用导出按钮
|
|
self.export_button.setEnabled(False)
|
|
self.status_label.setText("状态: 正在导出...")
|
|
|
|
# 运行导出脚本（使用外部窗口）
|
|
# print(params)
|
|
self.runner.run_export_map(params)
|
|
|
|
except Exception as e:
|
|
self.log_error(f"导出准备过程出错: {str(e)}")
|
|
traceback.print_exc()
|
|
|
|
def log_message(self, message):
|
|
"""添加日志消息"""
|
|
# 如果父窗口存在，使用父窗口的日志功能
|
|
if self.main_window and hasattr(self.main_window, 'log_signal'):
|
|
self.main_window.log_signal.emit(message)
|
|
else:
|
|
# 否则发射自己的信号
|
|
self.log_signal.emit(message)
|
|
|
|
def log_error(self, message):
|
|
"""添加错误日志"""
|
|
self.log_message(f"错误: {message}")
|
|
|
|
def on_script_log(self, task_id, message):
|
|
"""脚本输出日志"""
|
|
self.log_message(f"{task_id}: {message}")
|
|
|
|
def on_script_finished(self, task_id:str, success:bool, message:str):
|
|
"""导出完成事件处理"""
|
|
# 恢复导出按钮
|
|
self.export_button.setEnabled(True)
|
|
|
|
if success:
|
|
self.status_label.setText("状态: 导出完成")
|
|
QMessageBox.information(self, "成功", "地图导出完成！")
|
|
else:
|
|
self.status_label.setText("状态: 导出失败")
|
|
QMessageBox.warning(self, "失败", "地图导出过程中出现错误，请查看日志详情")
|
|
|
|
def validate_and_params(self):
|
|
# 获取输入参数
|
|
county_name = self.county_name_edit.text().strip()
|
|
if not county_name:
|
|
QMessageBox.warning(self, "错误", "请输入区县名称")
|
|
return
|
|
|
|
# 获取面要素列表
|
|
polygon_list = self.get_selected_polygon()
|
|
if not polygon_list:
|
|
QMessageBox.warning(self, "错误", "请选择至少一个要素")
|
|
return
|
|
|
|
# 获取配置文件
|
|
config_file = self.config_file_edit.text()
|
|
if not os.path.exists(config_file):
|
|
QMessageBox.warning(self, "错误", "配置文件不存在")
|
|
return
|
|
|
|
# 获取模板文件
|
|
template_aprx_file = self.template_aprx_file_edit.text()
|
|
if not os.path.exists(template_aprx_file):
|
|
QMessageBox.warning(self, "错误", "模板文件不存在")
|
|
return
|
|
|
|
# 获取输出路径
|
|
output_path = self.output_path_edit.text()
|
|
if not output_path:
|
|
QMessageBox.warning(self, "错误", "请选择输出路径")
|
|
return
|
|
|
|
# 获取数据源路径
|
|
data_source_path = self.data_source_path_edit.text()
|
|
if not os.path.exists(data_source_path) and not data_source_path.endswith(".gdb"):
|
|
QMessageBox.warning(self, "错误", "请确认是否是有效数据源")
|
|
return
|
|
|
|
# 获取符号系统路径
|
|
symbol_path = self.symbol_path_edit.text()
|
|
if not os.path.exists(symbol_path):
|
|
QMessageBox.warning(self, "错误", "符号系统路径不存在")
|
|
return
|
|
|
|
# 获取图片源路径
|
|
pic_path = self.pic_path_edit.text()
|
|
if not os.path.exists(pic_path):
|
|
QMessageBox.warning(self, "错误", "符号系统路径不存在")
|
|
return
|
|
|
|
# 是否强制重新生成
|
|
force_regenerate = self.force_regenerate_check.isChecked()
|
|
|
|
# 准备参数
|
|
return {
|
|
'config_file': config_file,
|
|
'county_name': county_name,
|
|
'polygon_list': polygon_list,
|
|
'template_aprx_file': template_aprx_file,
|
|
'output_path': output_path,
|
|
'data_source_path': data_source_path,
|
|
'symbol_path': symbol_path,
|
|
'force_regenerate': force_regenerate,
|
|
'pic_path': pic_path
|
|
}
|
|
|
|
def load_settings(self):
|
|
"""从主窗口加载设置"""
|
|
try:
|
|
if not self.main_window or not hasattr(self.main_window, 'settings'):
|
|
return
|
|
|
|
settings = self.main_window.settings
|
|
if 'export_map_settings' in settings:
|
|
paths = settings['export_map_settings']
|
|
|
|
# 直接使用get方法设置默认值
|
|
self.config_file_edit.setText(paths.get("config_file", ""))
|
|
self.county_name_edit.setText(paths.get("county_name", ""))
|
|
self.template_aprx_file_edit.setText(paths.get("template_aprx_file", ""))
|
|
self.output_path_edit.setText(paths.get("output_path", ""))
|
|
self.data_source_path_edit.setText(paths.get("data_source_path", ""))
|
|
self.symbol_path_edit.setText(paths.get("symbol_path", ""))
|
|
self.pic_path_edit.setText(paths.get("pic_path", ""))
|
|
|
|
except Exception as e:
|
|
self.log_error(f"加载设置失败: {str(e)}")
|
|
|
|
def get_settings(self):
|
|
"""获取当前设置，供主窗口保存使用"""
|
|
return {
|
|
'config_file': self.config_file_edit.text(),
|
|
'county_name': self.county_name_edit.text(),
|
|
'template_aprx_file': self.template_aprx_file_edit.text(),
|
|
'output_path': self.output_path_edit.text(),
|
|
'data_source_path': self.data_source_path_edit.text(),
|
|
'symbol_path': self.symbol_path_edit.text(),
|
|
'pic_path': self.pic_path_edit.text()
|
|
}
|
|
|
|
def update_config_file(self, config_file):
|
|
self.config_file_edit.setText(config_file) |