106 lines
5.1 KiB
Python
106 lines
5.1 KiB
Python
import os
|
|
import importlib # 动态导入模块
|
|
import logging
|
|
import sys
|
|
from qgis._core import QgsProject
|
|
from .map_utils import MapUtils
|
|
|
|
|
|
class Maps:
|
|
def __init__(self, model, data, config):
|
|
self.model = model # 震情基本数据
|
|
self.data = data # 额外数据
|
|
self.config = config # 配置参数
|
|
|
|
def load(self):
|
|
# ---加载地图模板---
|
|
project = QgsProject.instance() # 获取QGIS项目实例
|
|
project.read(self.model["path"]) # 读取地图模板（.qgs或.qgz文件）
|
|
|
|
db_config = self.config["db"] # 需在application.yml中添加db配置
|
|
|
|
# 遍历所有图层，更新PostgreSQL图层的连接
|
|
for layer in project.mapLayers().values():
|
|
if layer.providerType() == "postgres": # 仅处理 PostgreSQL 图层
|
|
try:
|
|
# 检查配置项是否完整
|
|
required_keys = ["host", "port", "database", "username", "password"]
|
|
missing_keys = [k for k in required_keys if k not in db_config]
|
|
if missing_keys:
|
|
logging.error(f"数据库配置缺失键：{missing_keys}，跳过图层 {layer.name()}")
|
|
continue
|
|
|
|
# 获取当前图层的数据源 URI
|
|
uri = layer.dataProvider().uri()
|
|
|
|
|
|
# 使用位置参数传递，而非关键字参数
|
|
uri.setConnection(
|
|
db_config["host"], # 主机
|
|
str(db_config["port"]), # 端口转换为字符串
|
|
db_config["database"], # 数据库名
|
|
db_config["username"], # 用户名
|
|
db_config["password"] # 密码
|
|
)
|
|
|
|
# 重新设置数据源，刷新连接
|
|
layer.setDataSource(uri.uri(), layer.name(), "postgres")
|
|
|
|
# 验证图层是否有效
|
|
if layer.isValid():
|
|
logging.info(f"图层 {layer.name()} 数据库连接更新成功")
|
|
else:
|
|
logging.error(f"图层 {layer.name()} 更新连接后仍无效，请检查配置")
|
|
except Exception as e:
|
|
logging.error(f"更新图层 {layer.name()} 连接失败：{str(e)}")
|
|
|
|
logging.info("读取project完成，模板路径：" + self.model["path"] + " 画幅：" + self.model["mapLayout"])
|
|
# logging.info("图层--" + str(len(project.mapLayersByName('eqcenter')))) # 日志记录特定图层数量
|
|
qLayout = project.layoutManager().layoutByName(self.model["mapLayout"]) # 获取指定名称的布局
|
|
imap = qLayout.itemById("Map") # 获取地图项
|
|
mapUtils = MapUtils(self.config, project, qLayout) # 创建地图工具实例
|
|
|
|
# 设置坐标系（当前注释掉，未启用）
|
|
# kid = 32000 + (700 if float(self.model["centerY"]) < 0 else 600) + (int(float(self.model["centerX"])/6)+31)
|
|
# logging.info("设置坐标系：" + str(kid))
|
|
# imap.setCrs(QgsCoordinateReferenceSystem(kid,QgsCoordinateReferenceSystem.CrsType.EpsgCrsId))
|
|
|
|
# 修改格网
|
|
# logging.info("修改格网")
|
|
# grid = imap.grid()
|
|
# grid.setLineSymbol(None)
|
|
# grid.setAnnotationDisplay(QgsLayoutItemMapGrid.DisplayMode.HideAll,QgsLayoutItemMapGrid.BorderSide.Left)
|
|
# grid.setAnnotationDisplay(QgsLayoutItemMapGrid.DisplayMode.HideAll, QgsLayoutItemMapGrid.BorderSide.Right)
|
|
# grid.setAnnotationDisplay(QgsLayoutItemMapGrid.DisplayMode.HideAll, QgsLayoutItemMapGrid.BorderSide.Bottom)
|
|
# grid.setAnnotationDisplay(QgsLayoutItemMapGrid.DisplayMode.HideAll, QgsLayoutItemMapGrid.BorderSide.Top)
|
|
|
|
# 按照过滤图层，避免以前的数据干扰(动态加载map_filter.py)
|
|
if os.path.exists(os.path.join(os.path.split(os.path.realpath(__file__))[0], 'map_filter.py')):
|
|
# 添加模块所在目录到 Python 搜索路径
|
|
sys.path.insert(0, os.path.split(os.path.realpath(__file__))[0])
|
|
importlib.import_module('map_filter').Filter(project, self.model)
|
|
|
|
# 缩放地图
|
|
logging.info("缩放地图")
|
|
# 调用缩放方法（参数：缩放规则、中心点坐标、缩放值）
|
|
mapUtils.Zoom(self.model["zoomRule"],
|
|
{'X': self.model["centerX"], 'Y': self.model["centerY"], 'value': self.model["zoomValue"]})
|
|
|
|
# 修改制图时间、单位、地图名称等文本
|
|
logging.info("正在修改地震信息...")
|
|
mapUtils.Update(self.model, "mapTitle") # 更新地图标题
|
|
mapUtils.Update(self.model, "mapTime") # 更新制图时间
|
|
mapUtils.Update(self.model, "mapUnit") # 更新单位
|
|
mapUtils.Update(self.model, "info") # 更新地震信息
|
|
|
|
# 修改比例尺
|
|
logging.info("修改比例尺")
|
|
mapUtils.UpdateScale()
|
|
|
|
# 导出图片
|
|
logging.info("导出图片")
|
|
mapUtils.Export(self.model["outFile"]) # 导出至指定路径
|
|
|
|
logging.info(self.model["event"] + "导出完成") # 记录导出完成日志
|
|
return self.model["name"]
|