1

13693261870

2024-09-05 95d9356edef73a1ad5b04e62b7aef6b77697a226

 ExportMap/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 *


# 获取完整路径
@@ -39,12 +39,15 @@


# 设置图层可见性
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()
@@ -56,12 +59,57 @@
            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, args):
    url = "crs=ESRI:104903&dpiMode=7&format=image/png&layers=Default&styles=default&tileMatrixSet=GoogleCRS84Quad01&url=http://219.153.49.10:7511/sj_raster/v6/wmts/service/system/10000201/1?ak=mf72ff9295c740ec0f37e61433e8a3ad8d"
    layer = QgsRasterLayer(url, "WmsLayer", "wms")
    prj.addMapLayer(layer)
    print(url)
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)


# 获取比例尺
def get_scale(scale):
    # scale = '1:' + str(math.ceil(map.scale() / 100) * 100)
    if scale > 10000:
        return "1:" + str(math.ceil(map.scale() / 10000)) + "万"
    elif scale > 1000:
        return "1:" + str(math.ceil(map.scale() / 1000)) + "千"
    elif scale > 100:
        return "1:" + str(math.ceil(map.scale() / 100) * 100)
    elif scale > 10:
        return "1:" + str(math.ceil(map.scale() / 10) * 10)
    else:
        return "1:" + str(math.ceil(scale))


# 初始化
@@ -71,13 +119,13 @@

# 加载工程
args = get_args()
prj = QgsProject.instance()
# prj.read(r'Test.qgz')
prj.read(os.path.join(args.src, args.qgz))
print("FileName: " + prj.fileName())
project = QgsProject.instance()
# project.read(r'Test.qgz')
project.read(os.path.join(args.src, args.qgz))
print("FileName: " + project.fileName())

# layout = QgsPrintLayout(prj)
layout = QgsLayout(prj)
# layout = QgsPrintLayout(project)
layout = QgsLayout(project)
# layout.initializeDefaults()

# 加载模板
@@ -91,36 +139,30 @@
# map = layout.itemById(r'地图')
map = layout.referenceMap()
map.zoomToExtent(map.extent())
scale = '1:' + str(math.ceil(map.scale() / 100) * 100)
scale = get_scale(map.scale())
print(scale)

# template_content = template_content.replace('1：25万', scale)
# doc.setContent(template_content)
# layout.loadFromTemplate(doc, QgsReadWriteContext(), True)
# map = layout.referenceMap()
# map.zoomToExtent(map.extent())

size = map.sizeWithUnits()
print('size: ' + str(size.width()) + " * " + str(size.height()) + " mm")
if size.width() > 280 or size.height() > 174:
    map.attemptResize(QgsLayoutSize(280, 174, QgsUnitTypes.LayoutMillimeters))
if size.width() > 233 or size.height() > 174:
    map.attemptResize(QgsLayoutSize(233, 174, QgsUnitTypes.LayoutMillimeters))
    print('size: ' + str(map.sizeWithUnits().width()) + " * " + str(map.sizeWithUnits().height()) + " mm")

# 设置图层
# "阀室", "站场", "管道中心线", "矢量注记", "矢量图", "高德影像注记", "高德影像", "影像注记", "影像图"
# showLayers = ["阀室", "管道中心线", "矢量注记", "矢量图"]
# showLayers = ["阀室", "站场", "管道中心线", "矢量注记", "矢量图", "高德影像注记", "高德影像", "影像注记", "影像图"]
showLayers = layout.customProperty("layers").split(',')
root = prj.layerTreeRoot()
root = project.layerTreeRoot()

# set_layers_valid(prj, showLayers)
set_item_visibility(root, showLayers)
add_map_layer(prj, args)
# set_layers_valid(project, showLayers)
set_item_visibility(root, layout, showLayers)
# add_map_layer(project, layout)

# sm = layout.itemById(r'说明')
'''table = layout.multiFrames()[0]
contents = table.tableContents()
contents[1][1] = scale
table.refreshAttributes()
# contents[0][1] = scale
contents[1][1].setContent(scale)
table.setTableContents(contents)
table.refresh()'''

# collection = layout.pageCollection()
@@ -132,6 +174,7 @@
# layout.updateBounds()
layout.renderContext()
layout.refresh()
# project.write("C:/360/a.qgz")

# 导出
exporter = QgsLayoutExporter(layout)