-
Notifications
You must be signed in to change notification settings - Fork 3
/
tb_esporta_shp.py
109 lines (89 loc) · 3.52 KB
/
tb_esporta_shp.py
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
# -*- coding: utf-8 -*-
# -------------------------------------------------------------------------------
# Name: tb_esporta_shp.py
# Author: Tarquini E.
# Created: 08-02-2018
# -------------------------------------------------------------------------------
import os
import shutil
import sqlite3
import sys
import webbrowser
import zipfile
from qgis.core import *
from qgis.gui import *
from qgis.PyQt import QtGui, uic
from qgis.PyQt.QtCore import *
from qgis.PyQt.QtGui import *
from qgis.PyQt.QtWidgets import *
from qgis.utils import *
from . import constants
from .setup_workers import setup_workers
from .workers.export_worker import ExportWorker
FORM_CLASS, _ = uic.loadUiType(os.path.join(
os.path.dirname(__file__), 'tb_esporta_shp.ui'))
class esporta_shp(QDialog, FORM_CLASS):
def __init__(self, parent=None):
"""Constructor."""
self.iface = iface
super(esporta_shp, self).__init__(parent)
self.setupUi(self)
self.plugin_dir = os.path.dirname(__file__)
def esporta_prog(self):
self.help_button.clicked.connect(lambda: webbrowser.open(
'https://www.youtube.com/watch?v=dYcMZSpu6HA&t=2s'))
self.dir_output.clear()
self.alert_text.hide()
self.button_box.setEnabled(False)
self.dir_output.textChanged.connect(self.disableButton)
self.show()
self.adjustSize()
result = self.exec_()
if result:
try:
in_dir = QgsProject.instance().readPath("./")
out_dir = self.dir_output.text()
if os.path.exists(out_dir):
# create export worker
worker = ExportWorker(in_dir, out_dir, self.plugin_dir)
# create export log file
logfile_path = os.path.join(in_dir, "Allegati", "log", str(
time.strftime("%Y-%m-%d_%H-%M-%S", time.gmtime())) + "_export_log.txt")
log_file = open(logfile_path, 'a')
log_file.write("EXPORT REPORT:" + "\n---------------\n\n")
# start export worker
setup_workers().start_worker(worker, self.iface,
'Starting export task...', log_file)
else:
QMessageBox.warning(
None, self.tr('WARNING!'), self.tr("The selected directory does not exist!"))
except Exception as z:
QMessageBox.critical(
None, 'ERROR!', 'Error:\n"' + str(z) + '"')
def disableButton(self):
conteggio = 0
check_campi = [self.dir_output.text()]
check_value = []
layers = QgsProject.instance().mapLayers().values()
for layer in layers:
if layer.name() in constants.LISTA_LAYER:
conteggio += 1
for x in check_campi:
if len(x) > 0:
value_campi = 1
check_value.append(value_campi)
else:
value_campi = 0
check_value.append(value_campi)
campi = sum(check_value)
if conteggio > 23 and campi > 0:
self.button_box.setEnabled(True)
self.alert_text.hide()
elif conteggio > 23 and campi == 0:
self.button_box.setEnabled(False)
self.alert_text.hide()
else:
self.button_box.setEnabled(False)
self.alert_text.show()
def tr(self, message):
return QCoreApplication.translate('esporta_shp', message)