Atlas.py 117 KB

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