You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

269 lines
9.8 KiB
Python

# -*- coding: utf-8 -*-
"""
/***************************************************************************
GemeindeMessdatenImport
A QGIS plugin
Importiert Messdaten im TXT-Format in die Gemeindeeigene Messdaten-Datenbank.
-------------------
begin : 2024-02-01
git sha : $Format:%H$
copyright : (C) 2024 by Sebastian Pertl
email : sebastian.pertl@fridolfing.bayern.de
***************************************************************************/
/***************************************************************************
* *
* This program is free software; you can redistribute it and/or modify *
* it under the terms of the GNU General Public License as published by *
* the Free Software Foundation; either version 2 of the License, or *
* (at your option) any later version. *
* *
***************************************************************************/
"""
from qgis.PyQt import uic
from qgis.PyQt import QtWidgets
from qgis.PyQt.QtWidgets import QFileDialog
from qgis.PyQt.QtCore import QSettings, QTranslator, QCoreApplication
from qgis.PyQt.QtGui import QIcon
from qgis.PyQt.QtWidgets import QAction, QComboBox, QLabel
from qgis.core import QgsVectorLayer, QgsVectorFileWriter, QgsProviderRegistry
from qgis.utils import iface
# Initialize Qt resources from file resources.py
from .resources import *
# Import the code for the dialog
#from .gdeMessdatenImport_dialog import GemeindeMessdatenImportDialog
import os
import os.path
import toml
class GemeindeTools:
"""QGIS Plugin Implementation."""
def __init__(self, iface):
"""Constructor.
:param iface: An interface instance that will be passed to this class
which provides the hook by which you can manipulate the QGIS
application at run time.
:type iface: QgsInterface
"""
# Save reference to the QGIS interface
self.iface = iface
# initialize plugin directory
self.plugin_dir = os.path.dirname(__file__)
# Declare instance attributes
# Eintrag Menüband Oben
self.actions = []
self.messdatenImportActions = []
self.menu = u'&Gemeinde Tools-Menu'
# Check if plugin was started the first time in current QGIS session
# Must be set in initGui() to survive plugin reloads
self.first_start = None
with open(os.path.join(self.plugin_dir, 'config.toml'), 'r') as f:
self.config = toml.load(f)
self.geschossNavi = ''
def initGui(self):
"""Create the menu entries and toolbar icons inside the QGIS GUI."""
self.messdatenImport = GemeindeMessdatenImport(self.config['database']['messdaten'])
self.messdatenImportMenu = u"Gemeinde-Tools: Messdaten Import"
self.toolBar1 = self.iface.addToolBar(self.messdatenImportMenu)
icon_path = os.path.join(self.plugin_dir, 'icon/txt.png')
action = QAction(QIcon(icon_path), u'Import Messdaten', self.iface.mainWindow())
action.triggered.connect(self.messdatenImport.run_import_txt)
self.toolBar1.addAction(action)
self.iface.addPluginToMenu(self.messdatenImportMenu, action) # Menüband oben
self.messdatenImportActions.append(action)
icon_path = os.path.join(self.plugin_dir, 'icon/dupl.png')
action = QAction(QIcon(icon_path), u'Lösche Vorhandene', self.iface.mainWindow())
action.triggered.connect(self.messdatenImport.run_del_dup)
self.toolBar1.addAction(action)
self.iface.addPluginToMenu(self.messdatenImportMenu, action)
self.messdatenImportActions.append(action)
icon_path = os.path.join(self.plugin_dir, 'icon/db.png')
action = QAction(QIcon(icon_path), u'Gemeinde Übertrag', self.iface.mainWindow())
action.triggered.connect(self.messdatenImport.run_save)
self.toolBar1.addAction(action)
self.iface.addPluginToMenu(self.messdatenImportMenu, action)
self.messdatenImportActions.append(action)
self.toolBar2 = self.iface.addToolBar("Gemeinde-Tools: Geschossnavigation")
self.lab = QLabel(self.iface.mainWindow())
self.lab.setText("Geschossauswahl: ")
self.toolBar2.addWidget(self.lab)
self.act = QComboBox(self.iface.mainWindow())
for k in self.config['geschossnavi']['ziele']:
self.act.addItem(self.config['geschossnavi']['ziele'][k])
#self.act.currentTextChanged.connect(self.geschossNavi.run)
self.toolBar2.addWidget(self.act)
self.first_start = True
def unload(self):
"""Removes the plugin menu item and icon from QGIS GUI."""
for action in self.messdatenImportActions:
self.iface.removePluginMenu(self.messdatenImportMenu, action)
self.iface.removeToolBarIcon(action)
class GemeindeMessdatenImport():
def __init__(self, inp_file, parent=None):
"""Constructor."""
self.inp_file = inp_file
self.md = QgsProviderRegistry.instance().providerMetadata("ogr")
self.con = self.md.createConnection( self.inp_file, {})
def run_import_txt(self):
qfd = QFileDialog()
filename = QFileDialog.getOpenFileName(qfd, "Wähle Vermessungsdaten","::{645ff040-5081-101b-9f08-00aa002f954e}", '*.txt')
csv = filename[0]
rutacsv = 'file:///' + csv + '?delimiter=,&xField=field_3&yField=field_2&useHeader=no'
tabla = QgsVectorLayer(rutacsv, os.path.basename(csv), 'delimitedtext')
opt = QgsVectorFileWriter.SaveVectorOptions()
opt.EditionCapability = QgsVectorFileWriter.CanAddNewLayer
opt.layerName = "messdaten_imp"
opt.actionOnExistingFile = QgsVectorFileWriter.CreateOrOverwriteLayer
QgsVectorFileWriter.writeAsVectorFormat(tabla, self.inp_file, opt)
# Datensätze bereinigen
query = """UPDATE messdaten_imp SET
field_6 = replace(field_6 ,"HSDV:",""),
field_7 = replace(field_7 ,"VSDV:",""),
field_8 = replace(field_8 ,"STATUS:",""),
field_9 = replace(field_9 ,"SATS:",""),
field_10 = replace(field_10,"AGE:",""),
field_11 = replace(field_11,"PDOP:",""),
field_12 = replace(field_12,"HDOP:",""),
field_13 = replace(field_13,"VDOP:",""),
field_14 = replace(field_14,"TDOP:",""),
field_15 = replace(field_15,"GDOP:",""),
field_16 = replace(field_16,"NSDV:",""),
field_17 = replace(field_17,"ESDV:",""),
field_18 = replace(field_18,"DATE:",""),
field_19 = replace(field_19,"TIME:","");"""
self.con.executeSql(query)
## Erzeuge Geometrie für Vorschau
#query = """UPDATE messdaten_imp SET geom = AsGPB(setsrid(makepoint(field_3, field_2), 25832));"""
#con.executeSql(query)
# Datum formatieren
query = """UPDATE messdaten_imp SET field_18 = substr(field_18,7,4)||'-'||substr(field_18,1,2)||'-'||substr(field_18,4,2);"""
self.con.executeSql(query)
iface.mapCanvas().refreshAllLayers()
def run_del_dup(self):
# Duplikate löschen
query = """DELETE FROM messdaten_imp WHERE field_3 || field_2 || field_4 in (SELECT rw || hw || höhe FROM messdaten);"""
self.con.executeSql(query)
query = """DELETE FROM messdaten_imp WHERE field_1 in (SELECT Nummer FROM messdaten);"""
self.con.executeSql(query)
iface.mapCanvas().refreshAllLayers()
def run_save(self):
# Daten übertragen
query = """INSERT INTO
messdaten
(
"fid",
"geom",
"Nummer",
"RW",
"HW",
"Höhe",
"Beschreibung",
"HSDV",
"VSDV",
"STATUS",
"SATS",
"AGE",
"PDOP",
"HDOP",
"VDOP",
"TDOP",
"GDOP",
"NSDV",
"ESDV",
"aufnahme_dat",
"aufnahme_uhr",
"import_dat",
"Anmerkung",
"Quelle"
)
SELECT
NULL,
NULL,
field_1,
field_3,
field_2,
field_4,
field_5,
field_6 ,
field_7 ,
field_8 ,
field_9 ,
field_10,
field_11,
field_12,
field_13,
field_14,
field_15,
field_16,
field_17, --ESDV
field_18,
field_19,
DATE('now'),
'',
'Vermessung Bauhof'
FROM messdaten_imp;"""
self.con.executeSql(query)
query = """UPDATE messdaten SET geom = setsrid(makepoint(RW, HW), 25832) WHERE "import_dat" = DATE('now')"""
self.con.executeSql(query)
# Anzahl der Objekte aktuallisieren
query = """DELETE FROM messdaten_imp WHERE field_1 in (SELECT Nummer FROM messdaten);"""
self.con.executeSql(query)
# Import-Tabelle leeren
query = """DELETE FROM messdaten_imp;"""
self.con.executeSql(query)
iface.mapCanvas().refreshAllLayers()