AddScriptFromFileAction.py 2.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  1. """
  2. ***************************************************************************
  3. EditScriptAction.py
  4. ---------------------
  5. Date : August 2012
  6. Copyright : (C) 2012 by Victor Olaya
  7. Email : volayaf at gmail dot com
  8. ***************************************************************************
  9. * *
  10. * This program is free software; you can redistribute it and/or modify *
  11. * it under the terms of the GNU General Public License as published by *
  12. * the Free Software Foundation; either version 2 of the License, or *
  13. * (at your option) any later version. *
  14. * *
  15. ***************************************************************************
  16. """
  17. __author__ = 'Victor Olaya'
  18. __date__ = 'April 2014'
  19. __copyright__ = '(C) 201, Victor Olaya'
  20. import os
  21. import shutil
  22. from qgis.PyQt.QtCore import QCoreApplication
  23. from qgis.PyQt.QtWidgets import QFileDialog
  24. from qgis.core import Qgis, QgsApplication, QgsMessageLog, QgsSettings
  25. from processing.gui.ToolboxAction import ToolboxAction
  26. from processing.script import ScriptUtils
  27. class AddScriptFromFileAction(ToolboxAction):
  28. def __init__(self):
  29. self.name = QCoreApplication.translate("AddScriptFromFileAction", "Add Script to Toolbox…")
  30. self.group = self.tr("Tools")
  31. def execute(self):
  32. settings = QgsSettings()
  33. lastDir = settings.value("processing/lastScriptsDir", "")
  34. files, _ = QFileDialog.getOpenFileNames(self.toolbox,
  35. self.tr("Add script(s)"),
  36. lastDir,
  37. self.tr("Processing scripts (*.py *.PY)"))
  38. if files:
  39. settings.setValue("processing/lastScriptsDir", os.path.dirname(files[0]))
  40. valid = 0
  41. for f in files:
  42. try:
  43. shutil.copy(f, ScriptUtils.scriptsFolders()[0])
  44. valid += 1
  45. except OSError as e:
  46. QgsMessageLog.logMessage(self.tr("Could not copy script '{}'\n{}").format(f, str(e)),
  47. "Processing",
  48. Qgis.Warning)
  49. if valid > 0:
  50. QgsApplication.processingRegistry().providerById("script").refreshAlgorithms()