Atlas.py 107 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252
  1. # -*- coding: utf-8 -*-
  2. """
  3. /***************************************************************************
  4. Atlas
  5. A QGIS plugin
  6. Atlas
  7. Generated by Plugin Builder: http://g-sherman.github.io/Qgis-Plugin-Builder/
  8. -------------------
  9. begin : 2019-02-15
  10. git sha : $Format:%H$
  11. copyright : (C) 2019 by jan vrobel
  12. email : vrobel.jan@seznam.cz
  13. ***************************************************************************/
  14. /***************************************************************************
  15. * *
  16. * This program is free software; you can redistribute it and/or modify *
  17. * it under the terms of the GNU General Public License as published by *
  18. * the Free Software Foundation; either version 2 of the License, or *
  19. * (at your option) any later version. *
  20. * *
  21. ***************************************************************************/
  22. """
  23. from PyQt5.QtCore import QSettings, QTranslator, qVersion, QCoreApplication, Qt, QFileSystemWatcher, QRegExp,QDir
  24. from PyQt5.QtGui import QIcon, QPixmap, QRegExpValidator, QDoubleValidator
  25. from PyQt5.QtWidgets import QAction, QTreeWidget, QTreeWidgetItem, QMessageBox, QLabel, QProgressDialog, QDialog, QProgressBar,QListWidgetItem
  26. # Initialize Qt resources from file resources.py
  27. from .resources import *
  28. import re
  29. #from .flaskServer import *
  30. from multiprocessing import Process
  31. from pathlib import Path
  32. # Import the code for the DockWidget
  33. from builtins import str
  34. from builtins import range
  35. from builtins import object
  36. import json
  37. import zipfile
  38. import tempfile
  39. import configparser
  40. import shutil
  41. import uuid
  42. import copy
  43. import webbrowser
  44. from .Atlas_dockwidget import AtlasDockWidget
  45. import os.path
  46. import tempfile
  47. import qgis.core
  48. from qgis.core import *
  49. import platform
  50. import os
  51. import qgis.utils
  52. import qgis.gui
  53. from qgis.gui import QgsMapCanvas
  54. from qgis.core import QgsApplication
  55. from qgis.utils import iface
  56. from PyQt5.QtWidgets import QApplication, QWidget, QPushButton, QMessageBox, QWidget, QInputDialog, QLineEdit, QFileDialog
  57. import requests
  58. import tempfile
  59. import time
  60. from urllib.request import urlopen
  61. import subprocess
  62. import threading
  63. #from flask import Flask, request, jsonify
  64. import base64
  65. import hashlib
  66. import html
  67. import re
  68. import urllib.parse
  69. from .ogr2ogr import main
  70. ## forms
  71. from .dlg_GetLayers import GetLayersDialog
  72. from .dlg_importMap import ImportMapDialog
  73. from .dlg_deleteMap import DeleteMapDialog
  74. from .dlg_importLayer import ImportLayerDialog
  75. from .dlg_addLayer import AddLayerDialog
  76. from .dlg_addMap import AddMapDialog
  77. from .dlg_createComposite import CreateCompositeDialog
  78. from .dlg_deleteLayerFromMap import DeleteLayerFromMapDialog
  79. from .dlg_editMap import EditMapDialog
  80. from .dlg_ConnectionManager import ConnectionManagerDialog
  81. from .dlg_userInfo import UserInfoDialog
  82. class Atlas:
  83. """QGIS Plugin Implementation."""
  84. def __init__(self, iface):
  85. """Constructor.
  86. :param iface: An interface instance that will be passed to this class
  87. which provides the hook by which you can manipulate the QGIS
  88. application at run time.
  89. :type iface: QgsInterface
  90. """
  91. # Save reference to the QGIS interface
  92. self.iface = iface
  93. # initialize plugin directory
  94. self.plugin_dir = os.path.dirname(__file__)
  95. ## init global variables
  96. # global filename
  97. self.client_id = None
  98. self.filename = None
  99. self.layerName = None
  100. self.username = 'browser'
  101. self.EPSG = 'EPSG:3857'
  102. self.composite = None
  103. self.thread1 = None
  104. self.compositeList = []
  105. self.compositeListOld = []
  106. self.CHUNK_SIZE = 1048576 ## s touto hodnotou pracuje layman klient (cca 1MB soubor)
  107. self.URI = "http://layman.lesprojekt.cz"
  108. self.access_token = None
  109. self.expires_in = None
  110. self.refresh_token = None
  111. self.laymanUsername = ""
  112. self.authHeader = None
  113. self.code_verifier = None
  114. self.code_challenge = None
  115. self.Agrimail = None
  116. self.loadedInMemory = False
  117. self.uri = 'http://layman.lesprojekt.cz/rest/'
  118. # global dlgGetLayers
  119. self.dlgGetLayers= GetLayersDialog()
  120. # initialize locale
  121. locale = QSettings().value('locale/userLocale')[0:2]
  122. locale_path = os.path.join(
  123. self.plugin_dir,
  124. 'i18n',
  125. 'Atlas_{}.qm'.format(locale))
  126. if os.path.exists(locale_path):
  127. self.translator = QTranslator()
  128. self.translator.load(locale_path)
  129. if qVersion() > '4.3.3':
  130. QCoreApplication.installTranslator(self.translator)
  131. # Declare instance attributes
  132. self.actions = []
  133. self.menu = self.tr(u'&Layman')
  134. # TODO: We are going to let the user set this up in a future iteration
  135. self.toolbar = self.iface.addToolBar(u'Layman')
  136. self.toolbar.setObjectName(u'Layman')
  137. #print "** INITIALIZING Atlas"
  138. self.pluginIsActive = False
  139. self.dockwidget = None
  140. ## prepare temp dir
  141. tempDir = tempfile.gettempdir() + os.sep + "atlas"
  142. try:
  143. os.mkdir(tempDir)
  144. print("Directory " , tempDir , " Created ")
  145. except FileExistsError:
  146. print("Directory " , tempDir , " already exists")
  147. # noinspection PyMethodMayBeStatic
  148. def tr(self, message):
  149. """Get the translation for a string using Qt translation API.
  150. We implement this ourselves since we do not inherit QObject.
  151. :param message: String for translation.
  152. :type message: str, QString
  153. :returns: Translated version of message.
  154. :rtype: QString
  155. """
  156. # noinspection PyTypeChecker,PyArgumentList,PyCallByClass
  157. return QCoreApplication.translate('Layman', message)
  158. def add_action(
  159. self,
  160. icon_path,
  161. text,
  162. callback,
  163. enabled_flag=True,
  164. add_to_menu=True,
  165. add_to_toolbar=True,
  166. status_tip=None,
  167. whats_this=None,
  168. parent=None):
  169. icon = QIcon(icon_path)
  170. action = QAction(icon, text, parent)
  171. action.triggered.connect(callback)
  172. action.setEnabled(enabled_flag)
  173. if status_tip is not None:
  174. action.setStatusTip(status_tip)
  175. if whats_this is not None:
  176. action.setWhatsThis(whats_this)
  177. if add_to_toolbar:
  178. self.toolbar.addAction(action)
  179. if add_to_menu:
  180. self.iface.addPluginToMenu(
  181. self.menu,
  182. action)
  183. self.actions.append(action)
  184. return action
  185. def initGui(self):
  186. """Create the menu entries and toolbar icons inside the QGIS GUI."""
  187. #icon_path = ':/plugins/Atlas/icon.png'
  188. #self.add_action(
  189. # icon_path,
  190. # text=self.tr(u''),
  191. # callback=self.run,
  192. # parent=self.iface.mainWindow())
  193. #self.first_start = True
  194. ################## user
  195. self.textbox = QLabel(self.iface.mainWindow())
  196. # Set width
  197. self.textbox.setFixedWidth(140)
  198. # Add textbox to toolbar
  199. self.txtAction = self.toolbar.addWidget(self.textbox)
  200. # Set tooltip
  201. self.txtAction.setToolTip(self.tr(u'Current Row Number'))
  202. # Set callback
  203. self.textbox.setText("Layman")
  204. ################ end usericon_path = self.plugin_dir + os.sep + 'icons' + os.sep + 'login.png'
  205. icon_path = self.plugin_dir + os.sep + 'icons' + os.sep + 'login.png'
  206. self.menu_Connection = self.add_action(
  207. icon_path,
  208. text=self.tr(u'Login'),
  209. callback=self.run_login,
  210. enabled_flag=True,
  211. parent=self.iface.mainWindow())
  212. icon_path = self.plugin_dir + os.sep + 'icons' + os.sep + 'save.png'
  213. self.menu_saveLocalFile = self.add_action(
  214. icon_path,
  215. text=self.tr(u'Save as to JSON and SLD'),
  216. callback=self.saveLocalFile,
  217. enabled_flag=False,
  218. parent=self.iface.mainWindow())
  219. self.first_start = True
  220. icon_path = self.plugin_dir + os.sep + 'icons' + os.sep + 'file.png'
  221. self.menu_loadJson = self.add_action(
  222. icon_path,
  223. text=self.tr(u'Load from JSON'),
  224. callback=self.loadLocalFile,
  225. enabled_flag=False,
  226. parent=self.iface.mainWindow())
  227. self.first_start = True
  228. icon_path = self.plugin_dir + os.sep + 'icons' + os.sep + 'layers-up.png'
  229. self.menu_ImportLayerDialog = self.add_action(
  230. icon_path,
  231. text=self.tr(u'Export layer to server'),
  232. callback=self.run_ImportLayerDialog,
  233. enabled_flag=False,
  234. parent=self.iface.mainWindow())
  235. icon_path = self.plugin_dir + os.sep + 'icons' + os.sep + 'layers-down.png'
  236. self.menu_AddLayerDialog = self.add_action(
  237. icon_path,
  238. text=self.tr(u'Load layer from server'),
  239. callback=self.run_AddLayerDialog,
  240. enabled_flag=False,
  241. parent=self.iface.mainWindow())
  242. icon_path = self.plugin_dir + os.sep + 'icons' + os.sep + 'upload-map.png'
  243. self.menu_ImportMapDialog = self.add_action(
  244. icon_path,
  245. text=self.tr(u'Manage maps'),
  246. callback=self.run_ImportMapDialog,
  247. enabled_flag=False,
  248. parent=self.iface.mainWindow())
  249. icon_path = self.plugin_dir + os.sep + 'icons' + os.sep + 'download_map.png'
  250. self.menu_AddMapDialog = self.add_action(
  251. icon_path,
  252. text=self.tr(u'Load map from server'),
  253. callback=self.run_AddMapDialog,
  254. enabled_flag=False,
  255. parent=self.iface.mainWindow())
  256. #icon_path = self.plugin_dir + os.sep + 'icons' + os.sep + 'delete.png'
  257. #self.menu_DeleteMapDialog = self.add_action(
  258. # icon_path,
  259. # text=self.tr(u'Delete map'),
  260. # callback=self.run_DeleteMapDialog,
  261. # enabled_flag=False,
  262. # parent=self.iface.mainWindow())
  263. #icon_path = self.plugin_dir + os.sep + 'icons' + os.sep + 'globus.png'
  264. #self.menu_CreateCompositeDialog = self.add_action(
  265. # icon_path,
  266. # text=self.tr(u'Create map'),
  267. # callback=self.run_CreateCompositeDialog,
  268. # enabled_flag=False,
  269. # parent=self.iface.mainWindow())
  270. icon_path = self.plugin_dir + os.sep + 'icons' + os.sep + 'account.svg'
  271. self.menu_UserInfoDialog = self.add_action(
  272. icon_path,
  273. text=self.tr(u'User info'),
  274. callback=self.run_UserInfoDialog,
  275. enabled_flag=False,
  276. parent=self.iface.mainWindow())
  277. #--------------------------------------------------------------------------
  278. def run_UserInfoDialog(self):
  279. self.dlg = UserInfoDialog()
  280. self.dlg.show()
  281. userEndpoint = "http://layman.lesprojekt.cz/rest/current-user"
  282. r = requests.get(url = userEndpoint, headers = self.authHeader)
  283. res = r.text
  284. res = self.fromByteToJson(r.content)
  285. print(res['claims'])
  286. self.dlg.label_layman.setText(res['claims']['preferred_username'])
  287. self.dlg.label_agrihub.setText(res['claims']['email'])
  288. self.dlg.setStyleSheet("#DialogBase {background: #f0f0f0 ;}")
  289. def run_EditMap(self, x):
  290. self.dlg = EditMapDialog()
  291. self.dlg.pushButton_save.setStyleSheet("#pushButton_save {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_save:hover{background: #66ab27 ;}#pushButton_save:disabled{background: #64818b ;}")
  292. self.dlg.pushButton_close.setStyleSheet("#pushButton_close {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_close:hover{background: #66ab27 ;}")
  293. self.dlg.setStyleSheet("#DialogBase {background: #f0f0f0 ;}")
  294. self.dlg.lineEdit_name.hide()
  295. self.dlg.label_2.hide()
  296. self.dlg.pushButton_close.clicked.connect(lambda: self.dlg.close())
  297. self.dlg.lineEdit_name.setText(self.compositeList[x]['name'])
  298. self.dlg.lineEdit_abstract.setText(self.compositeList[x]['abstract'])
  299. self.dlg.lineEdit_title.setText(self.compositeList[x]['title'])
  300. self.dlg.lineEdit_units.setText(self.compositeList[x]['units'])
  301. self.dlg.lineEdit_scale.setText(str(self.compositeList[x]['scale']))
  302. self.dlg.lineEdit_user.setText(self.compositeList[x]['user']['name'])
  303. self.dlg.rejected.connect(lambda: self.afterCloseEditMapDialog())
  304. self.dlg.pushButton_save.clicked.connect(lambda: self.modifyMap(x))
  305. self.dlg.rejected.connect(lambda: self.afterCloseCompositeDialog())
  306. self.dlg.show()
  307. result = self.dlg.exec_()
  308. def run_DeleteLayerFromMap(self):
  309. self.dlg = DeleteLayerFromMapDialog()
  310. self.dlg.pushButton.clicked.connect(lambda: self.deteteLayerFromComposite(self.dlg.listWidget.currentRow(),self.compositeList[self.dlg.listWidget.currentRow()]['layers'][self.dlg.listWidget_listLayers2.currentRow()]['params']['LAYERS']))
  311. #self.dlg.pushButton.clicked.connect(lambda: print(self.dlg.listWidget_2.currentRow()))
  312. layers = QgsProject.instance().mapLayers().values()
  313. for i in range (0, len(self.compositeList)):
  314. self.dlg.listWidget.addItem(self.compositeList[i]['name'])
  315. # self.dlg.pushButton.setEnabled(False)
  316. self.dlg.listWidget.itemClicked.connect(self.listCompositeLayers)
  317. self.dlg.listWidget_listLayers2.itemClicked.connect(self.showThumbnail)
  318. # self.dlg.listWidget_listMaps.itemClicked.connect(self.enableButton)
  319. self.dlg.show()
  320. self.dlg.pushButton_close.clicked.connect(lambda: self.dlg.close())
  321. result = self.dlg.exec_()
  322. def run_CreateCompositeDialog(self, fromImport = False):
  323. self.dlg = CreateCompositeDialog()
  324. self.dlg.label_info.hide()
  325. self.dlg.label_2.hide()
  326. self.dlg.lineEdit.hide()
  327. self.dlg.pushButton_CreateComposition.clicked.connect(lambda: self.createComposite(self.dlg.lineEdit.text(),self.dlg.lineEdit_2.text()))
  328. layers = QgsProject.instance().mapLayers().values()
  329. self.dlg.treeWidget.itemClicked.connect(self.setExtent)
  330. for layer in layers:
  331. # self.dlg.listWidget_listLayers.addItem(layer.name())
  332. if (layer.type() == QgsMapLayer.VectorLayer):
  333. item = QTreeWidgetItem([layer.name()])
  334. self.dlg.treeWidget.addTopLevelItem(item)
  335. ext = iface.mapCanvas().extent()
  336. self.dlg.lineEdit.setValidator(QRegExpValidator(QRegExp("[a-z]{1}[a-z0-9]{1,30}")))
  337. self.dlg.lineEdit_2.editingFinished.connect(self.checkNameCreateMap)
  338. self.dlg.lineEdit_3.setText(str(ext.xMinimum()))
  339. self.dlg.lineEdit_4.setText(str(ext.xMaximum()))
  340. self.dlg.lineEdit_5.setText(str(ext.yMinimum()))
  341. self.dlg.lineEdit_6.setText(str(ext.yMaximum()))
  342. self.dlg.pushButton_defaultExtent.clicked.connect(lambda: self.setDefaultExtent(ext))
  343. self.dlg.pushButton_defaultExtent.setStyleSheet("#pushButton_defaultExtent {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_defaultExtent:hover{background: #66ab27 ;}#pushButton_defaultExtent:disabled{background: #64818b ;}")
  344. self.dlg.pushButton_CreateComposition.setStyleSheet("#pushButton_CreateComposition {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_CreateComposition:hover{background: #66ab27 ;}#pushButton_CreateComposition:disabled{background: #64818b ;}")
  345. self.dlg.pushButton_close.setStyleSheet("#pushButton_close {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_close:hover{background: #66ab27 ;}")
  346. self.dlg.setStyleSheet("#DialogBase {background: #f0f0f0 ;}")
  347. self.dlg.show()
  348. self.dlg.pushButton_close.clicked.connect(lambda: self.dlg.close())
  349. if (fromImport):
  350. self.dlg.rejected.connect(lambda: self.afterCloseCompositeDialog())
  351. result = self.dlg.exec_()
  352. def run_ImportMapDialog(self):
  353. self.dlg = ImportMapDialog()
  354. self.dlg.label_import.hide()
  355. #self.dlg.listWidget_listLayers2.hide()
  356. self.dlg.pushButton_deleteMap.setEnabled(False)
  357. self.dlg.pushButton_editMeta.setEnabled(False)
  358. self.dlg.pushButton_addRaster.setEnabled(False)
  359. self.dlg.mMapLayerComboBox.setFilters(QgsMapLayerProxyModel.VectorLayer)
  360. self.dlg.pushButton.clicked.connect(lambda: self.addLayerToComposite(self.dlg.listWidget.currentRow()))
  361. self.dlg.pushButton_deleteMap.clicked.connect(lambda: self.deleteMap(self.dlg.listWidget.currentItem().text(),self.dlg.listWidget.currentRow()))
  362. # self.dlg.pushButton_deleteMap.clicked.connect(lambda: self.deleteMapFromCanvas(self.dlg.listWidget.currentRow()))
  363. self.dlg.pushButton_up.clicked.connect(lambda: self.reorderLayers(self.dlg.listWidget_listLayers.currentRow(), 1, self.dlg.listWidget.currentRow()))
  364. self.dlg.pushButton_down.clicked.connect(lambda: self.reorderLayers(self.dlg.listWidget_listLayers.currentRow(), -1, self.dlg.listWidget.currentRow()))
  365. if not self.loadedInMemory:
  366. self.loadAllComposites()
  367. self.dlg.pushButton_addMap.clicked.connect(lambda: self.showAddMapDialog())
  368. layers = QgsProject.instance().mapLayers().values()
  369. for i in range (0, len(self.compositeList)):
  370. print(self.compositeList[i])
  371. #self.dlg.listWidget.addItem(self.compositeList[i]['name'])
  372. self.dlg.listWidget.addItem(self.compositeList[i]['title'])
  373. self.dlg.progressBar.hide()
  374. self.dlg.pushButton.setIcon(QIcon(self.plugin_dir + os.sep + 'icons' + os.sep + 'plus.png'))
  375. self.dlg.pushButton_addMap.setIcon(QIcon(self.plugin_dir + os.sep + 'icons' + os.sep + 'plus.png'))
  376. self.dlg.pushButton_addRaster.setIcon(QIcon(self.plugin_dir + os.sep + 'icons' + os.sep + 'plus.png'))
  377. self.dlg.pushButton_deleteLayers.setIcon(QIcon(self.plugin_dir + os.sep + 'icons' + os.sep + 'minus.png'))
  378. self.dlg.pushButton_deleteMap.setIcon(QIcon(self.plugin_dir + os.sep + 'icons' + os.sep + 'minus.png'))
  379. self.dlg.pushButton_editMeta.setIcon(QIcon(self.plugin_dir + os.sep + 'icons' + os.sep + 'edit.png'))
  380. self.dlg.listWidget.itemClicked.connect(self.refreshLayerList)
  381. self.dlg.listWidget.itemClicked.connect(lambda: self.dlg.pushButton_deleteMap.setEnabled(True))
  382. self.dlg.listWidget.itemClicked.connect(lambda: self.dlg.pushButton_editMeta.setEnabled(True))
  383. self.dlg.listWidget.itemClicked.connect(lambda: self.dlg.pushButton_addRaster.setEnabled(True))
  384. self.dlg.listWidget_listLayers.itemClicked.connect(lambda: self.dlg.pushButton_down.setEnabled(True))
  385. self.dlg.listWidget_listLayers.itemClicked.connect(lambda: self.dlg.pushButton_up.setEnabled(True))
  386. self.dlg.pushButton_editMeta.clicked.connect(lambda: self.showEditMapDialog(self.dlg.listWidget.currentRow()))
  387. ###########nacitam vrstvy z laymana do comboboxu
  388. url = self.URI+'/rest/'+self.laymanUsername+'/layers'
  389. r = requests.get(url = url)
  390. # print(r.content())
  391. data = r.json()
  392. for row in range(0, len(data)):
  393. self.dlg.comboBox_raster.addItem(data[row]['name'])
  394. self.dlg.pushButton_addRaster.clicked.connect(lambda: self.addExistingLayerToComposite(self.dlg.comboBox_raster.currentText()))
  395. #### nahrát mapy ze serveru do comboboxu
  396. url = self.URI+'/rest/'+self.laymanUsername+'/maps'
  397. r = requests.get(url = url)
  398. data = r.json()
  399. for row in range(0, len(data)):
  400. self.dlg.comboBox_loadMap.addItem(data[row]['name'])
  401. self.dlg.pushButton_loadMap.clicked.connect(lambda: self.readMapJson(self.dlg.comboBox_loadMap.currentText(), 'WMS'))
  402. self.dlg.pushButton_loadMapWFS.clicked.connect(lambda: self.readMapJson(self.dlg.comboBox_loadMap.currentText(), 'WFS'))
  403. self.dlg.pushButton_loadMapWFS.hide()
  404. self.dlg.pushButton_loadMap.hide()
  405. self.dlg.comboBox_loadMap.hide()
  406. #############
  407. self.dlg.pushButton.setEnabled(False)
  408. self.dlg.pushButton_deleteLayers.setEnabled(False)
  409. self.dlg.listWidget_listLayers.itemClicked.connect(self.showSmallThumbnail)
  410. # self.dlg.listWidget_listLayers.itemClicked.connect(self.showThumbnail)
  411. self.dlg.listWidget.itemClicked.connect(self.enableButton)
  412. self.dlg.pushButton_deleteLayers.clicked.connect(lambda: self.deteteLayerFromComposite(self.dlg.listWidget.currentRow(),self.dlg.listWidget_listLayers.currentRow(), self.dlg.listWidget_listLayers.currentItem().text()))
  413. self.dlg.show()
  414. self.dlg.pushButton_close.clicked.connect(lambda: self.dlg.close())
  415. self.dlg.pushButton_down.setStyleSheet("#pushButton_down {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_down:hover{background: #66ab27 ;}#pushButton_down:disabled{background: #64818b ;}")
  416. self.dlg.pushButton_up.setStyleSheet("#pushButton_up {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_up:hover{background: #66ab27 ;}#pushButton_up:disabled{background: #64818b ;}")
  417. self.dlg.pushButton.setStyleSheet("#pushButton {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton:hover{background: #66ab27 ;}#pushButton:disabled{background: #64818b ;}")
  418. self.dlg.pushButton_close.setStyleSheet("#pushButton_close {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_close:hover{background: #66ab27 ;}#pushButton_close:disabled{background: #64818b ;}")
  419. self.dlg.pushButton_deleteLayers.setStyleSheet("#pushButton_deleteLayers {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_deleteLayers:hover{background: #66ab27 ;}#pushButton_deleteLayers:disabled{background: #64818b ;}")
  420. self.dlg.pushButton_deleteMap.setStyleSheet("#pushButton_deleteMap {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_deleteMap:hover{background: #66ab27 ;}#pushButton_deleteMap:disabled{background: #64818b ;}")
  421. self.dlg.pushButton_addRaster.setStyleSheet("#pushButton_addRaster {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_addRaster:hover{background: #66ab27 ;}#pushButton_addRaster:disabled{background: #64818b ;}")
  422. self.dlg.pushButton_addMap.setStyleSheet("#pushButton_addMap {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_addMap:hover{background: #66ab27 ;}#pushButton_addMap:disabled{background: #64818b ;}")
  423. self.dlg.pushButton_editMeta.setStyleSheet("#pushButton_editMeta {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_editMeta:hover{background: #66ab27 ;}#pushButton_editMeta:disabled{background: #64818b ;}")
  424. self.dlg.pushButton_loadMapWFS.setStyleSheet("#pushButton_loadMapWFS {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_loadMapWFS:hover{background: #66ab27 ;}#pushButton_loadMapWFS:disabled{background: #64818b ;}")
  425. self.dlg.pushButton_loadMap.setStyleSheet("#pushButton_loadMap {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_loadMap:hover{background: #66ab27 ;}#pushButton_loadMap:disabled{background: #64818b ;}")
  426. self.dlg.setStyleSheet("#DialogBase {background: #f0f0f0 ;}")
  427. result = self.dlg.exec_()
  428. def run_DeleteMapDialog(self):
  429. self.dlg = DeleteMapDialog()
  430. self.refreshListWidgetMaps()
  431. self.dlg.pushButton.setEnabled(False)
  432. self.dlg.treeWidget.itemClicked.connect(self.enableButton)
  433. self.dlg.pushButton.clicked.connect(lambda: self.deleteMapFromServer(self.dlg.treeWidget.selectedItems()[0].text(0)))
  434. self.dlg.pushButton_close.clicked.connect(lambda: self.dlg.close())
  435. self.dlg.show()
  436. result = self.dlg.exec_()
  437. def run_ImportLayerDialog(self):
  438. self.dlg = ImportLayerDialog()
  439. # self.dlg.pushButton.clicked.connect(lambda: self.postRequest(self.dlg.treeWidget.currentItem().text(0)))
  440. self.dlg.pushButton.clicked.connect(lambda: self.callPostRequest(self.dlg.treeWidget.selectedItems()))
  441. self.dlg.pushButton.setEnabled(False)
  442. # self.dlg.treeWidget.itemClicked.connect(self.enableButton)
  443. self.dlg.treeWidget.itemPressed.connect(self.enableButtonImport)
  444. self.dlg.treeWidget.itemClicked.connect(self.onItemClicked)
  445. layers = QgsProject.instance().mapLayers().values()
  446. for layer in layers:
  447. if (layer.type() == QgsMapLayer.VectorLayer):
  448. layerType = 'vector layer'
  449. else:
  450. layerType = 'raster layer'
  451. item = QTreeWidgetItem([layer.name(), layerType])
  452. self.dlg.treeWidget.addTopLevelItem(item)
  453. self.dlg.setWindowModality(Qt.ApplicationModal)
  454. self.dlg.pushButton_close.setStyleSheet("#pushButton_close {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_close:hover{background: #66ab27 ;}")
  455. self.dlg.pushButton.setStyleSheet("#pushButton {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton:hover{background: #66ab27 ;}#pushButton:disabled{background: #64818b ;}")
  456. self.dlg.setStyleSheet("#DialogBase {background: #f0f0f0 ;}")
  457. self.dlg.show()
  458. self.dlg.pushButton_close.clicked.connect(lambda: self.dlg.close())
  459. result = self.dlg.exec_()
  460. def run_login(self):
  461. self.dlg = ConnectionManagerDialog()
  462. self.dlg.show()
  463. self.dlg.pushButton_Connect.setEnabled(False)
  464. if (os.path.isfile(os.getenv("HOME") + os.sep + ".layman" + os.sep +'layman_user.INI')):
  465. config = self.loadIni()
  466. if len(config['DEFAULT']['login']) > 0:
  467. self.Agrimail = config['DEFAULT']['login']
  468. self.dlg.pushButton_Connect.setEnabled(True)
  469. self.dlg.lineEdit_userName.setText(config['DEFAULT']['login'] + "@lesprojekt.cz")
  470. self.dlg.lineEdit_AgriID.setText(config['DEFAULT']['id'])
  471. else:
  472. try:
  473. os.makedirs(os.getenv("HOME") + os.sep + ".layman")
  474. except:
  475. print("layman directory already exists")
  476. self.dlg.lineEdit_userName.setText("@lesprojekt.cz")
  477. self.dlg.pushButton_Connect.setEnabled(True)
  478. self.dlg.pushButton_ttt.hide()
  479. self.dlg.lineEdit_userName.textChanged.connect(self.checkUsername)
  480. self.dlg.pushButton_close.clicked.connect(lambda: self.dlg.close())
  481. self.dlg.pushButton_Connect.clicked.connect(lambda: self.openAuthLiferayUrl())
  482. self.dlg.pushButton_Continue.clicked.connect(lambda: self.getToken())
  483. self.dlg.pushButton_Continue.setEnabled(False)
  484. self.dlg.pushButton_ttt.clicked.connect(lambda: self.printVariables())
  485. self.dlg.pushButton_close.setStyleSheet("#pushButton_close {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_close:hover{background: #66ab27 ;}")
  486. self.dlg.pushButton_Connect.setStyleSheet("#pushButton_Connect {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_Connect:hover{background: #66ab27 ;}")
  487. self.dlg.pushButton_Continue.setStyleSheet("#pushButton_Continue {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_Continue:hover{background: #66ab27 ;} #pushButton_Continue:disabled{background: #64818b ;}")
  488. self.dlg.setStyleSheet("#DialogBase {background: #f0f0f0 ;}")
  489. result = self.dlg.exec_()
  490. def run_AddMapDialog(self):
  491. url = self.URI+'/rest/'+self.laymanUsername+'/maps'
  492. print(url)
  493. r = requests.get(url = url)
  494. print(r.content)
  495. data = r.json()
  496. self.dlg = AddMapDialog()
  497. for row in range(0, len(data)):
  498. url = self.URI+'/rest/'+self.laymanUsername+'/maps/'+data[row]['name']+'/file'
  499. r = requests.get(url = url)
  500. # print(r.content)
  501. d = r.json()
  502. print(d['title'])
  503. # item = QTreeWidgetItem([data[row]['name'], data[row]['url'], data[row]['uuid']])
  504. item = QTreeWidgetItem([d['title']])
  505. self.dlg.treeWidget.addTopLevelItem(item)
  506. self.dlg.pushButton.setEnabled(False)
  507. self.dlg.pushButton_mapWFS.setEnabled(False)
  508. self.dlg.label_info.hide()
  509. self.dlg.treeWidget.itemClicked.connect(self.showThumbnailMap)
  510. self.dlg.treeWidget.itemClicked.connect(self.enableButton)
  511. self.dlg.treeWidget.itemClicked.connect(self.enableLoadMapButtons)
  512. self.dlg.pushButton.clicked.connect(lambda: self.readMapJson(self.dlg.treeWidget.selectedItems()[0].text(0), 'WMS'))
  513. self.dlg.pushButton_mapWFS.clicked.connect(lambda: self.readMapJson(self.dlg.treeWidget.selectedItems()[0].text(0), 'WFS'))
  514. self.dlg.pushButton_close.clicked.connect(lambda: self.dlg.close())
  515. self.dlg.pushButton_mapWFS.setStyleSheet("#pushButton_mapWFS {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_mapWFS:hover{background: #66ab27 ;}#pushButton_mapWFS:disabled{background: #64818b ;}")
  516. self.dlg.pushButton.setStyleSheet("#pushButton {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton:hover{background: #66ab27 ;}#pushButton:disabled{background: #64818b ;}")
  517. self.dlg.pushButton_close.setStyleSheet("#pushButton_close {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_close:hover{background: #66ab27 ;}#pushButton_close:disabled{background: #64818b ;}")
  518. self.dlg.setStyleSheet("#DialogBase {background: #f0f0f0 ;}")
  519. self.dlg.show()
  520. result = self.dlg.exec_()
  521. def run_AddLayerDialog(self):
  522. if not self.loadedInMemory:
  523. self.loadAllComposites()
  524. url = self.URI+'/rest/'+self.laymanUsername+'/layers'
  525. r = requests.get(url = url)
  526. data = r.json()
  527. self.dlg = AddLayerDialog()
  528. self.dlg.pushButton_layerRedirect.hide()
  529. self.dlg.pushButton_layerRedirect.setEnabled(False)
  530. self.dlg.pushButton.setEnabled(False)
  531. self.dlg.pushButton_wfs.setEnabled(False)
  532. self.dlg.pushButton_delete.setEnabled(False)
  533. for row in range(0, len(data)):
  534. item = QTreeWidgetItem([data[row]['name']])
  535. self.dlg.treeWidget.addTopLevelItem(item)
  536. self.dlg.pushButton_delete.clicked.connect(lambda: self.layerDelete(self.dlg.treeWidget.selectedItems()[0].text(0)))
  537. self.dlg.pushButton_layerRedirect.clicked.connect(lambda: self.layerInfoRedirect(self.dlg.treeWidget.selectedItems()[0].text(0)))
  538. self.dlg.pushButton.clicked.connect(lambda: self.readLayerJson(self.dlg.treeWidget.selectedItems()[0].text(0), "WMS"))
  539. self.dlg.pushButton_wfs.clicked.connect(lambda: self.readLayerJson(self.dlg.treeWidget.selectedItems()[0].text(0), "WFS"))
  540. self.dlg.treeWidget.itemClicked.connect(self.showThumbnail)
  541. self.dlg.treeWidget.itemClicked.connect(self.enableDeleteButton)
  542. self.dlg.pushButton_close.clicked.connect(lambda: self.dlg.close())
  543. self.dlg.setWindowModality(Qt.ApplicationModal)
  544. self.dlg.pushButton_delete.setStyleSheet("#pushButton_delete {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_delete:hover{background: #66ab27 ;}#pushButton_delete:disabled{background: #64818b ;}")
  545. self.dlg.pushButton_close.setStyleSheet("#pushButton_close {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_close:hover{background: #66ab27 ;}#pushButton_close:disabled{background: #64818b ;}")
  546. self.dlg.pushButton_delete.setStyleSheet("#pushButton_delete {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_delete:hover{background: #66ab27 ;}#pushButton_delete:disabled{background: #64818b ;}")
  547. self.dlg.pushButton.setStyleSheet("#pushButton {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton:hover{background: #66ab27 ;}#pushButton:disabled{background: #64818b ;}")
  548. self.dlg.pushButton_wfs.setStyleSheet("#pushButton_wfs {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} #pushButton_wfs:hover{background: #66ab27 ;}#pushButton_wfs:disabled{background: #64818b ;}")
  549. self.dlg.setStyleSheet("#DialogBase {background: #f0f0f0 ;}")
  550. self.dlg.show()
  551. result = self.dlg.exec_()
  552. def showExistingLayers(self,x):
  553. self.dlg.listWidget_listLayers.clear()
  554. for i in range (0,len(self.compositeList[self.dlg.listWidget.currentRow()]['layers'])):
  555. self.dlg.listWidget_listLayers.addItem(self.compositeList[self.dlg.listWidget.currentRow()]['layers'][i]['params']['LAYERS'])
  556. #--------------------------- slots---------------------
  557. # @QtCore.pyqtSlot(QtWidgets.QTreeWidgetItem, int)
  558. def onItemClicked(self, it, col):
  559. if (it.text(1) == 'raster layer'):
  560. self.dlg.pushButton.setEnabled(False)
  561. else:
  562. self.dlg.pushButton.setEnabled(True)
  563. def setExtent(self, it, col):
  564. layer = QgsProject.instance().mapLayersByName(it.text(0))
  565. ext = layer[0].extent()
  566. xmin = ext.xMinimum()
  567. xmax = ext.xMaximum()
  568. ymin = ext.yMinimum()
  569. ymax = ext.yMaximum()
  570. self.dlg.lineEdit_3.setText(str(xmin))
  571. self.dlg.lineEdit_4.setText(str(xmax))
  572. self.dlg.lineEdit_5.setText(str(ymin))
  573. self.dlg.lineEdit_6.setText(str(ymax))
  574. self.dlg.label_4.setText("Extent of canvas: " + it.text(0))
  575. def enableDeleteButton(self, item, col):
  576. self.dlg.pushButton.setEnabled(True)
  577. self.dlg.pushButton_wfs.setEnabled(True)
  578. self.dlg.pushButton_layerRedirect.setEnabled(True)
  579. self.dlg.pushButton_delete.setEnabled(True)
  580. def enableButton(self, item, col):
  581. self.dlg.pushButton.setEnabled(True)
  582. self.dlg.pushButton_mapWFS.setEnabled(True)
  583. self.dlg.pushButton_deleteLayers.setEnabled(True)
  584. self.dlg.pushButton_editMeta.setEnabled(True)
  585. self.dlg.pushButton_addRaster.setEnabled(True)
  586. def checkUsername(self, name):
  587. n = name.split("@")
  588. if(len(n[0]) > 0):
  589. self.dlg.pushButton_Connect.setEnabled(True)
  590. self.Agrimail = n[0]
  591. else:
  592. self.dlg.pushButton_Connect.setEnabled(False)
  593. def checkLoadedMap(self, name):
  594. print(self.compositeList)
  595. loaded = False
  596. for i in range (0, len(self.compositeList)):
  597. if self.compositeList[i]['name'] == name:
  598. loaded = True
  599. return loaded
  600. def enableLoadMapButtons(self, item):
  601. self.dlg.pushButton_mapWFS.setEnabled(True)
  602. def enableButtonImport(self, item, column):
  603. if (len(self.dlg.treeWidget.selectedItems()) > 0):
  604. self.dlg.pushButton.setEnabled(True)
  605. else:
  606. self.dlg.pushButton.setEnabled(False)
  607. def enableButton(self, item):
  608. try: ## addMap nemá combobox
  609. if (self.dlg.mMapLayerComboBox.count() > 0):
  610. self.dlg.pushButton.setEnabled(True)
  611. self.dlg.pushButton_mapWFS.setEnabled(True)
  612. else:
  613. self.dlg.pushButton.setEnabled(False)
  614. except:
  615. self.dlg.pushButton.setEnabled(True) ## addMap nemá combobox nastavujeme funkční tlačítko
  616. try:
  617. self.dlg.pushButton_deleteLayers.setEnabled(True) # pro formulář importMap
  618. except:
  619. pass
  620. def listCompositeLayers(self, it):
  621. self.dlg.listWidget_listLayers2.clear()
  622. for i in range (0,len(self.compositeList[self.dlg.listWidget.currentRow()]['layers'])):
  623. self.dlg.listWidget_listLayers2.addItem(self.compositeList[self.dlg.listWidget.currentRow()]['layers'][i]['params']['LAYERS'])
  624. def deleteLayerThrowCompositions(self, name):
  625. for x in range (0,len(self.compositeList)):
  626. for i in range (0,len(self.compositeList[x]['layers'])):
  627. if (name == self.compositeList[x]['layers'][i]['params']['LAYERS']):
  628. inComposite = True
  629. print(self.compositeList[x]['layers'][i]['params']['LAYERS'])
  630. self.deteteLayerFromComposite(x, i, name)
  631. def checkLayersInComopsitions(self, name):
  632. inComposite = False
  633. for x in range (0,len(self.compositeList)):
  634. for i in range (0,len(self.compositeList[x]['layers'])):
  635. if (name == self.compositeList[x]['layers'][i]['params']['LAYERS']):
  636. inComposite = True
  637. print(self.compositeList[x]['layers'][i]['params']['LAYERS'])
  638. return inComposite
  639. def refreshCompositeList(self):
  640. self.dlg.listWidget.clear()
  641. print (len(self.compositeList))
  642. for i in range (0, len(self.compositeList)):
  643. # self.dlg.listWidget.addItem(self.compositeList[i]['name'])
  644. self.dlg.listWidget.addItem(self.compositeList[i]['title'])
  645. def refreshLayerList(self):
  646. self.dlg.listWidget_listLayers.clear()
  647. for i in range (0,len(self.compositeList[self.dlg.listWidget.currentRow()]['layers'])):
  648. self.dlg.listWidget_listLayers.addItem(self.compositeList[self.dlg.listWidget.currentRow()]['layers'][i]['params']['LAYERS'])
  649. print(self.compositeList[self.dlg.listWidget.currentRow()]['layers'][i]['params']['LAYERS'])
  650. def refreshListWidgetMaps(self):
  651. self.dlg.treeWidget.clear()
  652. url = self.URI+'/rest/'+self.laymanUsername+'/maps'
  653. print(url)
  654. r = requests.get(url = url)
  655. #print(r.content())
  656. data = r.json()
  657. for row in range(0, len(data)):
  658. item = QTreeWidgetItem([data[row]['name']])
  659. self.dlg.treeWidget.addTopLevelItem(item)
  660. def showAddMapDialog(self):
  661. self.old_dlg = self.dlg
  662. self.run_CreateCompositeDialog(True)
  663. ### connect
  664. def afterCloseCompositeDialog(self):
  665. self.dlg = self.old_dlg
  666. self.refreshCompositeList()
  667. def afterCloseEditMapDialog(self):
  668. self.dlg = self.old_dlg
  669. def showEditMapDialog(self, x):
  670. self.old_dlg = self.dlg
  671. self.run_EditMap(x)
  672. ############### pravdepodobne půjde smazat
  673. def refreshItems(self):
  674. self.dlg.listWidget_listLayers.clear()
  675. for i in range (0,len(self.compositeList)):
  676. # self.dlg.listWidget_listLayers.addItem(self.compositeList[i]['name'])
  677. self.dlg.listWidget_listLayers.addItem(self.compositeList[i]['title'])
  678. def saveEditedToComposite(self,x):
  679. self.compositeList[x]['abstract'] = self.dlg.lineEdit_abstract.text()
  680. self.compositeList[x]['title'] = self.dlg.lineEdit_title.text()
  681. def layerDelete(self, name):
  682. if (self.checkLayersInComopsitions(name) == True):
  683. msgbox = QMessageBox(QMessageBox.Question, "Delete layer", "This layers is included in other compositions. It will be delete from every composition.")
  684. msgbox.addButton(QMessageBox.Yes)
  685. msgbox.addButton(QMessageBox.No)
  686. msgbox.setDefaultButton(QMessageBox.No)
  687. reply = msgbox.exec()
  688. if (reply == QMessageBox.Yes):
  689. url = self.URI+'/rest/'+self.laymanUsername+'/layers/'+name
  690. response = requests.delete(url, headers = self.authHeader)
  691. print(response.content)
  692. print(response)
  693. self.addLayerRefresh()
  694. self.deleteLayerThrowCompositions(name)
  695. else:
  696. url = self.URI+'/rest/'+self.laymanUsername+'/layers/'+name
  697. response = requests.delete(url, headers = self.authHeader)
  698. print(response.content)
  699. print(response)
  700. self.addLayerRefresh()
  701. def addLayerRefresh(self):
  702. self.dlg.treeWidget.clear()
  703. url = self.URI+'/rest/'+self.laymanUsername+'/layers'
  704. r = requests.get(url = url)
  705. data = r.json()
  706. for row in range(0, len(data)):
  707. item = QTreeWidgetItem([data[row]['name']])
  708. self.dlg.treeWidget.addTopLevelItem(item)
  709. def checkIfMapExist(self, name):
  710. url = "http://layman.lesprojekt.cz/rest/"+self.laymanUsername+"/maps/"+str(name)+"/file"
  711. r = requests.get(url)
  712. print(r.content)
  713. if (r.status_code == 404):
  714. return False
  715. else:
  716. return True
  717. def reorderLayers(self, pos, order, x):
  718. self.compositeList[x]['layers']
  719. i = 0
  720. for lay in self.compositeList[x]['layers']:
  721. #print (lay)
  722. if i == pos:
  723. pom = self.compositeList[x]['layers'][i]
  724. self.compositeList[x]['layers'][i] = self.compositeList[x]['layers'][i + order]
  725. self.compositeList[x]['layers'][i + order] = pom
  726. i = i + 1
  727. print(self.compositeList[x]['layers'])
  728. self.importMap(x, 'mod')
  729. self.refreshLayerList()
  730. self.dlg.listWidget_listLayers.setCurrentRow(pos + order)
  731. def showThumbnail(self, it):
  732. try:
  733. layer = it.text(0) ##pro QTreeWidget
  734. except:
  735. layer = it.text()##pro listWidget
  736. try:
  737. url = self.URI+'/rest/' +self.laymanUsername+'/layers/'+str(layer)+'/thumbnail'
  738. data = urlopen(url).read()
  739. pixmap = QPixmap(200, 200)
  740. pixmap.loadFromData(data)
  741. smaller_pixmap = pixmap.scaled(200, 200, Qt.KeepAspectRatio, Qt.FastTransformation)
  742. self.dlg.label_thumbnail.setPixmap(smaller_pixmap)
  743. self.dlg.label_thumbnail.setAlignment(Qt.AlignCenter)
  744. except:
  745. self.dlg.label_thumbnail.setText(' Unable to load thumbnail.')
  746. def showSmallThumbnail(self, it):
  747. try:
  748. layer = it.text(0) ##pro QTreeWidget
  749. except:
  750. layer = it.text()##pro listWidget
  751. try:
  752. url = self.URI+'/rest/'+self.laymanUsername+'/layers/'+str(layer)+'/thumbnail'
  753. data = urlopen(url).read()
  754. pixmap = QPixmap(170, 170)
  755. pixmap.loadFromData(data)
  756. smaller_pixmap = pixmap.scaled(170, 170, Qt.KeepAspectRatio, Qt.FastTransformation)
  757. self.dlg.label_thumbnail.setPixmap(smaller_pixmap)
  758. self.dlg.label_thumbnail.setAlignment(Qt.AlignCenter)
  759. except:
  760. self.dlg.label_thumbnail.setText(' Unable to load thumbnail.')
  761. def showThumbnailMap(self, it):
  762. try:
  763. map = it.text(0) ##pro QTreeWidget
  764. except:
  765. map = it.text()##pro listWidget
  766. try:
  767. map = self.removeUnacceptableChars(str(map))
  768. url = self.URI+'/rest/'+self.laymanUsername+'/maps/'+str(map)+'/thumbnail'
  769. print(url)
  770. data = urlopen(url).read()
  771. pixmap = QPixmap(100, 100)
  772. pixmap.loadFromData(data)
  773. smaller_pixmap = pixmap.scaled(100, 100, Qt.KeepAspectRatio, Qt.FastTransformation)
  774. self.dlg.label_thumbnail.setPixmap(smaller_pixmap)
  775. # self.dlg.label_thumbnail.setAlignment(Qt.AlignCenter)
  776. except:
  777. self.dlg.label_thumbnail.setText(' Unable to load thumbnail.')
  778. def checkNameCreateMap(self):
  779. text = self.dlg.lineEdit_2.text()
  780. text = self.removeUnacceptableChars(text)
  781. print(text)
  782. if (self.checkIfMapExist(text)):
  783. self.dlg.pushButton_CreateComposition.setEnabled(False)
  784. self.dlg.label_info.show()
  785. else:
  786. self.dlg.pushButton_CreateComposition.setEnabled(True)
  787. self.dlg.label_info.hide()
  788. #----------------------------------------------------------
  789. def readLayerJson(self,layerName, service):
  790. if self.checkLayerOnLayman(layerName):
  791. url = self.URI+'/rest/'+self.laymanUsername+'/layers/'+layerName
  792. print (url)
  793. r = requests.get(url = url)
  794. data = r.json()
  795. if (service == "WMS"):
  796. print("loading WMS")
  797. print (data['wms']['url'])
  798. wmsUrl = data['wms']['url']
  799. format = 'png'
  800. epsg = 'EPSG:4326'
  801. self.loadWms(wmsUrl, layerName, format, epsg)
  802. if (service == "WFS"):
  803. wfsUrl = data['wfs']['url']
  804. print("loading WFS")
  805. self.loadWfs(wfsUrl, layerName)
  806. else:
  807. QMessageBox.information(None, "Layman", "Something went wrong with this layer: "+layerName)
  808. def loadAllComposites(self):
  809. url = self.URI+'/rest/' + self.laymanUsername + '/maps'
  810. try:
  811. r = requests.get(url = url)
  812. except:
  813. QMessageBox.information(None, "Error", "Connection with server failed!.")
  814. data = r.json()
  815. print(data)
  816. for i in data:
  817. print(i['name'])
  818. url = self.URI+'/rest/' + self.laymanUsername + '/maps/'+i['name']+'/file'
  819. r = requests.get(url = url)
  820. map = r.json()
  821. self.compositeList.append (map)
  822. self.loadedInMemory = True
  823. def readMapJson(self,name, service):
  824. nameWithDiacritics = name
  825. name = self.removeUnacceptableChars(name)
  826. print(name)
  827. print("readMapJson " + name)
  828. if not self.checkLoadedMap(name): ## je nactena mapa?
  829. url = self.URI+'/rest/'+self.laymanUsername+'/maps/'+name +'/file'
  830. r = requests.get(url = url)
  831. data = r.json()
  832. print(data)
  833. self.addComposite(data,service, name)
  834. try:
  835. self.refreshCompositeList() ## pouze pro import Form
  836. except:
  837. pass
  838. else:
  839. url = self.URI+'/rest/'+self.laymanUsername+'/maps/'+name +'/file'
  840. r = requests.get(url = url)
  841. data = r.json()
  842. self.loadService(data,service, name)
  843. # QMessageBox.information(None, "Message", "This map is already loaded in memory. Loading only layers to canvas")
  844. def deleteMapFromServer(self,name):
  845. url = self.URI+'/rest/'+self.laymanUsername+'/maps/'+name
  846. response = requests.delete(url, headers = self.authHeader)
  847. print(response)
  848. QMessageBox.information(None, "Message", "Composition deleted sucessfully.")
  849. # self.refreshMapList()
  850. self.refreshListWidgetMaps() ## pro treewidget
  851. def deleteMap(self,name, x):
  852. name = self.removeUnacceptableChars(name)
  853. url = self.URI+'/rest/'+self.laymanUsername+'/maps/'+name
  854. response = requests.delete(url, headers = self.authHeader)
  855. print(response)
  856. print(response.content)
  857. QMessageBox.information(None, "Message", "Composition deleted sucessfully.")
  858. del (self.compositeList[x])
  859. self.refreshCompositeList()## pro import map form
  860. self.dlg.listWidget_listLayers.clear()
  861. def deleteMapandLayers(self,name, x):
  862. try:
  863. url = self.URI+'/rest/'+self.laymanUsername+'/maps/'+name
  864. response = requests.delete(url, headers = self.authHeader)
  865. QMessageBox.information(None, "Message", "Composition deleted sucessfully.")
  866. except:
  867. QMessageBox.information(None, "Warning", "Deleting composition failed.")
  868. #self.refreshItems()
  869. self.deleteMapFromCanvas(x)
  870. self.dlg.listWidget_listLayers.clear()
  871. self.dlg.pushButton_deleteLayers.setEnabled(False)
  872. self.dlg.pushButton.setEnabled(False)
  873. self.dlg.pushButton_deleteMap.setEnabled(False)
  874. self.refreshCompositeList()
  875. def deleteMapFromCanvas(self, x):
  876. for i in range (0,len(self.compositeList[x]['layers'])): #[0]['params']['LAYERS']:
  877. name = self.compositeList[x]['layers'][i]['params']['LAYERS']
  878. lay = QgsProject.instance().mapLayersByName(name)[0]
  879. if (lay != None and lay.type() != QgsMapLayer.VectorLayer):
  880. QgsProject.instance().removeMapLayer(lay.id())
  881. del (self.compositeList[x])
  882. def deleteLayerFromCanvas(self, name):
  883. lay = QgsProject.instance().mapLayersByName(name)[0]
  884. if (lay != None and lay.type() != QgsMapLayer.VectorLayer):
  885. QgsProject.instance().removeMapLayer(lay.id())
  886. def getSelectedLayers(self):
  887. arr = self.dlg.listWidget_listLayers2.selectedItems()
  888. layers = []
  889. for a in arr:
  890. layer = QgsProject.instance().mapLayersByName(a.text())
  891. for lay in layer:
  892. if (lay.type() == QgsMapLayer.VectorLayer):
  893. layers.append(lay)
  894. return layers
  895. def run_getLayer(self):
  896. """Run method that performs all the real work"""
  897. # Create the dialog with elements (after translation) and keep reference
  898. # Only create GUI ONCE in callback, so that it will only load when the plugin is started
  899. if self.first_start == True:
  900. self.first_start = False
  901. self.dlgGetLayers.items.clear()
  902. # show the dialog
  903. self.dlgGetLayers.show()
  904. data= self.getExistingLayers()
  905. for x in range(len(data)):
  906. print(data[x]['name'])
  907. self.dlgGetLayers.items.addItem(data[x]['name'])
  908. # Run the dialog event loop
  909. print (self.dlgGetLayers)
  910. result = self.dlgGetLayers.exec_()
  911. # See if OK was pressed
  912. if result:
  913. # Do something useful here - delete the line containing pass and
  914. # substitute with your code.
  915. print ("test tlacitka")
  916. def onClosePlugin(self):
  917. """Cleanup necessary items here when plugin dockwidget is closed"""
  918. #print "** CLOSING Atlas"
  919. # disconnects
  920. self.dockwidget.closingPlugin.disconnect(self.onClosePlugin)
  921. self.pluginIsActive = False
  922. def unload(self):
  923. """Removes the plugin menu item and icon from QGIS GUI."""
  924. #print "** UNLOAD Atlas"
  925. for action in self.actions:
  926. self.iface.removePluginMenu(
  927. self.tr(u'&Atlas'),
  928. action)
  929. self.iface.removeToolBarIcon(action)
  930. # remove the toolbar
  931. del self.toolbar
  932. #--------------------------------------------------------------------------
  933. def bounds(self, layers):
  934. extent = None
  935. for layer in layers:
  936. if layer.type() == 0:
  937. transform = QgsCoordinateTransform(layer.crs(), QgsCoordinateReferenceSystem('EPSG:4326'), QgsProject.instance()) # WGS 84 / UTM zone 33N
  938. try:
  939. layerExtent = transform.transform(layer.extent())
  940. except QgsCsException:
  941. print("exception in transform layer srs")
  942. layerExtent = QgsRectangle(-180, -90, 180, 90)
  943. if extent is None:
  944. extent = layerExtent
  945. else:
  946. extent.combineExtentWith(layerExtent)
  947. return (extent.xMinimum(), extent.yMinimum(), extent.xMaximum(), extent.yMaximum())
  948. def copySymbols(self,symbol, tempPath, fileNames):
  949. for i in range(symbol.symbolLayerCount()):
  950. sl = symbol.symbolLayer(i)
  951. if isinstance(sl, QgsSvgMarkerSymbolLayer):
  952. symbolPath = sl.path();
  953. shutil.copy(symbolPath, tempPath)
  954. print("Copying " + str(sl.path()))
  955. fileNames.append(tempPath + os.sep + os.path.basename(symbolPath))
  956. else:
  957. print("Ignoring " + str(sl))
  958. def setDefaultExtent (self, ext):
  959. self.dlg.lineEdit_3.setText(str(ext.xMinimum()))
  960. self.dlg.lineEdit_4.setText(str(ext.xMaximum()))
  961. self.dlg.lineEdit_5.setText(str(ext.yMinimum()))
  962. self.dlg.lineEdit_6.setText(str(ext.yMaximum()))
  963. def loadJsonLayer(self, fileName):
  964. name = (os.path.splitext(os.path.basename(fileName))[0])
  965. vlayer = QgsVectorLayer(fileName, name,"ogr")
  966. sldPath = os.path.splitext(fileName[:-4])[0] + ".sld"
  967. print (sldPath)
  968. exists = os.path.isfile(sldPath)
  969. if (exists):
  970. vlayer.loadSldStyle(os.path.splitext(fileName)[0]+ ".sld")
  971. print("sld loaded")
  972. QgsProject.instance().addMapLayer(vlayer)
  973. vlayer.triggerRepaint()
  974. def checkLoadedLayer(self):
  975. layers = iface.layerTreeView().selectedLayers()
  976. if (len(layers) > 0):
  977. self.json_export()
  978. else:
  979. QMessageBox.information(None, "Message", "You must load layer first!")
  980. def getEmptyComposite(self, compositeName, compositeTitle):
  981. compositeEPSG = "epsg:4326"
  982. if (self.dlg.lineEdit_3.text() == "" or self.dlg.lineEdit_4.text() == "" or self.dlg.lineEdit_5.text() == "" or self.dlg.lineEdit_6.text() == ""):
  983. ext = iface.mapCanvas().extent()
  984. xmin = ext.xMinimum()
  985. xmax = ext.xMaximum()
  986. ymin = ext.yMinimum()
  987. ymax = ext.yMaximum()
  988. else:
  989. xmin = self.dlg.lineEdit_3.text()
  990. xmax = self.dlg.lineEdit_4.text()
  991. ymin = self.dlg.lineEdit_5.text()
  992. ymax = self.dlg.lineEdit_6.text()
  993. abstract = self.dlg.lineEdit_7.text()
  994. comp = {"abstract":abstract,"center":[1672068,6568819],"current_base_layer":{"title":"Composite_base_layer"},"extent":[str(xmin),str(ymin),str(xmax),str(ymax)],"groups":{"guest":"w"},"layers":[],"name":compositeName,"projection":compositeEPSG,"scale":1,"title":compositeTitle,"units":"m","user":{"email":"","name":self.laymanUsername}}
  995. self.dlg.close()
  996. iface.messageBar().pushWidget(iface.messageBar().createMessage("Atlas:", " Kompozice " + compositeName + " byla úspešně vytvořena."), Qgis.Success, duration=3)
  997. return comp
  998. def loadLocalFile(self):
  999. options = QFileDialog.Options()
  1000. dialog = QFileDialog()
  1001. dialog.setStyleSheet("QPushButton {color: #fff !important;text-transform: uppercase; text-decoration: none; background: #72c02c; padding: 20px; border-radius: 50px; display: inline-block; border: none;transition: all 0.4s ease 0s;} QPushButton:hover{background: #66ab27 ;}QPushButton:disabled{background: #64818b ;}");
  1002. fileName = dialog.getOpenFileName(None,"Load file", "","GeoJson Files (*.geojson);;Json Files (*.json)", options=options)
  1003. print ("načítám json ze souboru:" + fileName[0])
  1004. self.loadJsonLayer(fileName[0])
  1005. def modifyMap(self, x):
  1006. name = self.removeUnacceptableChars(self.dlg.lineEdit_title.text())
  1007. #self.compositeList[x]['name'] = self.dlg.lineEdit_name.text()
  1008. self.compositeList[x]['name'] = name
  1009. self.compositeList[x]['abstract'] = self.dlg.lineEdit_abstract.text()
  1010. self.compositeList[x]['title'] = self.dlg.lineEdit_title.text()
  1011. print(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'])
  1012. oldName = self.dlg.lineEdit_name.text()
  1013. response = requests.delete(self.URI+'/rest/'+self.laymanUsername+'/maps/'+oldName,headers = self.authHeader)
  1014. print(response.content)
  1015. self.dlg.close()
  1016. self.afterCloseEditMapDialog()
  1017. self.importMap(x, 'mod')
  1018. try:
  1019. self.refreshCompositeList()
  1020. iface.messageBar().pushWidget(iface.messageBar().createMessage("Import:", " Map metadata was saved successfully."), Qgis.Success, duration=3)
  1021. except:
  1022. pass
  1023. def saveLocalFile(self):
  1024. layer = self.iface.activeLayer()
  1025. path = iface.activeLayer().dataProvider().dataSourceUri()
  1026. path = path.split("|")[0].replace("'","")
  1027. if (layer == None):
  1028. QMessageBox.information(None, "Message", "You must load layer first!")
  1029. else:
  1030. # defaultDir = QDir().home()
  1031. defaultDir = os.path.dirname(path)
  1032. print(defaultDir)
  1033. print(defaultDir +os.sep+ str(layer.name()) + ".geojson")
  1034. print(QDir().homePath() +os.sep+ str(layer.name()) + ".geojson")
  1035. dialog = QFileDialog()
  1036. dialog.setFileMode(1)
  1037. dialog.setDirectory(defaultDir)
  1038. print(dialog.directory().path())
  1039. # layer_name = dialog.getSaveFileName(None, "Save file", QDir().homePath() +os.sep+ str(layer.name()) + ".geojson", "*.geojson")
  1040. layer_name = dialog.getSaveFileName(None, "Save file", defaultDir +os.sep+ str(layer.name()) + ".geojson", "*.geojson")
  1041. self.json_export_local(layer_name[0], layer)
  1042. def json_export_local(self, layer_name, lay):
  1043. import shutil
  1044. filePath = self.getTempPath(layer_name)
  1045. ogr_driver_name = "GeoJSON"
  1046. project = QgsProject.instance()
  1047. fileNames = []
  1048. ## zde musí být zajištěna vektorová vrstva
  1049. layer = lay
  1050. layerType = layer.type()
  1051. if layerType == QgsMapLayer.VectorLayer:
  1052. renderer = layer.renderer()
  1053. hasIcon = False
  1054. if isinstance(renderer, QgsSingleSymbolRenderer):
  1055. self.copySymbols(renderer.symbol(), tempfile.gettempdir(), fileNames)
  1056. hasIcon = True
  1057. layerCrs = qgis.utils.iface.activeLayer().crs().authid()
  1058. crs = QgsCoordinateReferenceSystem(layerCrs)
  1059. result2 = qgis.core.QgsVectorFileWriter.writeAsVectorFormat(layer, layer_name, "utf-8", crs, ogr_driver_name) # export jsonu do souboru
  1060. print(result2)
  1061. if(result2[0] == 2): ## testujeme zda není soubor otevřený v jiném procesu / návratový kod 2
  1062. tempFile = self.getTempPath(os.path.basename(layer_name.replace(".geojson", "") ))
  1063. print(tempFile)
  1064. #QMessageBox.information(None, "Layman", "It is not possible overwrite this file. File is already open in other process.")
  1065. sld_temp_filename = tempFile.replace("geojson", "sld")
  1066. layer.saveSldStyle(sld_temp_filename)
  1067. result2 = qgis.core.QgsVectorFileWriter.writeAsVectorFormat(layer, tempFile, "utf-8", crs, ogr_driver_name)
  1068. print(result2)
  1069. if(result2[0] == 2):
  1070. QMessageBox.information(None, "Layman", "It is not possible overwrite this file. File is already open in other process.")
  1071. return
  1072. if os.path.basename(layer_name.replace(".geojson", "")) != '':
  1073. QgsProject.instance().removeMapLayer(layer.id())
  1074. print(tempFile, layer_name)
  1075. shutil.copy(tempFile, layer_name)
  1076. shutil.copy(sld_temp_filename, layer_name.replace(".geojson", ".sld"))
  1077. self.loadJsonLayer(layer_name)
  1078. else:
  1079. sld_filename = layer_name.replace("geojson", "sld")
  1080. result3 = False
  1081. layer.saveSldStyle(sld_filename)
  1082. def json_export(self, layer_name):
  1083. filePath = self.getTempPath(layer_name)
  1084. ogr_driver_name = "GeoJSON"
  1085. project = QgsProject.instance()
  1086. fileNames = []
  1087. layer = QgsProject.instance().mapLayersByName(layer_name)[0]
  1088. layerType = layer.type()
  1089. if layerType == QgsMapLayer.VectorLayer:
  1090. renderer = layer.renderer()
  1091. hasIcon = False
  1092. if isinstance(renderer, QgsSingleSymbolRenderer):
  1093. self.copySymbols(renderer.symbol(), tempfile.gettempdir(), fileNames)
  1094. hasIcon = True
  1095. #layerCrs = qgis.utils.iface.activeLayer().crs().authid()
  1096. layerCrs = layer.crs().authid()
  1097. crs = QgsCoordinateReferenceSystem(layerCrs)# původně bylo
  1098. layer_filename = filePath
  1099. # crs = QgsCoordinateReferenceSystem(4326, QgsCoordinateReferenceSystem.EpsgCrsId)
  1100. ##crs = layer.crs().authid()
  1101. # tempOut = layer_filename.replace(".geojson", "_2.geojson")
  1102. result2 = qgis.core.QgsVectorFileWriter.writeAsVectorFormat(layer, layer_filename, "utf-8", crs, ogr_driver_name) # export jsonu do souboru
  1103. ##transformace
  1104. # input = tempOut
  1105. # out = layer_filename
  1106. # try:
  1107. # os.remove(layer_filename)
  1108. # except:
  1109. # pass
  1110. ## print("","-f", "Geojson", "-s_srs", layerCrs, "-t_srs", "epsg:4326", out, input)
  1111. # print("convert")
  1112. # print(main(["","-f", ogr_driver_name, "-s_srs", "EPSG:5514", "-t_srs", "EPSG:4326", out, input])) ## ogr2ogr
  1113. ##
  1114. sld_filename = filePath.replace("geojson", "sld")
  1115. result3 = False
  1116. layer.saveSldStyle(sld_filename)
  1117. def checkValidAttributes(self, layer_name):
  1118. layers = QgsProject.instance().mapLayersByName(layer_name)
  1119. if len(layers) > 1:
  1120. for l in layers:
  1121. if (isinstance(l, QgsVectorLayer)):
  1122. print("xxx")
  1123. layers.clear()
  1124. layers.append(l)
  1125. break
  1126. isValid = True
  1127. pom =layers[0].getFeature(1)
  1128. for nam in pom.fields().names():
  1129. if (nam == ''):
  1130. isValid = False
  1131. return isValid
  1132. def checkWgsExtent(self, layer):
  1133. WgsXmax = 180
  1134. WgsXmin = -180
  1135. WgsYmax = 90
  1136. WgsYmin = -90
  1137. extent = layer.extent()
  1138. if (extent.xMaximum() > WgsXmax or extent.xMinimum() < WgsXmin or extent.yMaximum() > WgsYmax or extent.yMinimum() < WgsYmin ):
  1139. return False
  1140. else:
  1141. return True
  1142. def callPostRequest(self, layers):
  1143. for item in layers:
  1144. print (item.text(0))
  1145. self.postRequest(item.text(0))
  1146. #def transformLayer2(self, layer):
  1147. # pass
  1148. def transformLayer(self, layer):
  1149. layer = iface.activeLayer()
  1150. crsSrc = layer.crs()
  1151. print(crsSrc.authid())
  1152. crsDest = QgsCoordinateReferenceSystem(4326, QgsCoordinateReferenceSystem.EpsgCrsId)
  1153. print(crsSrc, crsDest)
  1154. xform = QgsCoordinateTransform(crsSrc, crsDest, 5514, 4326)
  1155. feats=[]
  1156. if (layer.geometryType() == 0):
  1157. destLayer = QgsVectorLayer("Point?crs=EPSG:4326","lay","memory")
  1158. if (layer.geometryType() == 1):
  1159. destLayer = QgsVectorLayer("Linestring?crs=EPSG:4326","lay","memory")
  1160. if (layer.geometryType() == 2):
  1161. destLayer = QgsVectorLayer("Polygon?crs=EPSG:4326","lay","memory")
  1162. for f in layer.getFeatures():
  1163. g = f.geometry()
  1164. g.transform(xform)
  1165. print(g)
  1166. f.setGeometry(g)
  1167. feats.append(f)
  1168. destLayer.dataProvider().addFeatures(feats)
  1169. return destLayer
  1170. def postRequest(self, layer_name):
  1171. nameCheck = True
  1172. validExtent = True
  1173. layers = QgsProject.instance().mapLayersByName(layer_name)
  1174. layer_name = self.removeUnacceptableChars(layer_name)
  1175. layers[0].setName(layer_name)
  1176. if len(layers) > 1:
  1177. for l in layers:
  1178. if (isinstance(l, QgsVectorLayer)):
  1179. layers.clear()
  1180. layers.append(l)
  1181. break
  1182. if not (re.match('[a-zA-Z_]{1}', layer_name)): ## nesmí být nesmysl v názvu na prvním místě
  1183. QMessageBox.information(None, "Layman", "Diacritics or number in first character is not allowed.")
  1184. nameCheck = False
  1185. if not (re.match('[a-zA-Z0-9]', layer_name)): ## není povolena diakritika
  1186. QMessageBox.information(None, "Layman", "Diacritics is not allowed.")
  1187. nameCheck = False
  1188. print(layer_name)
  1189. #if (layers[0].crs().authid() != 'EPSG:4323'):
  1190. # layers[0] = self.transformLayer(layers[0])
  1191. if not self.checkWgsExtent(layers[0]):
  1192. QMessageBox.information(None, "Layman", "Extent of layer is out of WGS 84 range. (EPSG: 4326)")
  1193. validExtent = False
  1194. if (nameCheck and validExtent):
  1195. crs = layers[0].crs().authid()
  1196. crs = "EPSG:4326"
  1197. data = { 'name' : str(layer_name), 'title' : str(layer_name), 'crs' : str(crs) }
  1198. if (self.checkValidAttributes(layer_name)):
  1199. if (self.checkExistingLayer(layer_name)):
  1200. print("vrstva již existuje")
  1201. msgbox = QMessageBox(QMessageBox.Question, "Layman", "Layer "+layer_name+" already exists in server. Do you want overwrite it´s geometry?")
  1202. msgbox.addButton(QMessageBox.Yes)
  1203. msgbox.addButton(QMessageBox.No)
  1204. msgbox.setDefaultButton(QMessageBox.No)
  1205. reply = msgbox.exec()
  1206. if (reply == QMessageBox.Yes):
  1207. print("vrstva již existuje")
  1208. self.json_export(layer_name)
  1209. geoPath = self.getTempPath(layer_name)
  1210. #try:
  1211. if (os.path.getsize(geoPath) > self.CHUNK_SIZE):
  1212. self.postInChunks(layer_name, "patch")
  1213. else:
  1214. self.patchLayer(layer_name, data)
  1215. iface.messageBar().pushWidget(iface.messageBar().createMessage("Import:", " Layer " + layer_name + " was imported successfully."), Qgis.Success, duration=3)
  1216. #except:
  1217. # iface.messageBar().pushWidget(iface.messageBar().createMessage("Import:", " Layer " + layer_name + " was not imported successfully."), Qgis.Warning, duration=3)
  1218. else:
  1219. pass
  1220. else:
  1221. self.layerName = layer_name
  1222. # if (self.checkEpsg(layer_name)):
  1223. if(True): ##Pravděpodobně nebude třeba testovat EPSG, pokud se detekuje, je vrstva transformována.
  1224. # print (layer_name)
  1225. self.json_export(layer_name)
  1226. sldPath = self.getTempPath(self.layerName).replace("geojson", "sld")
  1227. geoPath = self.getTempPath(self.layerName)
  1228. if (os.path.getsize(geoPath) > self.CHUNK_SIZE):
  1229. self.postInChunks(layer_name, "post")
  1230. else:
  1231. if(os.path.isfile(sldPath)): ## existuje sld?
  1232. files = [('file', open(geoPath, 'rb')), ('sld', open(sldPath, 'rb'))]
  1233. else:
  1234. files = {'file': (geoPath, open(geoPath, 'rb')),}
  1235. print(files)
  1236. response = requests.post(self.URI+'/rest/'+self.laymanUsername+'/layers', files=files, data = data, headers = self.authHeader)
  1237. print(response.content)
  1238. print(response.status_code)
  1239. if response.status_code == 200:
  1240. iface.messageBar().pushWidget(iface.messageBar().createMessage("Import:", " Layer " + layer_name + " was imported successfully."), Qgis.Success, duration=3)
  1241. else:
  1242. iface.messageBar().pushWidget(iface.messageBar().createMessage("Import:", " Layer " + layer_name + " was not imported."), Qgis.Warning, duration=3)
  1243. else:
  1244. # QMessageBox.information(None, "Atlas", "Vrstva "+layer_name+" nelze nahrát. Použíjte EPSG:4326 nebo EPSG:3857")
  1245. QMessageBox.information(None, "Layman", "Use EPSG:4326")
  1246. else:
  1247. QMessageBox.information(None, "Layman", "Layer "+layer_name+" does not have attributes!")
  1248. def addExistingLayerToComposite(self, name):
  1249. x = self.dlg.listWidget.currentRow()
  1250. if self.checkLayerOnLayman(name):
  1251. inComposite = name in self.isRasterLayerInComposite(x, name)
  1252. if not(inComposite):
  1253. response = requests.get(self.URI+'/rest/'+self.laymanUsername+'/layers/'+str(name), verify=False)
  1254. res = self.fromByteToJson(response.content)
  1255. print(res)
  1256. wmsUrl = res['wms']['url']
  1257. self.compositeList[x]['layers'].append({"metadata":{},"visibility":True,"opacity":1,"title":str(name),"className":"HSLayers.Layer.WMS","singleTile":True,"wmsMaxScale":0,"legends":[""],"maxResolution":None,"minResolution":0,"url": wmsUrl ,"params":{"LAYERS": str(name),"INFO_FORMAT":"application/vnd.ogc.gml","FORMAT":"image/png","FROMCRS":"EPSG:3857","VERSION":"1.3.0"},"ratio":1.5,"dimensions":{}})
  1258. self.importMap(x, 'add', 1)
  1259. self.refreshLayerList()
  1260. else:
  1261. QMessageBox.information(None, "Message", "Composition already include layer "+name+"!")
  1262. else:
  1263. QMessageBox.information(None, "Layman", "Something went wrong with this layer: "+name)
  1264. def addExistingMapToMemory(self, name):
  1265. response = requests.get(self.URI+'/rest/'+self.laymanUsername+'/maps/'+str(name), verify=False)
  1266. res = self.fromByteToJson(response.content)
  1267. print(res)
  1268. wmsUrl = res['wms']['url']
  1269. x = self.dlg.listWidget.currentRow()
  1270. self.compositeList[x]['layers'].append({"metadata":{},"visibility":True,"opacity":1,"title":str(name),"className":"HSLayers.Layer.WMS","singleTile":True,"wmsMaxScale":0,"legends":[""],"maxResolution":None,"minResolution":0,"url": wmsUrl ,"params":{"LAYERS": str(name),"INFO_FORMAT":"application/vnd.ogc.gml","FORMAT":"image/png","FROMCRS":"EPSG:3857","VERSION":"1.3.0"},"ratio":1.5,"dimensions":{}})
  1271. self.importMap(x, 'add', 1)
  1272. self.refreshLayerList()
  1273. def layerInfoRedirect(self, name):
  1274. url = self.URI+'/rest/'+self.laymanUsername+"/layers/" + name
  1275. print(url)
  1276. response = requests.get(url)
  1277. r = self.fromByteToJson(response.content)
  1278. try:
  1279. url = r['metadata']['record_url']
  1280. webbrowser.open(url, new=2) ## redirect na micku pro více info
  1281. except:
  1282. QMessageBox.information(None, "Layman", "Link is unavailable.")
  1283. def checkLayerOnLayman(self, layer_name):
  1284. url = self.URI+'/rest/'+self.laymanUsername+"/layers/"+layer_name
  1285. r = requests.get(url = url, verify=False)
  1286. #print(r.content)
  1287. data = r.json()
  1288. try:
  1289. if data['wms']['status'] == 'NOT_AVAILABLE' or data['wms']['status'] == 'PENDING':
  1290. return False
  1291. else:
  1292. return True
  1293. except:
  1294. return True # validní vrstva nemá status
  1295. def addLayerToComposite(self,x):
  1296. self.compositeListOld = copy.deepcopy(self.compositeList) ## list pred upravou pro vraceni zmen
  1297. #layers = self.getSelectedLayers() #odkomentovat pro zapnuti nacitani vrstev z listwidgetu
  1298. layers = []
  1299. layers.append(self.dlg.mMapLayerComboBox.currentLayer())
  1300. successful = 0
  1301. print (len(layers))
  1302. try:
  1303. step = self.getProgressBarStep(len(layers))
  1304. except:
  1305. QMessageBox.information(None, "Layman", "No layer selected!")
  1306. for i in range (0, len(layers)):
  1307. print(type(layers[i]))
  1308. print("tst")
  1309. print (self.isLayerInComposite(x))
  1310. print (layers[i].name() in self.isLayerInComposite(x))
  1311. inComposite = layers[i].name() in self.isLayerInComposite(x)
  1312. if (self.checkExistingLayer(layers[i].name()) and inComposite):
  1313. QMessageBox.information(None, "Message", "Composition already include layer "+layers[i].name()+"!")
  1314. # self.postRequest(layers[i].name())
  1315. else:
  1316. self.postRequest(layers[i].name())
  1317. wmsStatus = 'PENDING'
  1318. j = 0
  1319. while ((wmsStatus == 'PENDING') and (j < 10)):
  1320. print (self.URI+'/rest/'+self.laymanUsername+'/layers/'+str(layers[i].name()))
  1321. response = requests.get(self.URI+'/rest/'+self.laymanUsername+'/layers/'+str(layers[i].name()), verify=False)
  1322. res = self.fromByteToJson(response.content)
  1323. try:
  1324. wmsStatus = res['wms']['status']
  1325. except:
  1326. wmsStatus = "done"
  1327. time.sleep(1)
  1328. j = j + 1
  1329. print(res)
  1330. try:
  1331. wmsUrl = res['wms']['url']
  1332. except:
  1333. wmsUrl = self.URI+'/geoserver/'+layers[i].name()+'/ows'
  1334. self.compositeList[x]['layers'].append({"metadata":{},"visibility":True,"opacity":1,"title":str(layers[i].name()),"className":"HSLayers.Layer.WMS","singleTile":True,"wmsMaxScale":0,"legends":[""],"maxResolution":None,"minResolution":0,"url": wmsUrl ,"params":{"LAYERS": str(layers[i].name()),"INFO_FORMAT":"application/vnd.ogc.gml","FORMAT":"image/png","FROMCRS":"EPSG:3857","VERSION":"1.3.0"},"ratio":1.5,"dimensions":{}})
  1335. successful = successful + 1
  1336. self.dlg.progressBar.setValue(self.dlg.progressBar.value()+step)
  1337. self.importMap(x, "add", successful)
  1338. def getProgressBarStep(self, count):
  1339. return (100/count)
  1340. def deteteLayerFromComposite(self, x, position, name): ## pro verzi s komboboxem
  1341. previousLayers = self.compositeList[x]['layers']
  1342. self.compositeList[x]['layers'] = []
  1343. for i in range (0,len(previousLayers)):
  1344. if (i != position):
  1345. self.compositeList[x]['layers'].append(previousLayers[i])
  1346. self.importMap(x, "del")
  1347. done = True
  1348. if (len(self.compositeList[x]['layers']) != 0 and len(self.compositeList[x]['layers']) != None):
  1349. for i in range (0, len(self.compositeList[x]['layers'])):
  1350. if (self.compositeList[x]['layers'][i]['params']['LAYERS'] == name):
  1351. done = False
  1352. if (done and QgsProject.instance().mapLayersByName(name)): ## kompozice může mít načtené vrstvy, které nejsou v canvasu
  1353. self.deleteLayerFromCanvas(name)
  1354. def importCleanComposite(self,x):
  1355. tempFile = tempfile.gettempdir() + os.sep + "atlas" + os.sep + "compsite.json"
  1356. with open(tempFile, 'w') as outfile:
  1357. json.dump(self.compositeList[x], outfile)
  1358. with open(tempFile, 'rb') as f:
  1359. d = json.load(f)
  1360. files = {'file': (tempFile, open(tempFile, 'rb')),}
  1361. data = { 'name' : self.compositeList[x]['name'], 'title' : self.compositeList[x]['title'], 'description' : self.compositeList[x]['abstract']}
  1362. print(self.URI+'/rest/'+self.laymanUsername+'/maps/')
  1363. #response = requests.post(self.URI+'/rest/'+self.laymanUsername+'/maps/', data = data, files=files, headers = self.authHeader, verify=False)
  1364. print(self.authHeader)
  1365. response = requests.post(self.URI+'/rest/'+self.laymanUsername+'/maps', files=files, data = data, headers = self.authHeader, verify=False)
  1366. print(response.content)
  1367. def showProgressBar(self, bar):
  1368. bar = QProgressBar()
  1369. bar.setRange(0,0) ## range 0,0 je nekonečný
  1370. self.showProgressBar(bar)
  1371. bar.show()
  1372. iface.mainWindow().statusBar().addWidget(bar)
  1373. def fromByteToJson(self, res):
  1374. pom = res
  1375. pom = pom.decode('utf_8')
  1376. pom = json.loads(pom)
  1377. return pom
  1378. def importMap(self, x, operation, s = 0): ##s je počet vrstev úspěšně nahraných na server
  1379. if (s != 0):
  1380. self.dlg.progressBar.show()
  1381. self.dlg.label_import.show()
  1382. tempFile = tempfile.gettempdir() + os.sep + "atlas" + os.sep + "compsite.json"
  1383. try:
  1384. existingLayers = self.isLayerInComposite(x)
  1385. except:
  1386. existingLayers = []
  1387. successful = s
  1388. with open(tempFile, 'w') as outfile:
  1389. json.dump(self.compositeList[x], outfile)
  1390. jsonPath = tempfile.gettempdir() + os.sep + "atlas" + os.sep + "compsite.json"
  1391. with open(jsonPath, 'rb') as f:
  1392. d = json.load(f)
  1393. files = {'file': (jsonPath, open(jsonPath, 'rb')),}
  1394. data = { 'name' : self.compositeList[x]['name'], 'title' : self.compositeList[x]['title'], 'description' : self.compositeList[x]['abstract']}
  1395. req = requests.get(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'])
  1396. mapCode = req.status_code ## test jestli vrstva na serveru existuje. Pokud ne = error 404
  1397. if (mapCode == 404):
  1398. if (operation == "add"):
  1399. msgbox = QMessageBox(QMessageBox.Question, "Import Map", "Chcete přidat do kompozice "+str(successful)+" vrstev?")
  1400. if (operation == "mod"):
  1401. msgbox = QMessageBox(QMessageBox.Question, "Import Map", "Chcete opravdu provést změny v této kompozici?")
  1402. self.dlg.close()
  1403. self.afterCloseEditMapDialog()
  1404. print(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'])
  1405. response = requests.delete(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'],headers = self.authHeader, verify=False)
  1406. print(response.content)
  1407. self.dlg.show()
  1408. if (operation == "del"):
  1409. msgbox = QMessageBox(QMessageBox.Question, "Import Map", "Chcete smazat vybranou vrstvu?")
  1410. msgbox.addButton(QMessageBox.Yes)
  1411. msgbox.addButton(QMessageBox.No)
  1412. msgbox.setDefaultButton(QMessageBox.No)
  1413. reply = msgbox.exec()
  1414. if (reply == QMessageBox.Yes):
  1415. response = requests.post(self.URI+'/rest/'+self.laymanUsername+'/maps', files=files, data = data, headers = self.authHeader, verify=False)
  1416. print(response.content)
  1417. data = response.content
  1418. self.dlg.progressBar.hide()
  1419. # self.refreshItems()## refresh formuláře
  1420. if (operation != "mod"):
  1421. self.refreshLayerList()
  1422. else:
  1423. if (operation == "add"):
  1424. self.dlg.progressBar.setValue(0)
  1425. else:
  1426. print(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'])
  1427. files = {'file': (jsonPath, open(jsonPath, 'rb')),}
  1428. response = requests.patch(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'], data = data, files=files, headers = self.authHeader, verify=False)
  1429. try:
  1430. self.refreshLayerList()
  1431. except:
  1432. ## nacházíme se v jiném formuláři, není co refreshovat
  1433. pass
  1434. #if (operation == "add"):
  1435. # print(len(existingLayers))
  1436. # msgbox = QMessageBox(QMessageBox.Question, "Layman", "Vybraná vrstva bude přidána do kompozice.")
  1437. # #if(len(existingLayers) == 0):
  1438. # # #if (successful != 0):
  1439. # # msgbox = QMessageBox(QMessageBox.Question, "Layman", "Vybraná vrstva bude přidána do kompozice.")
  1440. # #else:
  1441. # # # msgbox = QMessageBox(QMessageBox.Question, "Import Map", "Kompozice již obsauje vrstvy: "+(', '.join(existingLayers))+". Chcete vybrané vrstvy opravdu znovu přidat?.")
  1442. # # msgbox = QMessageBox(QMessageBox.Question, "Layman", "Vybraná vrstva bude přidána do kompozice.")
  1443. #if (operation == "del"):
  1444. # msgbox = QMessageBox(QMessageBox.Question, "Import Map", "Chcete smazat vybranou vrstvu?")
  1445. #if (operation == "mod"):
  1446. # msgbox = QMessageBox(QMessageBox.Question, "Import Map", "Chcete uložit upravenou mapu?")
  1447. #msgbox.addButton(QMessageBox.Yes)
  1448. #msgbox.addButton(QMessageBox.No)
  1449. #msgbox.setDefaultButton(QMessageBox.No)
  1450. #reply = msgbox.exec()
  1451. #if (reply == QMessageBox.Yes):
  1452. # print(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'])
  1453. # files = {'file': (jsonPath, open(jsonPath, 'rb')),}
  1454. # response = requests.patch(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'], data = data, files=files, headers = self.authHeader, verify=False)
  1455. # try:
  1456. # self.refreshLayerList()
  1457. # except:
  1458. # ## nacházíme se v jiném formuláři, není co refreshovat
  1459. # pass
  1460. #if (reply == QMessageBox.No):
  1461. # self.compositeList = copy.deepcopy(self.compositeListOld)
  1462. # #self.refreshItems()##refresh formulare
  1463. # self.refreshLayerList()
  1464. try: ## některé formuláře nemají progress bar
  1465. self.dlg.progressBar.hide()
  1466. self.dlg.label_import.hide()
  1467. except:
  1468. pass
  1469. def deleteLayer(self, layerName):
  1470. url = self.URI+'/rest/'+self.laymanUsername+"/layers/" + layerName
  1471. r = requests.delete(url)
  1472. print (r.content)
  1473. def getActiveLayer(self):
  1474. layer = self.iface.activeLayer()
  1475. return layer
  1476. def createComposite(self, name, title):
  1477. # if (name == "" or title == ""):
  1478. if (title == ""):
  1479. QMessageBox.information(None, "Message", "Není vyplněn titulek!")
  1480. else:
  1481. name = self.removeUnacceptableChars(title)
  1482. data = self.getEmptyComposite(name,title)
  1483. print (data)
  1484. self.compositeList.append(data)
  1485. x = len(self.compositeList) - 1
  1486. self.importCleanComposite(x)
  1487. try:
  1488. self.refreshCompositeList()
  1489. except:
  1490. pass
  1491. def patchLayer(self, layer_name, data):
  1492. self.layerName = layer_name
  1493. self.json_export(layer_name)
  1494. sldPath = self.getTempPath(self.layerName).replace("geojson", "sld")
  1495. geoPath = self.getTempPath(self.layerName)
  1496. print(sldPath)
  1497. if(os.path.isfile(sldPath)): ## existuje sld?
  1498. files = [('file', open(geoPath, 'rb')), ('sld', open(sldPath, 'rb'))]
  1499. print("sld")
  1500. else:
  1501. files = {'file': (geoPath, open(geoPath, 'rb')),}
  1502. print("ne sld")
  1503. url = self.URI+'/rest/'+self.laymanUsername+"/layers/" + layer_name
  1504. print(url)
  1505. r = requests.patch(url, files=files, data = data, headers = self.authHeader, verify=False)
  1506. print(r.content)
  1507. def getTempPath(self, name):
  1508. print (name)
  1509. if type(name) is tuple:
  1510. return (name[0] + name[1])
  1511. else:
  1512. tempFile = tempfile.gettempdir() + os.sep + name +'.geojson'
  1513. return tempFile
  1514. def sendLayer(self):
  1515. layer = self.getActiveLayer()
  1516. if (layer == None):
  1517. QMessageBox.information(None, "Message", "Neexistuje žádná vrstva k uložení")
  1518. else:
  1519. self.json_export()
  1520. self.postRequest()
  1521. def getExistingLayers(self):
  1522. url = self.URI+'/rest/'+self.laymanUsername+"/layers"
  1523. r = requests.get(url = url)
  1524. data = r.json()
  1525. for x in range(len(data)):
  1526. print(data[x]['name'])
  1527. return data
  1528. def checkExistingLayer(self, layerName):
  1529. layerName = self.removeUnacceptableChars(layerName)
  1530. url = self.URI+'/rest/'+self.laymanUsername+"/layers"
  1531. r = requests.get(url = url)
  1532. data = r.json()
  1533. pom = set()
  1534. for x in range(len(data)):
  1535. pom.add((data[x]['name']))
  1536. layerName = layerName.replace(" ", "_").lower()
  1537. print(layerName)
  1538. print(pom)
  1539. if (layerName in pom):
  1540. return True
  1541. else:
  1542. return False
  1543. def removeUnacceptableChars(self, input):
  1544. input = input.replace("ř","r")
  1545. input = input.replace("š","s")
  1546. input = input.replace("ž","z")
  1547. input = input.replace("ů","u")
  1548. input = input.replace("ú","u")
  1549. input = input.replace(" ","_")
  1550. input = input.replace("é","e")
  1551. input = input.replace("í","i")
  1552. input = input.replace("á","a")
  1553. input = input.replace("ó","o")
  1554. input = input.replace("č","c")
  1555. input = input.replace("ď","d")
  1556. input = input.replace("ě","e")
  1557. input = input.replace("ť","t")
  1558. input = re.sub(r'[?|$|.|!]',r'',input)
  1559. input = input.lower()
  1560. # iface.messageBar().pushWidget(iface.messageBar().createMessage("Layman:", "Diacritics in name of layer was replaced."), Qgis.Success, duration=3)
  1561. return input
  1562. def test (self):
  1563. item = self.dlgGetLayers.items.currentItem().text()
  1564. print( item)
  1565. def isLayerInComposite(self, x):
  1566. layers =[]
  1567. layers.append(self.dlg.mMapLayerComboBox.currentLayer())
  1568. print (self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name']+'/file')
  1569. req = requests.get (self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name']+'/file')
  1570. data = req.json()
  1571. print (data['abstract'])
  1572. print (len(self.compositeList[x]['layers']))
  1573. existingLayers = []
  1574. for i in range (0, len(data['layers'])):
  1575. for j in range (0, len(layers)):
  1576. if (data['layers'][i]['params']['LAYERS'] in layers[j].name() ):
  1577. existingLayers.append(data['layers'][i]['params']['LAYERS'])
  1578. return existingLayers
  1579. def isRasterLayerInComposite(self, x, name):
  1580. req = requests.get (self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name']+'/file')
  1581. data = req.json()
  1582. existingLayers = []
  1583. for i in range (0, len(data['layers'])):
  1584. if (data['layers'][i]['params']['LAYERS'] == name ):
  1585. existingLayers.append(data['layers'][i]['params']['LAYERS'])
  1586. return existingLayers
  1587. def addComposite(self, data, service, groupName = ''):
  1588. #self.compositeList.append (data)
  1589. print(len(data['layers']))
  1590. for x in range(0, len(data['layers'])):
  1591. # repairUrl = self.convertUrlFromHex(data['layers'][x]['url'])
  1592. repairUrl = self.URI+"/geoserver/"+self.laymanUsername+"/ows"
  1593. layerName = data['layers'][x]['params']['LAYERS']
  1594. format = data['layers'][x]['params']['FORMAT']
  1595. # epsg = str(data['groups']['projection']).upper()
  1596. epsg = 'EPSG:4326'
  1597. #abstract = data['data']['layers'][x]['params']['ABSTRACT']
  1598. wmsName = data['layers'][x]['params']['LAYERS']
  1599. if service == 'WMS':
  1600. self.loadWms(repairUrl, layerName, format,epsg, groupName)
  1601. if service == 'WFS':
  1602. self.loadWfs(repairUrl, layerName, groupName)
  1603. def loadService(self, data, service, groupName = ''):
  1604. print(len(data['layers']))
  1605. for x in range(0, len(data['layers'])):
  1606. #repairUrl = self.convertUrlFromHex(data['layers'][x]['url'])
  1607. repairUrl = "http://layman.lesprojekt.cz/geoserver/"+self.laymanUsername+"/ows"
  1608. layerName = data['layers'][x]['params']['LAYERS']
  1609. format = data['layers'][x]['params']['FORMAT']
  1610. # epsg = str(data['groups']['projection']).upper()
  1611. epsg = 'EPSG:4326'
  1612. #abstract = data['data']['layers'][x]['params']['ABSTRACT']
  1613. wmsName = data['layers'][x]['params']['LAYERS']
  1614. if self.checkLayerOnLayman(layerName):
  1615. if service == 'WMS':
  1616. self.loadWms(repairUrl, layerName, format,epsg, groupName)
  1617. if service == 'WFS':
  1618. self.loadWfs(repairUrl, layerName, groupName)
  1619. else:
  1620. QMessageBox.information(None, "Layman", "Layer: "+layerName + " is corrupted and will not be loaded.")
  1621. def loadWms(self, url, layerName, format, epsg, groupName = ''):
  1622. print(url)
  1623. urlWithParams = 'contextualWMSLegend=0&crs='+epsg+'&dpiMode=7&featureCount=10&format=image/png&layers='+layerName+'&styles=&url=' + url
  1624. print(urlWithParams)
  1625. rlayer = QgsRasterLayer(urlWithParams, layerName, 'wms')
  1626. if (groupName != ''):
  1627. self.addWmsToGroup(groupName,rlayer)
  1628. else:
  1629. QgsProject.instance().addMapLayer(rlayer)
  1630. def loadWfs(self, url, layerName, groupName = ''):
  1631. epsg = 'EPSG:4326'
  1632. uri = "http://layman.lesprojekt.cz/geoserver/"+self.laymanUsername+"/ows?srsname="+epsg+"&typename="+self.laymanUsername+":"+layerName+"&restrictToRequestBBOX=1&pagingEnabled=True &version=auto&request=GetFeature&service=WFS"
  1633. print(uri)
  1634. vlayer = QgsVectorLayer(uri, layerName, "WFS")
  1635. print(vlayer.isValid())
  1636. if (groupName != ''):
  1637. self.addWmsToGroup(groupName,vlayer)
  1638. else:
  1639. QgsProject.instance().addMapLayer(vlayer)
  1640. def addWmsToGroup(self, groupName, layer):
  1641. root = QgsProject.instance().layerTreeRoot()
  1642. group = root.findGroup(groupName)
  1643. if not(group):
  1644. group = root.addGroup(groupName)
  1645. QgsProject.instance().addMapLayer(layer,False)
  1646. group.insertChildNode(1,QgsLayerTreeLayer(layer))
  1647. def convertUrlFromHex(self, url):
  1648. url = url.replace('%3A',':')
  1649. url = url.replace('%2F','/')
  1650. url = url.replace('%3F','?')
  1651. url = url.replace('%3D','=')
  1652. url = url.replace('%26','&')
  1653. return url
  1654. def checkEpsg(self, name):
  1655. ret = False
  1656. layer = QgsProject.instance().mapLayersByName(name)
  1657. layerCRS = layer[0].crs().authid()
  1658. if ((layerCRS == 'EPSG:4326') or (layerCRS == 'EPSG:3857')):
  1659. ret = True
  1660. return ret
  1661. def registerLayer(self, name):
  1662. sldPath = self.getTempPath(name).replace("geojson", "sld")
  1663. geoPath = self.getTempPath(name)
  1664. url = "http://layman.lesprojekt.cz/rest/"+self.laymanUsername+"/layers"
  1665. files = {'sld': (sldPath, open(sldPath, 'rb')),} # nahrávám sld
  1666. payload = {
  1667. 'file': name+".geojson"
  1668. }
  1669. response = requests.request("POST", url, files = files, data=payload, headers = self.authHeader)
  1670. # print(response.text)
  1671. def read_in_chunks(self, file_object): ## cca 1MB chunk převzato z laymana test klienta
  1672. chunk_size=self.CHUNK_SIZE
  1673. while True:
  1674. data = file_object.read(chunk_size)
  1675. if not data:
  1676. break
  1677. yield data
  1678. def postInChunks(self, layer_name, reqType):
  1679. print(reqType)
  1680. if (reqType == "patch"):
  1681. url = self.URI+'/rest/'+self.laymanUsername+'/layers/'+layer_name
  1682. r = requests.delete(url,headers = self.authHeader)
  1683. # print(r.content)
  1684. self.registerLayer(layer_name)
  1685. filePath = os.path.join(tempfile.gettempdir(), "atlas_chunks" ) ## chunky se ukládají do adresáře v tempu
  1686. if not (os.path.exists(filePath)):
  1687. os.mkdir(filePath)
  1688. file = self.getTempPath(layer_name)
  1689. f = open(file, 'rb')
  1690. arr = []
  1691. for piece in self.read_in_chunks(f):
  1692. arr.append(piece)
  1693. url = self.URI+'/rest/'+self.laymanUsername+'/layers/'+layer_name+'/chunk'
  1694. resumableFilename = layer_name+'.geojson'
  1695. layman_original_parameter = "file"
  1696. resumableTotalChunks = len(arr)
  1697. for i in range (1, len(arr)+1): ##chunky jsou počítané od 1 proto +1
  1698. print("chunk" + str(i))
  1699. file = arr[i-1] # rozsekaná část souboru
  1700. resumableChunkNumber = i # cislo casti
  1701. payload = {
  1702. 'file' : "chunk"+str(i)+".geojson",
  1703. 'resumableFilename': resumableFilename,
  1704. 'layman_original_parameter': layman_original_parameter,
  1705. 'resumableChunkNumber': i,
  1706. 'resumableTotalChunks': resumableTotalChunks
  1707. }
  1708. f = open(filePath + os.sep+"chunk"+str(i)+".geojson", "wb")
  1709. f.write(bytearray(arr[i-1]))
  1710. f.close()
  1711. files = {'file': (layer_name+".geojson", open(filePath +os.sep+ "chunk"+str(i)+".geojson", 'rb')),}
  1712. response = requests.post(url, files = files, data=payload, headers = self.authHeader)
  1713. print(response.content)
  1714. ############################################# auth part############################
  1715. def startThread(self):
  1716. self.thread1 = threading.Thread(target=self.refreshToken)
  1717. self.thread1.start()
  1718. def refreshToken(self):
  1719. #self.expires_in = 60
  1720. path = tempfile.gettempdir() + os.sep + "atlas" + os.sep + "tsts.txt"
  1721. i = 0
  1722. tokenEndpoint = "https://www.agrihub.cz/o/oauth2/token"
  1723. while (i < self.expires_in):
  1724. f = open(path, "a")
  1725. f.write(str(i))
  1726. f.close()
  1727. time.sleep(1)
  1728. if i == self.expires_in - 4:
  1729. data = {'grant_type':'refresh_token',
  1730. 'refresh_token': self.refresh_token,
  1731. 'client_id':'id-3462f94b-875c-9185-4ced-b69841f24b3',
  1732. 'redirect_uri':'http://localhost:3000/client/authn/oauth2-liferay/callback',
  1733. 'code_verifier':'test'
  1734. }
  1735. r = requests.post(url = tokenEndpoint, data = data, headers = self.authHeader)
  1736. res = self.fromByteToJson(r.content)
  1737. self.access_token = res['access_token']
  1738. self.refresh_token = res['refresh_token']
  1739. self.expires_in = res['expires_in']
  1740. print(self.access_token)
  1741. self.setAuthHeader()
  1742. i = 0
  1743. else:
  1744. i += 1
  1745. def authOptained (self):
  1746. self.dlg.pushButton_Continue.setEnabled(True)
  1747. self.menu_Connection.setEnabled(True)
  1748. self.menu_saveLocalFile.setEnabled(True)
  1749. self.menu_loadJson.setEnabled(True)
  1750. self.menu_ImportLayerDialog.setEnabled(True)
  1751. self.menu_AddLayerDialog.setEnabled(True)
  1752. self.menu_AddMapDialog.setEnabled(True)
  1753. self.menu_ImportLayerDialog.setEnabled(True)
  1754. self.menu_ImportMapDialog.setEnabled(True)
  1755. # self.menu_DeleteMapDialog.setEnabled(True)
  1756. # self.menu_CreateCompositeDialog.setEnabled(True)
  1757. self.menu_UserInfoDialog.setEnabled(True)
  1758. self.textbox.setText("Layman: Logged user")
  1759. def getCodeVerifier(self):
  1760. code_verifier = base64.urlsafe_b64encode(os.urandom(40)).decode('utf-8')
  1761. code_verifier = re.sub('[^a-zA-Z0-9]+', '', code_verifier)
  1762. self.code_verifier = code_verifier
  1763. return (code_verifier)
  1764. def getCodeChallenge(self, code_verifier):
  1765. code_challenge = hashlib.sha256(code_verifier.encode('utf-8')).digest()
  1766. code_challenge = base64.urlsafe_b64encode(code_challenge).decode('utf-8')
  1767. code_challenge = code_challenge.replace('=', '')
  1768. self.code_challenge = code_challenge
  1769. return (code_challenge)
  1770. def printVariables(self):
  1771. print (self.access_token)
  1772. print(self.refresh_token)
  1773. print(self.authHeader)
  1774. print(self.laymanUsername)
  1775. def getAuthCode(self):
  1776. path = tempfile.gettempdir() + os.sep + "atlas" + os.sep + "auth.txt"
  1777. f = open(path, "r")
  1778. ret = f.read()
  1779. f.close()
  1780. return ret
  1781. def getToken(self):
  1782. self.saveIni()
  1783. tokenEndpoint = "https://www.agrihub.cz/o/oauth2/token"
  1784. # data to be sent to api
  1785. data = {'grant_type':'authorization_code',
  1786. 'client_id': self.client_id,
  1787. 'redirect_uri':'http://localhost:3000/client/authn/oauth2-liferay/callback',
  1788. 'code_verifier': self.code_verifier,
  1789. 'code': self.getAuthCode()}
  1790. # sending post request and saving response as response object
  1791. r = requests.post(url = tokenEndpoint, data = data, headers = self.authHeader)
  1792. res = self.fromByteToJson(r.content)
  1793. print (res)
  1794. try:
  1795. print(res['access_token'])
  1796. print(res['expires_in'])
  1797. print(res['refresh_token'])
  1798. except:
  1799. QMessageBox.information(None, "Message", "Autorization was not sucessfull! Please try it again.")
  1800. return
  1801. self.access_token = res['access_token']
  1802. self.refresh_token = res['refresh_token']
  1803. self.expires_in = res['expires_in']
  1804. self.setAuthHeader()
  1805. data ={
  1806. 'access_token': res['access_token'],
  1807. 'refresh_token': res['refresh_token'],
  1808. 'expires_in':res['expires_in']
  1809. }
  1810. path = tempfile.gettempdir() + os.sep + "atlas" + os.sep + "tokens.json"
  1811. with open(path, 'w') as outfile:
  1812. json.dump(data, outfile)
  1813. self.startThread()
  1814. self.registerUserIfNotExists()
  1815. self.dlg.close()
  1816. def setAuthHeader(self):
  1817. self.authHeader ={
  1818. "Authorization": "Bearer " + self.access_token,
  1819. "AuthorizationIssUrl" : 'https://www.agrihub.cz/o/oauth2/authorize'
  1820. }
  1821. def registerUserIfNotExists(self):
  1822. userEndpoint = "http://layman.lesprojekt.cz/rest/current-user"
  1823. #self.laymanUsername = self.dlg.lineEdit_userName.text()
  1824. id = self.client_id.replace('-', '')
  1825. user = {'username':self.Agrimail}
  1826. #user = self.Agrimail
  1827. print("authheader: "+ str(self.authHeader))
  1828. r = requests.patch(url = userEndpoint, data = user, headers = self.authHeader)
  1829. res = r.text
  1830. res = self.fromByteToJson(r.content)
  1831. print (res)
  1832. print(user)
  1833. print(res)
  1834. try:
  1835. # if res['message'] == 'User already reserved username.': # res['code'] == 34, code 35 je pokud již jiný uživatel má účet, který chceme registrovat
  1836. if res['code'] == 34: # res['code'] == 34, code 35 je pokud již jiný uživatel má účet, který chceme registrovat (code 35 pravděpodobně nemůže nastat)
  1837. print("user exists")
  1838. self.laymanUsername = res['detail']['username']
  1839. #self.laymanUsername = res['username']
  1840. #else:
  1841. # self.laymanUsername = user['username']
  1842. except:
  1843. print("creating new user: " + res['username'])
  1844. self.laymanUsername = res['username']
  1845. #API_ENDPOINT = "http://layman.lesprojekt.cz/rest/current-user"
  1846. #headers ={
  1847. # "Authorization": "Bearer "+ self.access_token,
  1848. # "AuthorizationIssUrl" : 'https://www.agrihub.cz/o/oauth2/authorize'
  1849. #}
  1850. def openAuthLiferayUrl(self):
  1851. self.getCodeChallenge(self.getCodeVerifier()) ##generování kódů
  1852. self.client_id = self.dlg.lineEdit_AgriID.text()
  1853. path = tempfile.gettempdir() + os.sep + "atlas" + os.sep + "auth.txt"
  1854. self.watcher = QFileSystemWatcher()
  1855. self.watcher.addPath(path)
  1856. self.watcher.fileChanged.connect(self.authOptained)
  1857. #################
  1858. print ("##### flask server is starting #####")
  1859. thread = StartFlaskDaemon()
  1860. thread.daemon = True
  1861. thread.start()
  1862. url = 'https://www.agrihub.cz/o/oauth2/authorize?response_type=code&client_id='+self.client_id+'&redirect_uri=http%3A%2F%2Flocalhost:3000%2Fclient%2Fauthn%2Foauth2-liferay%2Fcallback&code_challenge='+self.code_challenge ##n4bQgYhMfWWaL-qgxVrQFaO_TxsrC4Is0V1sFbDwCgg'
  1863. webbrowser.open(url, new=2)
  1864. def saveIni(self):
  1865. file = os.getenv("HOME") + os.sep + ".layman" + os.sep + 'layman_user.INI'
  1866. dir = os.getenv("HOME") + os.sep + ".layman"
  1867. if not (os.path.isdir(dir)):
  1868. try:
  1869. os.mkdir(dir)
  1870. except OSError:
  1871. print ("vytváření adresáře selhalo")
  1872. config = configparser.ConfigParser()
  1873. config['DEFAULT']['login'] = self.Agrimail
  1874. config['DEFAULT']['id'] = self.client_id
  1875. with open(file, 'w') as configfile:
  1876. config.write(configfile)
  1877. def loadIni(self):
  1878. file = os.getenv("HOME") + os.sep + ".layman" + os.sep +'layman_user.INI'
  1879. config = configparser.ConfigParser()
  1880. config.read(file)
  1881. return config
  1882. def run(self):
  1883. """Run method that loads and starts the plugin"""
  1884. if not self.pluginIsActive:
  1885. self.pluginIsActive = True
  1886. #print "** STARTING Atlas"
  1887. # dockwidget may not exist if:
  1888. # first run of plugin
  1889. # removed on close (see self.onClosePlugin method)
  1890. if self.dockwidget == None:
  1891. # Create the dockwidget (after translation) and keep reference
  1892. self.dockwidget = AtlasDockWidget()
  1893. # connect to provide cleanup on closing of dockwidget
  1894. self.dockwidget.closingPlugin.connect(self.onClosePlugin)
  1895. self.dockwidget.pushButton.clicked.connect(self.sendLayer)
  1896. self.dockwidget.pushButton_2.clicked.connect(self.loadLocalFile)
  1897. self.dockwidget.pushButton_3.clicked.connect(self.saveLocalFile)
  1898. self.dockwidget.pushButton_getLayers.clicked.connect(self.run_getLayer)
  1899. self.dlgGetLayers.pushButtonxx.clicked.connect(lambda: print(self.dlgGetLayers.items.currentItem().text()))
  1900. # show the dockwidget
  1901. # TODO: fix to allow choice of dock location
  1902. self.iface.addDockWidget(Qt.TopDockWidgetArea, self.dockwidget)
  1903. self.dockwidget.show()
  1904. class StartFlaskDaemon(threading.Thread):
  1905. def run(self):
  1906. from subprocess import Popen, PIPE
  1907. import platform
  1908. import sys
  1909. from pathlib import Path
  1910. d= os.path.dirname(Path(__file__).absolute()) + os.sep+"flask_listener" + os.sep + "flask_listener.py"
  1911. if platform.system() == 'Windows':
  1912. process = subprocess.Popen(["python", d],stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
  1913. else:
  1914. process = subprocess.Popen(["python " + d],stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
  1915. stdout, stderr = process.communicate()
  1916. print (stdout, stderr)
  1917. '''Start your thread here'''