Atlas.py 117 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441
  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)+'/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)+'/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)+'/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. print (sldPath)
  1011. exists = os.path.isfile(sldPath)
  1012. if (exists):
  1013. vlayer.loadSldStyle(os.path.splitext(fileName)[0]+ ".sld")
  1014. print("sld loaded")
  1015. QgsProject.instance().addMapLayer(vlayer)
  1016. vlayer.triggerRepaint()
  1017. def checkLoadedLayer(self):
  1018. layers = iface.layerTreeView().selectedLayers()
  1019. if (len(layers) > 0):
  1020. self.json_export()
  1021. else:
  1022. QMessageBox.information(None, "Message", "You must load layer first!")
  1023. def getEmptyComposite(self, compositeName, compositeTitle):
  1024. compositeEPSG = "epsg:4326"
  1025. if (self.dlg.lineEdit_3.text() == "" or self.dlg.lineEdit_4.text() == "" or self.dlg.lineEdit_5.text() == "" or self.dlg.lineEdit_6.text() == ""):
  1026. ext = iface.mapCanvas().extent()
  1027. xmin = ext.xMinimum()
  1028. xmax = ext.xMaximum()
  1029. ymin = ext.yMinimum()
  1030. ymax = ext.yMaximum()
  1031. else:
  1032. xmin = self.dlg.lineEdit_3.text()
  1033. xmax = self.dlg.lineEdit_4.text()
  1034. ymin = self.dlg.lineEdit_5.text()
  1035. ymax = self.dlg.lineEdit_6.text()
  1036. abstract = self.dlg.lineEdit_7.text()
  1037. 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}}
  1038. self.dlg.close()
  1039. iface.messageBar().pushWidget(iface.messageBar().createMessage("Atlas:", " Kompozice " + compositeName + " byla úspešně vytvořena."), Qgis.Success, duration=3)
  1040. return comp
  1041. def loadLocalFile(self):
  1042. options = QFileDialog.Options()
  1043. dialog = QFileDialog()
  1044. 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 ;}");
  1045. fileName = dialog.getOpenFileName(None,"Load file", "","GeoJson Files (*.geojson);;Json Files (*.json)", options=options)
  1046. print ("načítám json ze souboru:" + fileName[0])
  1047. self.loadJsonLayer(fileName[0])
  1048. def modifyMap(self, x):
  1049. name = self.removeUnacceptableChars(self.dlg.lineEdit_title.text())
  1050. #self.compositeList[x]['name'] = self.dlg.lineEdit_name.text()
  1051. self.compositeList[x]['name'] = name
  1052. self.compositeList[x]['abstract'] = self.dlg.lineEdit_abstract.text()
  1053. self.compositeList[x]['title'] = self.dlg.lineEdit_title.text()
  1054. self.compositeList[x]['extent'][0] = self.dlg.lineEdit_xmin.text()
  1055. self.compositeList[x]['extent'][2] = self.dlg.lineEdit_xmax.text()
  1056. self.compositeList[x]['extent'][1] = self.dlg.lineEdit_ymin.text()
  1057. self.compositeList[x]['extent'][3] = self.dlg.lineEdit_ymax.text()
  1058. print(self.compositeList[x]['extent'])
  1059. print(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'])
  1060. oldName = self.dlg.lineEdit_name.text()
  1061. response = requests.delete(self.URI+'/rest/'+self.laymanUsername+'/maps/'+oldName,headers = self.authHeader)
  1062. print(response.content)
  1063. self.dlg.close()
  1064. self.afterCloseEditMapDialog()
  1065. self.importMap(x, 'mod')
  1066. try:
  1067. self.refreshCompositeList()
  1068. iface.messageBar().pushWidget(iface.messageBar().createMessage("Import:", " Map metadata was saved successfully."), Qgis.Success, duration=3)
  1069. except:
  1070. pass
  1071. def saveLocalFile(self):
  1072. layer = self.iface.activeLayer()
  1073. path = iface.activeLayer().dataProvider().dataSourceUri()
  1074. path = path.split("|")[0].replace("'","")
  1075. if (layer == None):
  1076. QMessageBox.information(None, "Message", "You must load layer first!")
  1077. else:
  1078. # defaultDir = QDir().home()
  1079. defaultDir = os.path.dirname(path)
  1080. print(defaultDir)
  1081. print(defaultDir +os.sep+ str(layer.name()) + ".geojson")
  1082. print(QDir().homePath() +os.sep+ str(layer.name()) + ".geojson")
  1083. dialog = QFileDialog()
  1084. dialog.setFileMode(1)
  1085. dialog.setDirectory(defaultDir)
  1086. print(dialog.directory().path())
  1087. # layer_name = dialog.getSaveFileName(None, "Save file", QDir().homePath() +os.sep+ str(layer.name()) + ".geojson", "*.geojson")
  1088. layer_name = dialog.getSaveFileName(None, "Save file", defaultDir +os.sep+ str(layer.name()) + ".geojson", "*.geojson")
  1089. self.json_export_local(layer_name[0], layer)
  1090. def json_export_local(self, layer_name, lay):
  1091. import shutil
  1092. filePath = self.getTempPath(layer_name)
  1093. ogr_driver_name = "GeoJSON"
  1094. project = QgsProject.instance()
  1095. fileNames = []
  1096. ## zde musí být zajištěna vektorová vrstva
  1097. layer = lay
  1098. layerType = layer.type()
  1099. if layerType == QgsMapLayer.VectorLayer:
  1100. renderer = layer.renderer()
  1101. hasIcon = False
  1102. if isinstance(renderer, QgsSingleSymbolRenderer):
  1103. self.copySymbols(renderer.symbol(), tempfile.gettempdir(), fileNames)
  1104. hasIcon = True
  1105. layerCrs = qgis.utils.iface.activeLayer().crs().authid()
  1106. crs = QgsCoordinateReferenceSystem(layerCrs)
  1107. result2 = qgis.core.QgsVectorFileWriter.writeAsVectorFormat(layer, layer_name, "utf-8", crs, ogr_driver_name) # export jsonu do souboru
  1108. print(result2)
  1109. if(result2[0] == 2): ## testujeme zda není soubor otevřený v jiném procesu / návratový kod 2
  1110. tempFile = self.getTempPath(os.path.basename(layer_name.replace(".geojson", "") ))
  1111. print(tempFile)
  1112. #QMessageBox.information(None, "Layman", "It is not possible overwrite this file. File is already open in other process.")
  1113. sld_temp_filename = tempFile.replace("geojson", "sld")
  1114. layer.saveSldStyle(sld_temp_filename)
  1115. result2 = qgis.core.QgsVectorFileWriter.writeAsVectorFormat(layer, tempFile, "utf-8", crs, ogr_driver_name)
  1116. print(result2)
  1117. if(result2[0] == 2):
  1118. QMessageBox.information(None, "Layman", "It is not possible overwrite this file. File is already open in other process.")
  1119. return
  1120. if os.path.basename(layer_name.replace(".geojson", "")) != '':
  1121. QgsProject.instance().removeMapLayer(layer.id())
  1122. print(tempFile, layer_name)
  1123. shutil.copy(tempFile, layer_name)
  1124. shutil.copy(sld_temp_filename, layer_name.replace(".geojson", ".sld"))
  1125. self.loadJsonLayer(layer_name)
  1126. else:
  1127. sld_filename = layer_name.replace("geojson", "sld")
  1128. result3 = False
  1129. layer.saveSldStyle(sld_filename)
  1130. def json_export(self, layer_name):
  1131. filePath = self.getTempPath(layer_name)
  1132. ogr_driver_name = "GeoJSON"
  1133. project = QgsProject.instance()
  1134. fileNames = []
  1135. layer = QgsProject.instance().mapLayersByName(layer_name)[0]
  1136. layerType = layer.type()
  1137. if layerType == QgsMapLayer.VectorLayer:
  1138. renderer = layer.renderer()
  1139. hasIcon = False
  1140. if isinstance(renderer, QgsSingleSymbolRenderer):
  1141. self.copySymbols(renderer.symbol(), tempfile.gettempdir(), fileNames)
  1142. hasIcon = True
  1143. #layerCrs = qgis.utils.iface.activeLayer().crs().authid()
  1144. layerCrs = layer.crs().authid()
  1145. crs = QgsCoordinateReferenceSystem(layerCrs)# původně bylo
  1146. layer_filename = filePath
  1147. # crs = QgsCoordinateReferenceSystem(4326, QgsCoordinateReferenceSystem.EpsgCrsId)
  1148. ##crs = layer.crs().authid()
  1149. # tempOut = layer_filename.replace(".geojson", "_2.geojson")
  1150. result2 = qgis.core.QgsVectorFileWriter.writeAsVectorFormat(layer, layer_filename, "utf-8", crs, ogr_driver_name) # export jsonu do souboru
  1151. ##transformace
  1152. # input = tempOut
  1153. # out = layer_filename
  1154. # try:
  1155. # os.remove(layer_filename)
  1156. # except:
  1157. # pass
  1158. ## print("","-f", "Geojson", "-s_srs", layerCrs, "-t_srs", "epsg:4326", out, input)
  1159. # print("convert")
  1160. # print(main(["","-f", ogr_driver_name, "-s_srs", "EPSG:5514", "-t_srs", "EPSG:4326", out, input])) ## ogr2ogr
  1161. ##
  1162. sld_filename = filePath.replace("geojson", "sld")
  1163. result3 = False
  1164. layer.saveSldStyle(sld_filename)
  1165. def checkValidAttributes(self, layer_name):
  1166. layers = QgsProject.instance().mapLayersByName(layer_name)
  1167. if len(layers) > 1:
  1168. for l in layers:
  1169. if (isinstance(l, QgsVectorLayer)):
  1170. print("xxx")
  1171. layers.clear()
  1172. layers.append(l)
  1173. break
  1174. isValid = True
  1175. pom =layers[0].getFeature(1)
  1176. for nam in pom.fields().names():
  1177. if (nam == ''):
  1178. isValid = False
  1179. return isValid
  1180. def checkWgsExtent(self, layer):
  1181. WgsXmax = 180
  1182. WgsXmin = -180
  1183. WgsYmax = 90
  1184. WgsYmin = -90
  1185. extent = layer.extent()
  1186. if (extent.xMaximum() > WgsXmax or extent.xMinimum() < WgsXmin or extent.yMaximum() > WgsYmax or extent.yMinimum() < WgsYmin ):
  1187. return False
  1188. else:
  1189. return True
  1190. def callPostRequest(self, layers):
  1191. for item in layers:
  1192. print (item.text(0))
  1193. self.postRequest(item.text(0))
  1194. #def transformLayer2(self, layer):
  1195. # pass
  1196. def setCurrentLayer(name):
  1197. layers = QgsProject.instance().mapLayersByName(name)
  1198. if(len(layers) >1):
  1199. layerss = QgsProject.instance().mapLayers().values()
  1200. i = 0
  1201. for layer in layerss:
  1202. if layers[0].name == layer.name():
  1203. if (i == item.currentIndex()):
  1204. i = i + 1
  1205. break
  1206. i = i + 1
  1207. return i
  1208. def transformLayer(self, layer):
  1209. layer = iface.activeLayer()
  1210. crsSrc = layer.crs()
  1211. print(crsSrc.authid())
  1212. crsDest = QgsCoordinateReferenceSystem(4326, QgsCoordinateReferenceSystem.EpsgCrsId)
  1213. print(crsSrc, crsDest)
  1214. xform = QgsCoordinateTransform(crsSrc, crsDest, 5514, 4326)
  1215. feats=[]
  1216. if (layer.geometryType() == 0):
  1217. destLayer = QgsVectorLayer("Point?crs=EPSG:4326","lay","memory")
  1218. if (layer.geometryType() == 1):
  1219. destLayer = QgsVectorLayer("Linestring?crs=EPSG:4326","lay","memory")
  1220. if (layer.geometryType() == 2):
  1221. destLayer = QgsVectorLayer("Polygon?crs=EPSG:4326","lay","memory")
  1222. for f in layer.getFeatures():
  1223. g = f.geometry()
  1224. g.transform(xform)
  1225. print(g)
  1226. f.setGeometry(g)
  1227. feats.append(f)
  1228. destLayer.dataProvider().addFeatures(feats)
  1229. return destLayer
  1230. def postRequest(self, layer_name):
  1231. nameCheck = True
  1232. validExtent = True
  1233. layers = QgsProject.instance().mapLayersByName(layer_name)
  1234. # layer_name = self.removeUnacceptableChars(layer_name)
  1235. layers[0].setName(layer_name)
  1236. if len(layers) > 1:
  1237. for l in layers:
  1238. if (isinstance(l, QgsVectorLayer)):
  1239. layers.clear()
  1240. layers.append(l)
  1241. break
  1242. #if not (re.match('[a-zA-Z_]{1}', layer_name)): ## nesmí být nesmysl v názvu na prvním místě
  1243. # QMessageBox.information(None, "Layman", "Diacritics or number in first character is not allowed.")
  1244. # nameCheck = False
  1245. #if not (re.match('[a-zA-Z0-9]', layer_name)): ## není povolena diakritika
  1246. # QMessageBox.information(None, "Layman", "Diacritics is not allowed.")
  1247. # nameCheck = False
  1248. if (re.match('[0-9]{1}', layer_name)): ## nesmí být nesmysl v názvu na prvním místě
  1249. QMessageBox.information(None, "Layman", "Number in first character is not allowed.")
  1250. nameCheck = False
  1251. print(layer_name)
  1252. #if (layers[0].crs().authid() != 'EPSG:4323'):
  1253. # layers[0] = self.transformLayer(layers[0])
  1254. if not self.checkWgsExtent(layers[0]):
  1255. QMessageBox.information(None, "Layman", "Extent of layer is out of WGS 84 range. (EPSG: 4326)")
  1256. validExtent = False
  1257. if (nameCheck and validExtent):
  1258. crs = layers[0].crs().authid()
  1259. crs = "EPSG:4326"
  1260. data = { 'name' : str(layer_name), 'title' : str(layer_name), 'crs' : str(crs) }
  1261. if (self.checkValidAttributes(layer_name)):
  1262. if (self.checkExistingLayer(layer_name)):
  1263. print("vrstva již existuje")
  1264. msgbox = QMessageBox(QMessageBox.Question, "Layman", "Layer "+layer_name+" already exists in server. Do you want overwrite it´s geometry?")
  1265. msgbox.addButton(QMessageBox.Yes)
  1266. msgbox.addButton(QMessageBox.No)
  1267. msgbox.setDefaultButton(QMessageBox.No)
  1268. reply = msgbox.exec()
  1269. if (reply == QMessageBox.Yes):
  1270. print("vrstva již existuje")
  1271. self.json_export(layer_name)
  1272. geoPath = self.getTempPath(layer_name)
  1273. #try:
  1274. if (os.path.getsize(geoPath) > self.CHUNK_SIZE):
  1275. self.postInChunks(layer_name, "patch")
  1276. # iface.messageBar().pushWidget(iface.messageBar().createMessage("Import:", " Layer " + layer_name + " was imported successfully."), Qgis.Success, duration=3)
  1277. else:
  1278. self.patchLayer(layer_name, data)
  1279. iface.messageBar().pushWidget(iface.messageBar().createMessage("Import:", " Layer " + layer_name + " was imported successfully."), Qgis.Success, duration=3)
  1280. #except:
  1281. # iface.messageBar().pushWidget(iface.messageBar().createMessage("Import:", " Layer " + layer_name + " was not imported successfully."), Qgis.Warning, duration=3)
  1282. else:
  1283. pass
  1284. else:
  1285. self.layerName = layer_name
  1286. # if (self.checkEpsg(layer_name)):
  1287. if(True): ##Pravděpodobně nebude třeba testovat EPSG, pokud se detekuje, je vrstva transformována.
  1288. # print (layer_name)
  1289. self.json_export(layer_name)
  1290. sldPath = self.getTempPath(self.layerName).replace("geojson", "sld")
  1291. geoPath = self.getTempPath(self.layerName)
  1292. if (os.path.getsize(geoPath) > self.CHUNK_SIZE):
  1293. self.postInChunks(layer_name, "post")
  1294. else:
  1295. if(os.path.isfile(sldPath)): ## existuje sld?
  1296. files = [('file', open(geoPath, 'rb')), ('sld', open(sldPath, 'rb'))]
  1297. else:
  1298. files = {'file': (geoPath, open(geoPath, 'rb')),}
  1299. print(files)
  1300. response = requests.post(self.URI+'/rest/'+self.laymanUsername+'/layers', files=files, data = data, headers = self.authHeader)
  1301. print(response.content)
  1302. print(response.status_code)
  1303. if response.status_code == 200:
  1304. iface.messageBar().pushWidget(iface.messageBar().createMessage("Import:", " Layer " + layer_name + " was imported successfully."), Qgis.Success, duration=3)
  1305. else:
  1306. iface.messageBar().pushWidget(iface.messageBar().createMessage("Import:", " Layer " + layer_name + " was not imported."), Qgis.Warning, duration=3)
  1307. else:
  1308. # QMessageBox.information(None, "Atlas", "Vrstva "+layer_name+" nelze nahrát. Použíjte EPSG:4326 nebo EPSG:3857")
  1309. QMessageBox.information(None, "Layman", "Use EPSG:4326")
  1310. else:
  1311. QMessageBox.information(None, "Layman", "Layer "+layer_name+" does not have attributes!")
  1312. def addExistingLayerToComposite(self, name):
  1313. x = self.dlg.listWidget.currentRow()
  1314. if self.checkLayerOnLayman(name):
  1315. inComposite = name in self.isRasterLayerInComposite(x, name)
  1316. if not(inComposite):
  1317. response = requests.get(self.URI+'/rest/'+self.laymanUsername+'/layers/'+str(name), verify=False)
  1318. res = self.fromByteToJson(response.content)
  1319. print(res)
  1320. wmsUrl = res['wms']['url']
  1321. 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":{}})
  1322. self.importMap(x, 'add', 1)
  1323. self.refreshLayerList()
  1324. else:
  1325. QMessageBox.information(None, "Message", "Composition already include layer "+name+"!")
  1326. else:
  1327. QMessageBox.information(None, "Layman", "Something went wrong with this layer: "+name)
  1328. def addExistingMapToMemory(self, name):
  1329. response = requests.get(self.URI+'/rest/'+self.laymanUsername+'/maps/'+str(name), verify=False)
  1330. res = self.fromByteToJson(response.content)
  1331. print(res)
  1332. wmsUrl = res['wms']['url']
  1333. x = self.dlg.listWidget.currentRow()
  1334. 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":{}})
  1335. self.importMap(x, 'add', 1)
  1336. self.refreshLayerList()
  1337. def layerInfoRedirect(self, name):
  1338. url = self.URI+'/rest/'+self.laymanUsername+"/layers/" + name
  1339. print(url)
  1340. response = requests.get(url)
  1341. r = self.fromByteToJson(response.content)
  1342. try:
  1343. url = r['metadata']['record_url']
  1344. webbrowser.open(url, new=2) ## redirect na micku pro více info
  1345. except:
  1346. QMessageBox.information(None, "Layman", "Link is unavailable.")
  1347. def checkLayerOnLayman(self, layer_name):
  1348. url = self.URI+'/rest/'+self.laymanUsername+"/layers/"+layer_name
  1349. r = requests.get(url = url, verify=False)
  1350. #print(r.content)
  1351. data = r.json()
  1352. try:
  1353. if data['wms']['status'] == 'NOT_AVAILABLE' or data['wms']['status'] == 'PENDING':
  1354. return False
  1355. else:
  1356. return True
  1357. except:
  1358. return True # validní vrstva nemá status
  1359. def addLayerToComposite(self,x):
  1360. self.compositeListOld = copy.deepcopy(self.compositeList) ## list pred upravou pro vraceni zmen
  1361. #layers = self.getSelectedLayers() #odkomentovat pro zapnuti nacitani vrstev z listwidgetu
  1362. layers = []
  1363. layers.append(self.dlg.mMapLayerComboBox.currentLayer())
  1364. successful = 0
  1365. print (len(layers))
  1366. try:
  1367. step = self.getProgressBarStep(len(layers))
  1368. except:
  1369. QMessageBox.information(None, "Layman", "No layer selected!")
  1370. for i in range (0, len(layers)):
  1371. print(type(layers[i]))
  1372. print (self.isLayerInComposite(x))
  1373. print (layers[i].name() in self.isLayerInComposite(x))
  1374. inComposite = layers[i].name() in self.isLayerInComposite(x)
  1375. layerName = self.removeUnacceptableChars(layers[i].name())
  1376. if (self.checkExistingLayer(layers[i].name()) and inComposite):
  1377. j = self.getLayerInCompositePosition(x)
  1378. print("j je: " + str(j))
  1379. # QMessageBox.information(None, "Message", "Composition already include layer "+layers[i].name()+"!")
  1380. # self.deteteLayerFromComposite(x, j, layers[i].name())
  1381. self.postRequest(layers[i].name())
  1382. layerName = self.removeUnacceptableChars(layers[i].name())
  1383. wmsStatus = 'PENDING'
  1384. j = 0
  1385. while ((wmsStatus == 'PENDING') and (j < 10)):
  1386. print (self.URI+'/rest/'+self.laymanUsername+'/layers/'+str(layerName))
  1387. response = requests.get(self.URI+'/rest/'+self.laymanUsername+'/layers/'+str(layerName), verify=False)
  1388. res = self.fromByteToJson(response.content)
  1389. try:
  1390. wmsStatus = res['wms']['status']
  1391. except:
  1392. wmsStatus = "done"
  1393. time.sleep(1)
  1394. j = j + 1
  1395. #print(res)
  1396. #try:
  1397. # wmsUrl = res['wms']['url']
  1398. #except:
  1399. # wmsUrl = self.URI+'/geoserver/'+layers[i].name()+'/ows'
  1400. #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":{}})
  1401. #successful = successful + 1
  1402. #self.dlg.progressBar.setValue(self.dlg.progressBar.value()+step)
  1403. #self.importMap(x, "add", successful)
  1404. else:
  1405. self.postRequest(layers[i].name())
  1406. wmsStatus = 'PENDING'
  1407. j = 0
  1408. while ((wmsStatus == 'PENDING') and (j < 10)):
  1409. print (self.URI+'/rest/'+self.laymanUsername+'/layers/'+str(layerName))
  1410. response = requests.get(self.URI+'/rest/'+self.laymanUsername+'/layers/'+str(layerName), verify=False)
  1411. res = self.fromByteToJson(response.content)
  1412. try:
  1413. wmsStatus = res['wms']['status']
  1414. except:
  1415. wmsStatus = "done"
  1416. time.sleep(1)
  1417. j = j + 1
  1418. print(res)
  1419. try:
  1420. wmsUrl = res['wms']['url']
  1421. except:
  1422. wmsUrl = self.URI+'/geoserver/'+layerName+'/ows'
  1423. 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":{}})
  1424. successful = successful + 1
  1425. self.dlg.progressBar.setValue(self.dlg.progressBar.value()+step)
  1426. self.importMap(x, "add", successful)
  1427. def getProgressBarStep(self, count):
  1428. return (100/count)
  1429. def deteteLayerFromComposite(self, x, position, name): ## pro verzi s komboboxem
  1430. previousLayers = self.compositeList[x]['layers']
  1431. self.compositeList[x]['layers'] = []
  1432. for i in range (0,len(previousLayers)):
  1433. if (i != position):
  1434. self.compositeList[x]['layers'].append(previousLayers[i])
  1435. self.importMap(x, "del")
  1436. done = True
  1437. if (len(self.compositeList[x]['layers']) != 0 and len(self.compositeList[x]['layers']) != None):
  1438. for i in range (0, len(self.compositeList[x]['layers'])):
  1439. if (self.compositeList[x]['layers'][i]['params']['LAYERS'] == name):
  1440. done = False
  1441. if (done and QgsProject.instance().mapLayersByName(name)): ## kompozice může mít načtené vrstvy, které nejsou v canvasu
  1442. self.deleteLayerFromCanvas(name)
  1443. def importCleanComposite(self,x):
  1444. tempFile = tempfile.gettempdir() + os.sep + "atlas" + os.sep + "compsite.json"
  1445. with open(tempFile, 'w') as outfile:
  1446. json.dump(self.compositeList[x], outfile)
  1447. with open(tempFile, 'rb') as f:
  1448. d = json.load(f)
  1449. files = {'file': (tempFile, open(tempFile, 'rb')),}
  1450. data = { 'name' : self.compositeList[x]['name'], 'title' : self.compositeList[x]['title'], 'description' : self.compositeList[x]['abstract']}
  1451. print(self.URI+'/rest/'+self.laymanUsername+'/maps/')
  1452. #response = requests.post(self.URI+'/rest/'+self.laymanUsername+'/maps/', data = data, files=files, headers = self.authHeader, verify=False)
  1453. print(self.authHeader)
  1454. response = requests.post(self.URI+'/rest/'+self.laymanUsername+'/maps', files=files, data = data, headers = self.authHeader, verify=False)
  1455. print(response.content)
  1456. def showProgressBar(self, bar):
  1457. bar = QProgressBar()
  1458. bar.setRange(0,0) ## range 0,0 je nekonečný
  1459. self.showProgressBar(bar)
  1460. bar.show()
  1461. iface.mainWindow().statusBar().addWidget(bar)
  1462. def fromByteToJson(self, res):
  1463. pom = res
  1464. pom = pom.decode('utf_8')
  1465. pom = json.loads(pom)
  1466. return pom
  1467. def importMap(self, x, operation, s = 0): ##s je počet vrstev úspěšně nahraných na server
  1468. if (s != 0):
  1469. self.dlg.progressBar.show()
  1470. self.dlg.label_import.show()
  1471. tempFile = tempfile.gettempdir() + os.sep + "atlas" + os.sep + "compsite.json"
  1472. try:
  1473. existingLayers = self.isLayerInComposite(x)
  1474. except:
  1475. existingLayers = []
  1476. successful = s
  1477. with open(tempFile, 'w') as outfile:
  1478. json.dump(self.compositeList[x], outfile)
  1479. jsonPath = tempfile.gettempdir() + os.sep + "atlas" + os.sep + "compsite.json"
  1480. with open(jsonPath, 'rb') as f:
  1481. d = json.load(f)
  1482. files = {'file': (jsonPath, open(jsonPath, 'rb')),}
  1483. data = { 'name' : self.compositeList[x]['name'], 'title' : self.compositeList[x]['title'], 'description' : self.compositeList[x]['abstract']}
  1484. req = requests.get(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'])
  1485. mapCode = req.status_code ## test jestli vrstva na serveru existuje. Pokud ne = error 404
  1486. if (mapCode == 404):
  1487. if (operation == "add"):
  1488. msgbox = QMessageBox(QMessageBox.Question, "Import Map", "Chcete přidat do kompozice "+str(successful)+" vrstev?")
  1489. if (operation == "mod"):
  1490. #msgbox = QMessageBox(QMessageBox.Question, "Import Map", "Do you want save changes?")
  1491. self.dlg.close()
  1492. self.afterCloseEditMapDialog()
  1493. print(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'])
  1494. response = requests.delete(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'],headers = self.authHeader, verify=False)
  1495. print(response.content)
  1496. response = requests.post(self.URI+'/rest/'+self.laymanUsername+'/maps', files=files, data = data, headers = self.authHeader)
  1497. self.dlg.show()
  1498. return
  1499. if (operation == "del"):
  1500. msgbox = QMessageBox(QMessageBox.Question, "Import Map", "Chcete smazat vybranou vrstvu?")
  1501. try: # jedná se o mod pokud je except
  1502. msgbox.addButton(QMessageBox.Yes)
  1503. msgbox.addButton(QMessageBox.No)
  1504. msgbox.setDefaultButton(QMessageBox.No)
  1505. reply = msgbox.exec()
  1506. except:
  1507. pass
  1508. if (reply == QMessageBox.Yes):
  1509. response = requests.post(self.URI+'/rest/'+self.laymanUsername+'/maps', files=files, data = data, headers = self.authHeader)
  1510. print(response.content)
  1511. data = response.content
  1512. self.dlg.progressBar.hide()
  1513. # self.refreshItems()## refresh formuláře
  1514. if (operation != "mod"):
  1515. self.refreshLayerList()
  1516. else:
  1517. if (operation == "add"):
  1518. self.dlg.progressBar.setValue(0)
  1519. else:
  1520. print(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'])
  1521. files = {'file': (jsonPath, open(jsonPath, 'rb')),}
  1522. response = requests.patch(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'], data = data, files=files, headers = self.authHeader, verify=False)
  1523. try:
  1524. self.refreshLayerList()
  1525. except:
  1526. ## nacházíme se v jiném formuláři, není co refreshovat
  1527. pass
  1528. #if (operation == "add"):
  1529. # print(len(existingLayers))
  1530. # msgbox = QMessageBox(QMessageBox.Question, "Layman", "Vybraná vrstva bude přidána do kompozice.")
  1531. # #if(len(existingLayers) == 0):
  1532. # # #if (successful != 0):
  1533. # # msgbox = QMessageBox(QMessageBox.Question, "Layman", "Vybraná vrstva bude přidána do kompozice.")
  1534. # #else:
  1535. # # # msgbox = QMessageBox(QMessageBox.Question, "Import Map", "Kompozice již obsauje vrstvy: "+(', '.join(existingLayers))+". Chcete vybrané vrstvy opravdu znovu přidat?.")
  1536. # # msgbox = QMessageBox(QMessageBox.Question, "Layman", "Vybraná vrstva bude přidána do kompozice.")
  1537. #if (operation == "del"):
  1538. # msgbox = QMessageBox(QMessageBox.Question, "Import Map", "Chcete smazat vybranou vrstvu?")
  1539. #if (operation == "mod"):
  1540. # msgbox = QMessageBox(QMessageBox.Question, "Import Map", "Chcete uložit upravenou mapu?")
  1541. #msgbox.addButton(QMessageBox.Yes)
  1542. #msgbox.addButton(QMessageBox.No)
  1543. #msgbox.setDefaultButton(QMessageBox.No)
  1544. #reply = msgbox.exec()
  1545. #if (reply == QMessageBox.Yes):
  1546. # print(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'])
  1547. # files = {'file': (jsonPath, open(jsonPath, 'rb')),}
  1548. # response = requests.patch(self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name'], data = data, files=files, headers = self.authHeader, verify=False)
  1549. # try:
  1550. # self.refreshLayerList()
  1551. # except:
  1552. # ## nacházíme se v jiném formuláři, není co refreshovat
  1553. # pass
  1554. #if (reply == QMessageBox.No):
  1555. # self.compositeList = copy.deepcopy(self.compositeListOld)
  1556. # #self.refreshItems()##refresh formulare
  1557. # self.refreshLayerList()
  1558. try: ## některé formuláře nemají progress bar
  1559. self.dlg.progressBar.hide()
  1560. self.dlg.label_import.hide()
  1561. except:
  1562. pass
  1563. def deleteLayer(self, layerName):
  1564. url = self.URI+'/rest/'+self.laymanUsername+"/layers/" + layerName
  1565. r = requests.delete(url)
  1566. print (r.content)
  1567. def getActiveLayer(self):
  1568. layer = self.iface.activeLayer()
  1569. return layer
  1570. def createComposite(self, name, title):
  1571. # if (name == "" or title == ""):
  1572. if (title == ""):
  1573. QMessageBox.information(None, "Message", "Není vyplněn titulek!")
  1574. else:
  1575. name = self.removeUnacceptableChars(title)
  1576. data = self.getEmptyComposite(name,title)
  1577. print (data)
  1578. self.compositeList.append(data)
  1579. x = len(self.compositeList) - 1
  1580. self.importCleanComposite(x)
  1581. try:
  1582. self.refreshCompositeList()
  1583. except:
  1584. pass
  1585. def patchLayer(self, layer_name, data):
  1586. self.layerName = layer_name
  1587. self.json_export(layer_name)
  1588. sldPath = self.getTempPath(self.layerName).replace("geojson", "sld")
  1589. geoPath = self.getTempPath(self.layerName)
  1590. print(sldPath)
  1591. if(os.path.isfile(sldPath)): ## existuje sld?
  1592. files = [('file', open(geoPath, 'rb')), ('sld', open(sldPath, 'rb'))]
  1593. print("sld")
  1594. else:
  1595. files = {'file': (geoPath, open(geoPath, 'rb')),}
  1596. print("ne sld")
  1597. url = self.URI+'/rest/'+self.laymanUsername+"/layers/" + layer_name
  1598. print(url)
  1599. r = requests.patch(url, files=files, data = data, headers = self.authHeader, verify=False)
  1600. print(r.content)
  1601. def getTempPath(self, name):
  1602. print (name)
  1603. if type(name) is tuple:
  1604. return (name[0] + name[1])
  1605. else:
  1606. tempFile = tempfile.gettempdir() + os.sep + name +'.geojson'
  1607. return tempFile
  1608. def sendLayer(self):
  1609. layer = self.getActiveLayer()
  1610. if (layer == None):
  1611. QMessageBox.information(None, "Message", "Neexistuje žádná vrstva k uložení")
  1612. else:
  1613. self.json_export()
  1614. self.postRequest()
  1615. def getExistingLayers(self):
  1616. url = self.URI+'/rest/'+self.laymanUsername+"/layers"
  1617. r = requests.get(url = url)
  1618. data = r.json()
  1619. for x in range(len(data)):
  1620. print(data[x]['name'])
  1621. return data
  1622. def checkExistingLayer(self, layerName):
  1623. layerName = self.removeUnacceptableChars(layerName)
  1624. url = self.URI+'/rest/'+self.laymanUsername+"/layers"
  1625. r = requests.get(url = url)
  1626. data = r.json()
  1627. pom = set()
  1628. for x in range(len(data)):
  1629. pom.add((data[x]['name']))
  1630. layerName = layerName.replace(" ", "_").lower()
  1631. print(layerName)
  1632. print(pom)
  1633. if (layerName in pom):
  1634. return True
  1635. else:
  1636. return False
  1637. def removeUnacceptableChars(self, input):
  1638. input = input.lower()
  1639. input = input.replace("ř","r")
  1640. input = input.replace("š","s")
  1641. input = input.replace("ž","z")
  1642. input = input.replace("ů","u")
  1643. input = input.replace("ú","u")
  1644. input = input.replace(" ","_")
  1645. input = input.replace("é","e")
  1646. input = input.replace("í","i")
  1647. input = input.replace("ý","y")
  1648. input = input.replace("á","a")
  1649. input = input.replace("ó","o")
  1650. input = input.replace("č","c")
  1651. input = input.replace("ď","d")
  1652. input = input.replace("ě","e")
  1653. input = input.replace("ť","t")
  1654. input = re.sub(r'[?|$|.|!]',r'',input)
  1655. # iface.messageBar().pushWidget(iface.messageBar().createMessage("Layman:", "Diacritics in name of layer was replaced."), Qgis.Success, duration=3)
  1656. return input
  1657. def removeUnacceptableChars2(self, input):
  1658. input = input.encode('utf-8')
  1659. input = input.hex()
  1660. # iface.messageBar().pushWidget(iface.messageBar().createMessage("Layman:", "Diacritics in name of layer was replaced."), Qgis.Success, duration=3)
  1661. return input
  1662. def test (self):
  1663. item = self.dlgGetLayers.items.currentItem().text()
  1664. print( item)
  1665. def isLayerInComposite(self, x):
  1666. layers =[]
  1667. layers.append(self.dlg.mMapLayerComboBox.currentLayer())
  1668. print (self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name']+'/file')
  1669. req = requests.get (self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name']+'/file')
  1670. data = req.json()
  1671. print (data['abstract'])
  1672. print (len(self.compositeList[x]['layers']))
  1673. existingLayers = []
  1674. for i in range (0, len(data['layers'])):
  1675. for j in range (0, len(layers)):
  1676. if (data['layers'][i]['params']['LAYERS'] in layers[j].name() ):
  1677. existingLayers.append(data['layers'][i]['params']['LAYERS'])
  1678. return existingLayers
  1679. def getLayerInCompositePosition(self, x):
  1680. layers =[]
  1681. layers.append(self.dlg.mMapLayerComboBox.currentLayer())
  1682. print (self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name']+'/file')
  1683. req = requests.get (self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name']+'/file')
  1684. data = req.json()
  1685. print (data['abstract'])
  1686. print (len(self.compositeList[x]['layers']))
  1687. existingLayers = []
  1688. j = 0
  1689. for i in range (0, len(data['layers'])):
  1690. for j in range (0, len(layers)):
  1691. if (data['layers'][i]['params']['LAYERS'] in layers[j].name() ):
  1692. # existingLayers.append(data['layers'][i]['params']['LAYERS'])
  1693. j = i
  1694. print("founded")
  1695. return j
  1696. def isRasterLayerInComposite(self, x, name):
  1697. req = requests.get (self.URI+'/rest/'+self.laymanUsername+'/maps/'+self.compositeList[x]['name']+'/file')
  1698. data = req.json()
  1699. existingLayers = []
  1700. for i in range (0, len(data['layers'])):
  1701. if (data['layers'][i]['params']['LAYERS'] == name ):
  1702. existingLayers.append(data['layers'][i]['params']['LAYERS'])
  1703. return existingLayers
  1704. def addComposite(self, data, service, groupName = ''):
  1705. #self.compositeList.append (data)
  1706. print(len(data['layers']))
  1707. for x in range(0, len(data['layers'])):
  1708. # repairUrl = self.convertUrlFromHex(data['layers'][x]['url'])
  1709. repairUrl = self.URI+"/geoserver/"+self.laymanUsername+"/ows"
  1710. layerName = data['layers'][x]['params']['LAYERS']
  1711. format = data['layers'][x]['params']['FORMAT']
  1712. # epsg = str(data['groups']['projection']).upper()
  1713. epsg = 'EPSG:4326'
  1714. #abstract = data['data']['layers'][x]['params']['ABSTRACT']
  1715. wmsName = data['layers'][x]['params']['LAYERS']
  1716. print("groupName " +groupName)
  1717. print("groupName " +layerName)
  1718. print("groupName " +format)
  1719. print("groupName " +epsg)
  1720. layerNameTitle = self.getLayerTitle(layerName)
  1721. print("groupName " +layerNameTitle)
  1722. if service == 'WMS':
  1723. self.loadWms(repairUrl, layerName,layerNameTitle, format,epsg, groupName)
  1724. if service == 'WFS':
  1725. self.loadWfs(repairUrl, layerName,layerNameTitle, groupName)
  1726. def loadService(self, data, service, groupName = ''):
  1727. #print(len(data['layers']))
  1728. for x in range(len(data['layers'])- 1, -1, -1): ## descending order
  1729. #repairUrl = self.convertUrlFromHex(data['layers'][x]['url'])
  1730. repairUrl = self.URI+"/geoserver/"+self.laymanUsername+"/ows"
  1731. layerName = data['layers'][x]['params']['LAYERS']
  1732. format = data['layers'][x]['params']['FORMAT']
  1733. # epsg = str(data['groups']['projection']).upper()
  1734. epsg = 'EPSG:4326'
  1735. #abstract = data['data']['layers'][x]['params']['ABSTRACT']
  1736. wmsName = data['layers'][x]['params']['LAYERS']
  1737. if self.checkLayerOnLayman(layerName):
  1738. if service == 'WMS':
  1739. print("zzzzzzz" +layerName)
  1740. self.loadWms(repairUrl, layerName, format,epsg, groupName)
  1741. if service == 'WFS':
  1742. self.loadWfs(repairUrl, layerName, groupName)
  1743. else:
  1744. QMessageBox.information(None, "Layman", "Layer: "+layerName + " is corrupted and will not be loaded.")
  1745. def getLayerTitle(self, layerName):
  1746. layerName = self.removeUnacceptableChars(layerName)
  1747. url = self.URI+'/rest/'+self.laymanUsername+'/layers/'+layerName
  1748. r = requests.get(url = url)
  1749. data = r.json()
  1750. print(data)
  1751. title = data['title']
  1752. return title
  1753. def loadWms(self, url, layerName,layerNameTitle, format, epsg, groupName = ''):
  1754. print(url)
  1755. layerName = self.removeUnacceptableChars(layerName)
  1756. epsg = "EPSG:4326"
  1757. urlWithParams = 'contextualWMSLegend=0&crs='+epsg+'&IgnoreReportedLayerExtents=1&dpiMode=7&featureCount=10&format=image/png&layers='+layerName+'&styles=&url=' + url
  1758. print(urlWithParams)
  1759. rlayer = QgsRasterLayer(urlWithParams, layerNameTitle, 'wms')
  1760. try:
  1761. print("extents")
  1762. print(rlayer.ignoreExtents())
  1763. except:
  1764. print("ignoreExtents works only with qgis 3.10 and higher")
  1765. pass # pro qgis 3.10 a vys
  1766. if (groupName != ''):
  1767. self.addWmsToGroup(groupName,rlayer)
  1768. else:
  1769. QgsProject.instance().addMapLayer(rlayer)
  1770. def loadWfs(self, url, layerName,layerNameTitle, groupName = ''):
  1771. layerName = self.removeUnacceptableChars(layerName)
  1772. epsg = 'EPSG:4326'
  1773. uri = self.URI+"/geoserver/"+self.laymanUsername+"/ows?srsname="+epsg+"&typename="+self.laymanUsername+":"+layerName+"&restrictToRequestBBOX=1&pagingEnabled=True&version=auto&request=GetFeature&service=WFS"
  1774. print(uri)
  1775. vlayer = QgsVectorLayer(uri, layerNameTitle, "WFS")
  1776. print(vlayer.isValid())
  1777. if (groupName != ''):
  1778. self.addWmsToGroup(groupName,vlayer)
  1779. else:
  1780. QgsProject.instance().addMapLayer(vlayer)
  1781. def addWmsToGroup(self, groupName, layer):
  1782. root = QgsProject.instance().layerTreeRoot()
  1783. group = root.findGroup(groupName)
  1784. if not(group):
  1785. group = root.addGroup(groupName)
  1786. group = self.reorderToTop(groupName)
  1787. QgsProject.instance().addMapLayer(layer,False)
  1788. group.insertChildNode(0,QgsLayerTreeLayer(layer))
  1789. def reorderToTop(self, name):
  1790. root = QgsProject.instance().layerTreeRoot()
  1791. for ch in root.children():
  1792. if ch.name() == name:
  1793. _ch = ch.clone()
  1794. root.insertChildNode(0, _ch)
  1795. root.removeChildNode(ch)
  1796. return _ch
  1797. def convertUrlFromHex(self, url):
  1798. url = url.replace('%3A',':')
  1799. url = url.replace('%2F','/')
  1800. url = url.replace('%3F','?')
  1801. url = url.replace('%3D','=')
  1802. url = url.replace('%26','&')
  1803. return url
  1804. def checkEpsg(self, name):
  1805. ret = False
  1806. layer = QgsProject.instance().mapLayersByName(name)
  1807. layerCRS = layer[0].crs().authid()
  1808. if ((layerCRS == 'EPSG:4326') or (layerCRS == 'EPSG:3857')):
  1809. ret = True
  1810. return ret
  1811. def registerLayer(self, name):
  1812. sldPath = self.getTempPath(name).replace("geojson", "sld")
  1813. geoPath = self.getTempPath(name)
  1814. url = "http://layman.lesprojekt.cz/rest/"+self.laymanUsername+"/layers"
  1815. files = {'sld': (sldPath, open(sldPath, 'rb')),} # nahrávám sld
  1816. payload = {
  1817. 'file': name+".geojson"
  1818. }
  1819. response = requests.request("POST", url, files = files, data=payload, headers = self.authHeader)
  1820. # print(response.text)
  1821. def read_in_chunks(self, file_object): ## cca 1MB chunk převzato z laymana test klienta
  1822. chunk_size=self.CHUNK_SIZE
  1823. while True:
  1824. data = file_object.read(chunk_size)
  1825. if not data:
  1826. break
  1827. yield data
  1828. def postInChunks(self, layer_name, reqType):
  1829. print(reqType)
  1830. if (reqType == "patch"):
  1831. url = self.URI+'/rest/'+self.laymanUsername+'/layers/'+layer_name
  1832. r = requests.delete(url,headers = self.authHeader)
  1833. # print(r.content)
  1834. self.registerLayer(layer_name)
  1835. filePath = os.path.join(tempfile.gettempdir(), "atlas_chunks" ) ## chunky se ukládají do adresáře v tempu
  1836. if not (os.path.exists(filePath)):
  1837. os.mkdir(filePath)
  1838. file = self.getTempPath(layer_name)
  1839. f = open(file, 'rb')
  1840. arr = []
  1841. for piece in self.read_in_chunks(f):
  1842. arr.append(piece)
  1843. url = self.URI+'/rest/'+self.laymanUsername+'/layers/'+layer_name+'/chunk'
  1844. resumableFilename = layer_name+'.geojson'
  1845. layman_original_parameter = "file"
  1846. resumableTotalChunks = len(arr)
  1847. try:
  1848. for i in range (1, len(arr)+1): ##chunky jsou počítané od 1 proto +1
  1849. print("chunk" + str(i))
  1850. file = arr[i-1] # rozsekaná část souboru
  1851. resumableChunkNumber = i # cislo casti
  1852. payload = {
  1853. 'file' : "chunk"+str(i)+".geojson",
  1854. 'resumableFilename': resumableFilename,
  1855. 'layman_original_parameter': layman_original_parameter,
  1856. 'resumableChunkNumber': i,
  1857. 'resumableTotalChunks': resumableTotalChunks
  1858. }
  1859. f = open(filePath + os.sep+"chunk"+str(i)+".geojson", "wb")
  1860. f.write(bytearray(arr[i-1]))
  1861. f.close()
  1862. files = {'file': (layer_name+".geojson", open(filePath +os.sep+ "chunk"+str(i)+".geojson", 'rb')),}
  1863. response = requests.post(url, files = files, data=payload, headers = self.authHeader)
  1864. print(response.content)
  1865. iface.messageBar().pushWidget(iface.messageBar().createMessage("Import:", " Layer " + layer_name + " was imported successfully."), Qgis.Success, duration=3)
  1866. except:
  1867. iface.messageBar().pushWidget(iface.messageBar().createMessage("Import:", " Layer " + layer_name + " was not imported successfully!"), Qgis.Warning, duration=3)
  1868. ############################################# auth part############################
  1869. def startThread(self):
  1870. self.thread1 = threading.Thread(target=self.refreshToken)
  1871. self.thread1.start()
  1872. def refreshToken(self):
  1873. #self.expires_in = 60
  1874. path = tempfile.gettempdir() + os.sep + "atlas" + os.sep + "tsts.txt"
  1875. i = 0
  1876. tokenEndpoint = "https://www."+self.liferayServer+"/o/oauth2/token"
  1877. while (i < self.expires_in):
  1878. f = open(path, "a")
  1879. f.write(str(i))
  1880. f.close()
  1881. time.sleep(1)
  1882. if i == self.expires_in - 4:
  1883. data = {'grant_type':'refresh_token',
  1884. 'refresh_token': self.refresh_token,
  1885. 'client_id':'id-3462f94b-875c-9185-4ced-b69841f24b3',
  1886. 'redirect_uri':'http://localhost:3000/client/authn/oauth2-liferay/callback',
  1887. 'code_verifier':'test'
  1888. }
  1889. r = requests.post(url = tokenEndpoint, data = data, headers = self.authHeader)
  1890. res = self.fromByteToJson(r.content)
  1891. self.access_token = res['access_token']
  1892. self.refresh_token = res['refresh_token']
  1893. self.expires_in = res['expires_in']
  1894. print(self.access_token)
  1895. self.setAuthHeader()
  1896. i = 0
  1897. else:
  1898. i += 1
  1899. def authOptained (self):
  1900. self.dlg.pushButton_Continue.setEnabled(True)
  1901. self.menu_Connection.setEnabled(True)
  1902. self.menu_saveLocalFile.setEnabled(True)
  1903. self.menu_loadJson.setEnabled(True)
  1904. self.menu_ImportLayerDialog.setEnabled(True)
  1905. self.menu_AddLayerDialog.setEnabled(True)
  1906. self.menu_AddMapDialog.setEnabled(True)
  1907. self.menu_ImportLayerDialog.setEnabled(True)
  1908. self.menu_ImportMapDialog.setEnabled(True)
  1909. # self.menu_DeleteMapDialog.setEnabled(True)
  1910. # self.menu_CreateCompositeDialog.setEnabled(True)
  1911. self.menu_UserInfoDialog.setEnabled(True)
  1912. self.textbox.setText("Layman: Logged user")
  1913. def getCodeVerifier(self):
  1914. code_verifier = base64.urlsafe_b64encode(os.urandom(40)).decode('utf-8')
  1915. code_verifier = re.sub('[^a-zA-Z0-9]+', '', code_verifier)
  1916. self.code_verifier = code_verifier
  1917. return (code_verifier)
  1918. def getCodeChallenge(self, code_verifier):
  1919. code_challenge = hashlib.sha256(code_verifier.encode('utf-8')).digest()
  1920. code_challenge = base64.urlsafe_b64encode(code_challenge).decode('utf-8')
  1921. code_challenge = code_challenge.replace('=', '')
  1922. self.code_challenge = code_challenge
  1923. return (code_challenge)
  1924. def printVariables(self):
  1925. print (self.access_token)
  1926. print(self.refresh_token)
  1927. print(self.authHeader)
  1928. print(self.laymanUsername)
  1929. def getAuthCode(self):
  1930. path = tempfile.gettempdir() + os.sep + "atlas" + os.sep + "auth.txt"
  1931. f = open(path, "r")
  1932. ret = f.read()
  1933. f.close()
  1934. return ret
  1935. def getToken(self):
  1936. self.saveIni()
  1937. tokenEndpoint = "https://www."+self.liferayServer+"/o/oauth2/token"
  1938. # data to be sent to api
  1939. data = {'grant_type':'authorization_code',
  1940. 'client_id': self.client_id,
  1941. 'redirect_uri':'http://localhost:3000/client/authn/oauth2-liferay/callback',
  1942. 'code_verifier': self.code_verifier,
  1943. 'code': self.getAuthCode()}
  1944. # sending post request and saving response as response object
  1945. r = requests.post(url = tokenEndpoint, data = data, headers = self.authHeader)
  1946. res = self.fromByteToJson(r.content)
  1947. print (res)
  1948. try:
  1949. print(res['access_token'])
  1950. print(res['expires_in'])
  1951. print(res['refresh_token'])
  1952. except:
  1953. QMessageBox.information(None, "Message", "Autorization was not sucessfull! Please try it again.")
  1954. return
  1955. self.access_token = res['access_token']
  1956. self.refresh_token = res['refresh_token']
  1957. self.expires_in = res['expires_in']
  1958. self.setAuthHeader()
  1959. data ={
  1960. 'access_token': res['access_token'],
  1961. 'refresh_token': res['refresh_token'],
  1962. 'expires_in':res['expires_in']
  1963. }
  1964. path = tempfile.gettempdir() + os.sep + "atlas" + os.sep + "tokens.json"
  1965. with open(path, 'w') as outfile:
  1966. json.dump(data, outfile)
  1967. self.startThread()
  1968. self.registerUserIfNotExists()
  1969. self.dlg.close()
  1970. def setAuthHeader(self):
  1971. self.authHeader ={
  1972. "Authorization": "Bearer " + self.access_token,
  1973. "AuthorizationIssUrl" : 'https://www.'+self.liferayServer+'/o/oauth2/authorize'
  1974. }
  1975. def registerUserIfNotExists(self):
  1976. userEndpoint = "http://layman.lesprojekt.cz/rest/current-user"
  1977. #self.laymanUsername = self.dlg.lineEdit_userName.text()
  1978. id = self.client_id.replace('-', '')
  1979. user = {'username':self.Agrimail}
  1980. #user = self.Agrimail
  1981. print("authheader: "+ str(self.authHeader))
  1982. r = requests.patch(url = userEndpoint, data = user, headers = self.authHeader)
  1983. res = r.text
  1984. res = self.fromByteToJson(r.content)
  1985. print (res)
  1986. print(user)
  1987. print(res)
  1988. try:
  1989. # if res['message'] == 'User already reserved username.': # res['code'] == 34, code 35 je pokud již jiný uživatel má účet, který chceme registrovat
  1990. 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)
  1991. print("user exists")
  1992. self.laymanUsername = res['detail']['username']
  1993. #self.laymanUsername = res['username']
  1994. #else:
  1995. # self.laymanUsername = user['username']
  1996. except:
  1997. print("creating new user: " + res['username'])
  1998. self.laymanUsername = res['username']
  1999. #API_ENDPOINT = "http://layman.lesprojekt.cz/rest/current-user"
  2000. #headers ={
  2001. # "Authorization": "Bearer "+ self.access_token,
  2002. # "AuthorizationIssUrl" : 'https://www.'+self.liferayServer+'/o/oauth2/authorize'
  2003. #}
  2004. def openAuthLiferayUrl(self):
  2005. self.liferayServer = self.dlg.lineEdit_server.text()
  2006. self.URI = self.dlg.lineEdit_serverLayman.text()
  2007. self.getCodeChallenge(self.getCodeVerifier()) ##generování kódů
  2008. self.client_id = self.dlg.lineEdit_AgriID.text()
  2009. path = tempfile.gettempdir() + os.sep + "atlas" + os.sep + "auth.txt"
  2010. self.watcher = QFileSystemWatcher()
  2011. self.watcher.addPath(path)
  2012. self.watcher.fileChanged.connect(self.authOptained)
  2013. #################
  2014. print ("##### flask server is starting #####")
  2015. self.flaskThread = StartFlaskDaemon()
  2016. self.flaskThread.daemon = True
  2017. self.flaskThread.start()
  2018. 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'
  2019. webbrowser.open(url, new=2)
  2020. def saveIni(self):
  2021. file = os.getenv("HOME") + os.sep + ".layman" + os.sep + 'layman_user.INI'
  2022. dir = os.getenv("HOME") + os.sep + ".layman"
  2023. if not (os.path.isdir(dir)):
  2024. try:
  2025. os.mkdir(dir)
  2026. except OSError:
  2027. print ("vytváření adresáře selhalo")
  2028. config = configparser.ConfigParser()
  2029. config['DEFAULT']['login'] = self.Agrimail
  2030. config['DEFAULT']['id'] = self.client_id
  2031. print("saveliferay" + self.liferayServer)
  2032. config['DEFAULT']['server'] = self.liferayServer
  2033. config['DEFAULT']['layman'] = self.URI
  2034. with open(file, 'w') as configfile:
  2035. config.write(configfile)
  2036. def loadIni(self):
  2037. file = os.getenv("HOME") + os.sep + ".layman" + os.sep +'layman_user.INI'
  2038. config = configparser.ConfigParser()
  2039. config.read(file)
  2040. return config
  2041. def layerChanged(self):
  2042. print(iface.activeLayer())
  2043. if (iface.activeLayer() != None):
  2044. self.menu_saveLocalFile.setEnabled(True)
  2045. else:
  2046. self.menu_saveLocalFile.setEnabled(False)
  2047. def run(self):
  2048. """Run method that loads and starts the plugin"""
  2049. if not self.pluginIsActive:
  2050. self.pluginIsActive = True
  2051. #print "** STARTING Atlas"
  2052. # dockwidget may not exist if:
  2053. # first run of plugin
  2054. # removed on close (see self.onClosePlugin method)
  2055. if self.dockwidget == None:
  2056. # Create the dockwidget (after translation) and keep reference
  2057. self.dockwidget = AtlasDockWidget()
  2058. # connect to provide cleanup on closing of dockwidget
  2059. self.dockwidget.closingPlugin.connect(self.onClosePlugin)
  2060. self.dockwidget.pushButton.clicked.connect(self.sendLayer)
  2061. self.dockwidget.pushButton_2.clicked.connect(self.loadLocalFile)
  2062. self.dockwidget.pushButton_3.clicked.connect(self.saveLocalFile)
  2063. self.dockwidget.pushButton_getLayers.clicked.connect(self.run_getLayer)
  2064. self.dlgGetLayers.pushButtonxx.clicked.connect(lambda: print(self.dlgGetLayers.items.currentItem().text()))
  2065. # show the dockwidget
  2066. # TODO: fix to allow choice of dock location
  2067. self.iface.addDockWidget(Qt.TopDockWidgetArea, self.dockwidget)
  2068. self.dockwidget.show()
  2069. class StartFlaskDaemon(threading.Thread):
  2070. def run(self):
  2071. from subprocess import Popen, PIPE
  2072. import platform
  2073. import sys
  2074. from pathlib import Path
  2075. d= os.path.dirname(Path(__file__).absolute()) + os.sep+"flask_listener" + os.sep + "flask_listener.py"
  2076. if platform.system() == 'Windows':
  2077. process = subprocess.Popen(["python", d],stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
  2078. else:
  2079. process = subprocess.Popen(["python " + d],stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
  2080. stdout, stderr = process.communicate()
  2081. print (stdout, stderr)
  2082. '''Start your thread here'''