1

13693261870

2022-11-08 9330f09dd592dcb744e1491e3afdb96e94af4eb3

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
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(project, showLayers):
    for layer in project.mapLayers().values():
        flag = layer.name() in showLayers
        print(layer.name() + ": " + str(flag))
        # if not flag:
        # project.removeMapLayer(layer)
        layer.setValid(flag)
 
 
def setItemVisibility(root, showLayers):
    print('Group: ' + root.name())
    for node in root.children():
        # QgsLayerTreeLayer
        if isinstance(node, QgsLayerTreeGroup):
            setItemVisibility(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)
            node.setItemVisibilityChecked(flag)
 
 
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, showLayers)
setItemVisibility(root, showLayers)
 
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()