mirror of
https://github.com/qgis/QGIS.git
synced 2025-03-06 00:05:02 -05:00
300 lines
11 KiB
Python
300 lines
11 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
EditScriptDialog.py
|
|
---------------------
|
|
Date : December 2012
|
|
Copyright : (C) 2012 by Alexander Bruy
|
|
Email : alexander dot bruy at gmail dot com
|
|
***************************************************************************
|
|
* *
|
|
* This program is free software; you can redistribute it and/or modify *
|
|
* it under the terms of the GNU General Public License as published by *
|
|
* the Free Software Foundation; either version 2 of the License, or *
|
|
* (at your option) any later version. *
|
|
* *
|
|
***************************************************************************
|
|
"""
|
|
|
|
__author__ = 'Alexander Bruy'
|
|
__date__ = 'December 2012'
|
|
__copyright__ = '(C) 2012, Alexander Bruy'
|
|
|
|
# This will get replaced with a git SHA1 when you do a git archive
|
|
|
|
__revision__ = '$Format:%H$'
|
|
|
|
import os
|
|
import codecs
|
|
|
|
from qgis.PyQt import uic
|
|
from qgis.PyQt.QtCore import Qt
|
|
from qgis.PyQt.QtGui import QCursor
|
|
from qgis.PyQt.QtWidgets import (QMessageBox,
|
|
QFileDialog)
|
|
|
|
from qgis.gui import QgsGui
|
|
from qgis.core import QgsApplication, QgsSettings
|
|
from qgis.utils import iface, OverrideCursor
|
|
|
|
from processing.gui.AlgorithmDialog import AlgorithmDialog
|
|
from processing.script import ScriptUtils
|
|
|
|
pluginPath = os.path.split(os.path.dirname(__file__))[0]
|
|
WIDGET, BASE = uic.loadUiType(
|
|
os.path.join(pluginPath, "ui", "DlgScriptEditor.ui"))
|
|
|
|
|
|
class ScriptEditorDialog(BASE, WIDGET):
|
|
hasChanged = False
|
|
|
|
def __init__(self, filePath=None, parent=None):
|
|
super(ScriptEditorDialog, self).__init__(parent)
|
|
self.setupUi(self)
|
|
|
|
QgsGui.instance().enableAutoGeometryRestore(self)
|
|
|
|
#~ self.setWindowFlags(Qt.WindowMinimizeButtonHint |
|
|
#~ Qt.WindowMaximizeButtonHint |
|
|
#~ Qt.WindowCloseButtonHint)
|
|
|
|
self.searchWidget.setVisible(False)
|
|
|
|
self.toolBar.setIconSize(iface.iconSize())
|
|
|
|
self.actionOpenScript.setIcon(
|
|
QgsApplication.getThemeIcon('/mActionFileOpen.svg'))
|
|
self.actionSaveScript.setIcon(
|
|
QgsApplication.getThemeIcon('/mActionFileSave.svg'))
|
|
self.actionSaveScriptAs.setIcon(
|
|
QgsApplication.getThemeIcon('/mActionFileSaveAs.svg'))
|
|
self.actionEditScriptHelp.setIcon(
|
|
QgsApplication.getThemeIcon('/mActionEditHelpContent.svg'))
|
|
self.actionRunScript.setIcon(
|
|
QgsApplication.getThemeIcon('/mActionStart.svg'))
|
|
self.actionCut.setIcon(
|
|
QgsApplication.getThemeIcon('/mActionEditCut.svg'))
|
|
self.actionCopy.setIcon(
|
|
QgsApplication.getThemeIcon('/mActionEditCopy.svg'))
|
|
self.actionPaste.setIcon(
|
|
QgsApplication.getThemeIcon('/mActionEditPaste.svg'))
|
|
self.actionUndo.setIcon(
|
|
QgsApplication.getThemeIcon('/mActionUndo.svg'))
|
|
self.actionRedo.setIcon(
|
|
QgsApplication.getThemeIcon('/mActionRedo.svg'))
|
|
self.actionFindReplace.setIcon(
|
|
QgsApplication.getThemeIcon('/mActionFindReplace.svg'))
|
|
self.actionIncreaseFontSize.setIcon(
|
|
QgsApplication.getThemeIcon('/mActionIncreaseFont.svg'))
|
|
self.actionDecreaseFontSize.setIcon(
|
|
QgsApplication.getThemeIcon('/mActionDecreaseFont.svg'))
|
|
|
|
# Connect signals and slots
|
|
self.actionOpenScript.triggered.connect(self.openScript)
|
|
self.actionSaveScript.triggered.connect(self.save)
|
|
self.actionSaveScriptAs.triggered.connect(self.saveAs)
|
|
#self.actionEditScriptHelp.triggered.connect(self.editHelp)
|
|
self.actionRunScript.triggered.connect(self.runAlgorithm)
|
|
self.actionCut.triggered.connect(self.editor.cut)
|
|
self.actionCopy.triggered.connect(self.editor.copy)
|
|
self.actionPaste.triggered.connect(self.editor.paste)
|
|
self.actionUndo.triggered.connect(self.editor.undo)
|
|
self.actionRedo.triggered.connect(self.editor.redo)
|
|
self.actionFindReplace.toggled.connect(self.toggleSearchBox)
|
|
self.actionIncreaseFontSize.triggered.connect(self.editor.zoomIn)
|
|
self.actionDecreaseFontSize.triggered.connect(self.editor.zoomOut)
|
|
self.editor.textChanged.connect(lambda: self.setHasChanged(True))
|
|
|
|
self.btnFind.clicked.connect(self.find)
|
|
self.btnReplace.clicked.connect(self.replace)
|
|
self.lastSearch = None
|
|
|
|
self.filePath = filePath
|
|
if self.filePath is not None:
|
|
self._loadFile(self.filePath)
|
|
|
|
self.needUpdate = False
|
|
self.setHasChanged(False)
|
|
|
|
#self.snippets = {}
|
|
#path = os.path.join(os.path.dirname(os.path.dirname(__file__)), "script", "snippets.py")
|
|
#with codecs.open(path, "r", encoding="utf-8") as f:
|
|
# lines = f.readlines()
|
|
#snippetlines = []
|
|
#name = None
|
|
#for line in lines:
|
|
# if line.startswith("##"):
|
|
# if snippetlines:
|
|
# self.snippets[name] = "".join(snippetlines)
|
|
# name = line[2:]
|
|
# snippetlines = []
|
|
# else:
|
|
# snippetlines.append(line)
|
|
#if snippetlines:
|
|
# self.snippets[name] = "".join(snippetlines)
|
|
#if self.snippets:
|
|
# self.btnSnippets.setVisible(False)
|
|
|
|
#if self.alg is not None:
|
|
# pass
|
|
# self.filename = self.alg.descriptionFile
|
|
# self.editor.setText(self.alg.script)
|
|
#else:
|
|
# self.filename = None
|
|
|
|
#def showSnippets(self, evt):
|
|
# popupmenu = QMenu()
|
|
# for name, snippet in list(self.snippets.items()):
|
|
# action = QAction(self.tr(name), self.btnSnippets)
|
|
# action.triggered[()].connect(lambda snippet=snippet: self.editor.insert(snippet))
|
|
# popupmenu.addAction(action)
|
|
# popupmenu.exec_(QCursor.pos())
|
|
|
|
def closeEvent(self, event):
|
|
if self.hasChanged:
|
|
ret = QMessageBox.question(self,
|
|
self.tr("Unsaved changes"),
|
|
self.tr("There are unsaved changes in the script. Continue?"),
|
|
QMessageBox.Yes | QMessageBox.No,
|
|
QMessageBox.No
|
|
)
|
|
if ret == QMessageBox.Yes:
|
|
self.updateProvider()
|
|
event.accept()
|
|
else:
|
|
event.ignore()
|
|
else:
|
|
self.updateProvider()
|
|
event.accept()
|
|
|
|
def updateProvider(self):
|
|
if self.needUpdate:
|
|
QgsApplication.processingRegistry().providerById("script").refreshAlgorithms()
|
|
|
|
def openScript(self):
|
|
if self.hasChanged:
|
|
ret = QMessageBox.warning(self,
|
|
self.tr("Unsaved changes"),
|
|
self.tr("There are unsaved changes in the script. Continue?"),
|
|
QMessageBox.Yes | QMessageBox.No, QMessageBox.No)
|
|
if ret == QMessageBox.No:
|
|
return
|
|
|
|
scriptDir = ScriptUtils.scriptsFolders()[0]
|
|
fileName, _ = QFileDialog.getOpenFileName(self,
|
|
self.tr("Open script"),
|
|
scriptDir,
|
|
self.tr("Script files (*.py *.PY)"))
|
|
|
|
if fileName == "":
|
|
return
|
|
|
|
with OverrideCursor(Qt.WaitCursor):
|
|
self._loadFile(fileName)
|
|
self.filePath = fileName
|
|
|
|
def save(self):
|
|
self.saveScript(False)
|
|
|
|
def saveAs(self):
|
|
self.saveScript(True)
|
|
|
|
def saveScript(self, saveAs):
|
|
if self.filePath is None or saveAs:
|
|
scriptDir = ScriptUtils.scriptsFolders()[0]
|
|
newPath, _ = QFileDialog.getSaveFileName(self,
|
|
self.tr("Save script"),
|
|
scriptDir,
|
|
self.tr("Script files (*.py *.PY)"))
|
|
|
|
if newPath:
|
|
if not newPath.lower().endswith(".py"):
|
|
newPath += ".py"
|
|
|
|
self.filePath = newPath
|
|
|
|
text = self.editor.text()
|
|
try:
|
|
with codecs.open(self.filePath, "w", encoding="utf-8") as f:
|
|
f.write(text)
|
|
except IOError as e:
|
|
QMessageBox.warning(self,
|
|
self.tr("I/O error"),
|
|
self.tr("Unable to save edits:\n{}").format(str(e))
|
|
)
|
|
return
|
|
self.needUpdate = True
|
|
self.setHasChanged(False)
|
|
#else:
|
|
# self.filePath = None
|
|
|
|
def setHasChanged(self, hasChanged):
|
|
self.hasChanged = hasChanged
|
|
self.actionSaveScript.setEnabled(hasChanged)
|
|
|
|
def runAlgorithm(self):
|
|
#~ if self.filePath is None or self.hasChanged:
|
|
#~ QMessageBox.warning(self,
|
|
#~ self.tr("Unsaved changes"),
|
|
#~ self.tr("There are unsaved changes in script. "
|
|
#~ "Please save it and try again.")
|
|
#~ )
|
|
#~ return
|
|
|
|
#~ algName = os.path.splitext(os.path.basename(self.filePath))[0]
|
|
#~ alg = ScriptUtils.loadAlgorithm(algName, self.filePath)
|
|
#~ alg.setProvider(QgsApplication.processingRegistry().providerById("script"))
|
|
#~ print("ALG", alg)
|
|
|
|
d = {}
|
|
#print(globals())
|
|
#print(locals())
|
|
exec(self.editor.text(), d)
|
|
#print(d)
|
|
#print(d.keys())
|
|
#print(d["SpatialIndex"])
|
|
alg = d["SpatialIndex"]()
|
|
alg.setProvider(QgsApplication.processingRegistry().providerById("script"))
|
|
|
|
dlg = alg.createCustomParametersWidget(self)
|
|
if not dlg:
|
|
dlg = AlgorithmDialog(alg)
|
|
|
|
canvas = iface.mapCanvas()
|
|
prevMapTool = canvas.mapTool()
|
|
|
|
dlg.show()
|
|
|
|
if canvas.mapTool() != prevMapTool:
|
|
try:
|
|
canvas.mapTool().reset()
|
|
except:
|
|
pass
|
|
canvas.setMapTool(prevMapTool)
|
|
|
|
def find(self):
|
|
textToFind = self.leFindText.text()
|
|
caseSensitive = self.chkCaseSensitive.isChecked()
|
|
wholeWord = self.chkWholeWord.isChecked()
|
|
if self.lastSearch is None or textToFind != self.lastSearch:
|
|
self.editor.findFirst(textToFind, False, caseSensitive, wholeWord, True)
|
|
else:
|
|
self.editor.findNext()
|
|
|
|
def replace(self):
|
|
textToReplace = self.leReplaceText.text()
|
|
self.editor.replaceSelectedText(textToReplace)
|
|
|
|
def toggleSearchBox(self, checked):
|
|
self.searchWidget.setVisible(checked)
|
|
|
|
def _loadFile(self, filePath):
|
|
with codecs.open(filePath, "r", encoding="utf-8") as f:
|
|
txt = f.read()
|
|
|
|
self.editor.setText(txt)
|
|
self.hasChanged = False
|
|
self.editor.setModified(False)
|
|
self.editor.recolor()
|