1

13693261870

2022-11-08 d59ed5d1ab6910ed25b547b23c2c20510e8bc8d2

 QGIS/render.py

@@ -1,5 +1,6 @@
#!/usr/bin/env
# -*- coding: utf-8 -*-
import os
import sys
import math
import argparse
@@ -11,11 +12,16 @@
from qgis.PyQt.QtXml import *


# 获取完整路径
def get_full_path():
    return os.path.split(sys.argv[0])[0]


# 获取参数
def getArgs():
    argv = sys.argv[1:]
    print("argv = ", argv)
def get_args():
    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)

@@ -23,9 +29,9 @@


# 设置图层可用性
def setLayersValid(project, showLayers):
    for layer in project.mapLayers().values():
        flag = layer.name() in showLayers
def set_layers_valid(prj, show_layers):
    for layer in prj.mapLayers().values():
        flag = layer.name() in show_layers
        print(layer.name() + ": " + str(flag))
        # if not flag:
        # project.removeMapLayer(layer)
@@ -33,19 +39,19 @@


# 设置图层可见性
def setItemVisibility(root, showLayers):
    print(root.name() + ': Group')
    for node in root.children():
def set_item_visibility(root_node, show_layers):
    print(root_node.name() + ': Group')
    for node in root_node.children():
        # QgsLayerTreeLayer
        if isinstance(node, QgsLayerTreeGroup):
            setItemVisibility(node, showLayers)
            set_item_visibility(node, show_layers)
        # QgsRasterLayer, QgsVectorLayer
        else:
            name = node.layer().name()
            flag = name in showLayers
            print(name + ": " + str(flag))
            # if not flag:
            # root.removeLayer(node.layer())
            # root_node.removeLayer(node.layer())
            # node.layer().setValid(False)
            node.setItemVisibilityChecked(flag)

@@ -56,10 +62,10 @@
qgs.initQgis()

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

# layout = QgsPrintLayout(project)
@@ -68,7 +74,7 @@

# 加载模板
# with open(r'Test.qpt', 'r', encoding='utf-8') as f:
with open(args.qpt, 'r', encoding='utf-8') as f:
with open(os.path.join(args.src, args.qpt), 'r', encoding='utf-8') as f:
    template_content = f.read()
doc = QDomDocument()
doc.setContent(template_content)
@@ -83,8 +89,8 @@
# showLayers = ["阀室", "管道中心线", "矢量注记", "矢量图"]
showLayers = layout.customProperty("layers").split(',')
root = project.layerTreeRoot()
# setLayersValid(project, showLayers)
setItemVisibility(root, showLayers)
# set_layers_valid(project, showLayers)
set_item_visibility(root, showLayers)

# 刷新
layout.refresh()