管道基础大数据平台系统开发-【CS】-ExportMap
13693261870
2023-10-23 2fd93008d2ce0052353e5016f2c15d1890ffd2e6
MoonExp/Sources/MoonExp.py
@@ -4,12 +4,12 @@
import sys
import math
import argparse
from qgis.core import *
from qgis.gui import *
from qgis.core import *
from qgis.PyQt.QtGui import *
from qgis.PyQt.QtXml import *
from qgis.PyQt.QtCore import *
from qgis.PyQt.QtWidgets import *
from qgis.PyQt.QtXml import *
# 获取完整路径
@@ -22,8 +22,8 @@
    print("argv = ", sys.argv[1:])
    parser = argparse.ArgumentParser(description='ArgUtils')
    parser.add_argument("-src", type=str, default=get_full_path(), required=False)
    parser.add_argument("-qgz", type=str, default="Test.qgz", required=False)
    parser.add_argument("-qpt", type=str, default="Test.qpt", required=False)
    parser.add_argument("-qgz", type=str, default="Moon.qgz", required=False)
    parser.add_argument("-qpt", type=str, default="Moon.qpt", required=False)
    return parser.parse_args()
@@ -34,17 +34,20 @@
        flag = layer.name() in show_layers
        print(layer.name() + ": " + str(flag))
        # if not flag:
        # project.removeMapLayer(layer)
        # prj.removeMapLayer(layer)
        layer.setValid(flag)
# 设置图层可见性
def set_item_visibility(root_node, show_layers):
def set_item_visibility(root_node, layout, show_layers):
    print(root_node.name() + ': Group')
    for node in root_node.children():
        # QgsLayerTreeLayer
        if isinstance(node, QgsLayerTreeGroup):
            set_item_visibility(node, show_layers)
            if node.name() == "底图":
                add_raster_layer(node, layout)
            else:
                set_item_visibility(node, layout, show_layers)
        # QgsRasterLayer, QgsVectorLayer
        else:
            name = node.layer().name()
@@ -54,6 +57,44 @@
            # root_node.removeLayer(node.layer())
            # node.layer().setValid(False)
            node.setItemVisibilityChecked(flag)
# 添加Raster图层:QgsLayerTreeGroup
def add_raster_layer(node, layout):
    wmsUrl = layout.customProperty("wmsUrl")
    if wmsUrl is None:
        print("wmsUrl 为空!")
        return
    uris = wmsUrl.split(',')
    # uris = uris[::-1]
    i = 0
    for uri in uris:
        i = i + 1
        if uri is None:
            continue
        url = "crs=ESRI:104903&dpiMode=7&format=image/png&layers=Default&styles=default&tileMatrixSet=GoogleCRS84Quad01&url=" + uri
        print("WMTS地址:" + url)
        layer = QgsRasterLayer(url, "WmsLayer_" + str(i), "wms")
        if layer.isValid():
            ls = project.layerStore()
            ls.addMapLayer(layer)
            node.addLayer(layer)
# 添加底图:layer.isValid(),layer.metadataUri(),layer.providerType()
def add_map_layer(prj, layout):
    uris = layout.customProperty("wmsUrl").split(',')
    for uri in uris:
        if uri is None:
            continue
        url = "crs=ESRI:104903&dpiMode=7&format=image/png&layers=Default&styles=default&tileMatrixSet=GoogleCRS84Quad01&url=" + uri
        print("WMS地址:" + url)
        layer = QgsRasterLayer(url, "WmsLayer", "wms")
        # if layer.isValid():
        prj.addMapLayer(layer)
# 初始化
@@ -103,15 +144,17 @@
# showLayers = ["阀室", "管道中心线", "矢量注记", "矢量图"]
showLayers = layout.customProperty("layers").split(',')
root = project.layerTreeRoot()
# set_layers_valid(project, showLayers)
set_item_visibility(root, showLayers)
set_item_visibility(root, layout, showLayers)
# add_map_layer(project, layout)
# sm = layout.itemById(r'说明')
table = layout.multiFrames()[0]
'''table = layout.multiFrames()[0]
contents = table.tableContents()
contents[1][1] = scale
table.refreshAttributes()
table.refresh()
table.refresh()'''
# collection = layout.pageCollection()
# page = collection.page(0)
@@ -122,6 +165,7 @@
# layout.updateBounds()
layout.renderContext()
layout.refresh()
# project.write("C:/360/a.qgz")
# 导出
exporter = QgsLayoutExporter(layout)