#!/usr/bin/env
|
# -*- coding: utf-8 -*-
|
import os
|
import sys
|
import math
|
import argparse
|
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 *
|
|
|
# 获取完整路径
|
def get_full_path():
|
return os.path.split(sys.argv[0])[0]
|
|
|
# 获取参数
|
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="Moon.qgz", required=False)
|
parser.add_argument("-qpt", type=str, default="Moon.qpt", required=False)
|
|
return parser.parse_args()
|
|
|
# 设置图层可用性
|
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:
|
# prj.removeMapLayer(layer)
|
layer.setValid(flag)
|
|
|
# 设置图层可见性
|
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):
|
if node.name() == "底图":
|
add_raster_layer(node, layout)
|
else:
|
set_item_visibility(node, layout, show_layers)
|
# QgsRasterLayer, QgsVectorLayer
|
else:
|
name = node.layer().name()
|
flag = name in showLayers
|
print(name + ": " + str(flag))
|
# if not flag:
|
# root_node.removeLayer(node.layer())
|
# node.layer().setValid(False)
|
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, 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))
|
|
|
# 初始化
|
# QgsApplication.setPrefixPath(r"E:/terrait/TianJin/LFServer/QGIS/", True)
|
qgs = QgsApplication([], False)
|
qgs.initQgis()
|
|
# 加载工程
|
args = get_args()
|
project = QgsProject.instance()
|
# project.read(r'Test.qgz')
|
project.read(os.path.join(args.src, args.qgz))
|
print("FileName: " + project.fileName())
|
|
# layout = QgsPrintLayout(project)
|
layout = QgsLayout(project)
|
# layout.initializeDefaults()
|
|
# 加载模板
|
# with open(r'Test.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)
|
layout.loadFromTemplate(doc, QgsReadWriteContext(), True)
|
|
# map = layout.itemById(r'地图')
|
map = layout.referenceMap()
|
map.zoomToExtent(map.extent())
|
scale = get_scale(map.scale())
|
print(scale)
|
|
size = map.sizeWithUnits()
|
print('size: ' + str(size.width()) + " * " + str(size.height()) + " mm")
|
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 = layout.customProperty("layers").split(',')
|
root = project.layerTreeRoot()
|
|
# 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[0][1] = scale
|
contents[1][1].setContent(scale)
|
table.setTableContents(contents)
|
table.refresh()'''
|
|
# collection = layout.pageCollection()
|
# page = collection.page(0)
|
# page.refreshItemSize()
|
|
# 刷新
|
map.refresh()
|
# layout.updateBounds()
|
layout.renderContext()
|
layout.refresh()
|
# project.write("C:/360/a.qgz")
|
|
# 导出
|
exporter = QgsLayoutExporter(layout)
|
# img_path = os.path.join(r"E:/terrait/TianJin/LFServer/QGIS", "render.png")
|
img_path = layout.customProperty("imgPath")
|
exporter.exportToImage(img_path, QgsLayoutExporter.ImageExportSettings())
|
|
# 退出
|
qgs.exitQgis()
|
