import os
|
import math
|
from qgis.core import *
|
from qgis.gui import *
|
from qgis.PyQt.QtGui import *
|
from qgis.PyQt.QtCore import *
|
from qgis.PyQt.QtWidgets import *
|
from qgis.PyQt.QtXml import *
|
|
|
def setLayersValid(root, showLayers):
|
print('Group: ' + root.name())
|
for node in root.children():
|
# QgsLayerTreeLayer
|
if isinstance(node, QgsLayerTreeGroup):
|
setChild(node, showLayers)
|
# QgsRasterLayer, QgsVectorLayer
|
else:
|
name = node.layer().name()
|
flag = name in showLayers
|
print(str(flag) + ", " + name)
|
|
if not flag:
|
# root.removeLayer(node.layer())
|
node.layer().setValid(False)
|
|
|
QgsApplication.setPrefixPath(r"C:/Program Files/QGIS 3.16/bin", True)
|
qgs = QgsApplication([], True)
|
qgs.initQgis()
|
|
project = QgsProject.instance()
|
project.read(r'E:/terrait/TianJin/LFServer/QGIS/Test.qgz')
|
print("FileName: " + project.fileName())
|
|
# layout = QgsPrintLayout(project)
|
layout = QgsLayout(project)
|
# layout.initializeDefaults()
|
|
with open(r"E:/terrait/TianJin/LFServer/QGIS/Template.qpt", 'r', encoding='utf-8') as f:
|
template_content = f.read()
|
doc = QDomDocument()
|
doc.setContent(template_content)
|
layout.loadFromTemplate(doc, QgsReadWriteContext(), True)
|
|
# map = layout.itemById(r'地图')
|
map = layout.referenceMap()
|
print('1:' + str(math.ceil(map.scale())))
|
|
# "阀室", "站场", "管道中心线", "矢量注记", "矢量图", "影像注记", "影像图"
|
showLayers = ["阀室", "管道中心线", "影像注记", "影像图"]
|
# root = project.layerTreeRoot()
|
# setLayersValid(project, root, showLayers)
|
for layer in project.mapLayers().values():
|
flag = layer.name() in showLayers
|
if not flag:
|
layer.setValid(flag)
|
print(layer.name() + ": " + str(flag))
|
|
layout.refresh()
|
map.refresh()
|
|
exporter = QgsLayoutExporter(layout)
|
img_path = os.path.join(r"E:/terrait/TianJin/LFServer/QGIS", "render.png")
|
exporter.exportToImage(img_path, QgsLayoutExporter.ImageExportSettings())
|
|
qgs.exitQgis()
|