| | |
|---|
| | | 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 * |
|---|
| | | |
|---|
| | | |
|---|
| | | # 获取完整路径 |
|---|
| | |
|---|
| | | |
|---|
| | | |
|---|
| | | # 设置图层可见性 |
|---|
| | | 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() |
|---|
| | |
|---|
| | | 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)) |
|---|
| | | |
|---|
| | | |
|---|
| | | # 初始化 |
|---|
| | |
|---|
| | | |
|---|
| | | # 加载工程 |
|---|
| | | 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() |
|---|
| | | |
|---|
| | | # 加载模板 |
|---|
| | |
|---|
| | | # 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") |
|---|
| | |
|---|
| | | 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() |
|---|
| | |
|---|
| | | # layout.updateBounds() |
|---|
| | | layout.renderContext() |
|---|
| | | layout.refresh() |
|---|
| | | # project.write("C:/360/a.qgz") |
|---|
| | | |
|---|
| | | # 导出 |
|---|
| | | exporter = QgsLayoutExporter(layout) |
|---|
