1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
|
"""
***************************************************************************
ModelerAlgorithmProvider.py
---------------------
Date : August 2012
Copyright : (C) 2012 by Victor Olaya
Email : volayaf 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__ = "Victor Olaya"
__date__ = "August 2012"
__copyright__ = "(C) 2012, Victor Olaya"
import os
from qgis.core import (
Qgis,
QgsApplication,
QgsProcessingProvider,
QgsMessageLog,
QgsProcessingModelAlgorithm,
QgsRuntimeProfiler,
)
from processing.core.ProcessingConfig import ProcessingConfig, Setting
from processing.gui.ContextAction import ContextAction
from processing.gui.ProviderActions import ProviderActions, ProviderContextMenuActions
from processing.modeler.AddModelFromFileAction import AddModelFromFileAction
from processing.modeler.CreateNewModelAction import CreateNewModelAction
from processing.modeler.DeleteModelAction import DeleteModelAction
from processing.modeler.EditModelAction import EditModelAction
from processing.modeler.ExportModelAsPythonScriptAction import (
ExportModelAsPythonScriptAction,
)
from processing.modeler.OpenModelFromFileAction import OpenModelFromFileAction
from processing.modeler.ModelerUtils import ModelerUtils
pluginPath = os.path.split(os.path.dirname(__file__))[0]
class ModelerAlgorithmProvider(QgsProcessingProvider):
def __init__(self):
super().__init__()
self.actions = [
CreateNewModelAction(),
OpenModelFromFileAction(),
AddModelFromFileAction(),
]
sep_action = ContextAction()
sep_action.is_separator = True
self.contextMenuActions = [
EditModelAction(),
DeleteModelAction(),
sep_action,
ExportModelAsPythonScriptAction(),
]
self.algs = []
self.isLoading = False
# must reload models if providers list is changed - previously unavailable algorithms
# which models depend on may now be available
QgsApplication.processingRegistry().providerAdded.connect(self.onProviderAdded)
def onProviderAdded(self, provider_id):
if provider_id == self.id():
return
self.refreshAlgorithms()
def load(self):
with QgsRuntimeProfiler.profile("Model Provider"):
ProcessingConfig.settingIcons[self.name()] = self.icon()
ProcessingConfig.addSetting(
Setting(
self.name(),
ModelerUtils.MODELS_FOLDER,
self.tr("Models folder", "ModelerAlgorithmProvider"),
ModelerUtils.defaultModelsFolder(),
valuetype=Setting.MULTIPLE_FOLDERS,
)
)
ProviderActions.registerProviderActions(self, self.actions)
ProviderContextMenuActions.registerProviderContextMenuActions(
self.contextMenuActions
)
ProcessingConfig.readSettings()
self.refreshAlgorithms()
return True
def unload(self):
ProviderActions.deregisterProviderActions(self)
ProviderContextMenuActions.deregisterProviderContextMenuActions(
self.contextMenuActions
)
def modelsFolder(self):
return ModelerUtils.modelsFolders()[0]
def name(self):
return self.tr("Models", "ModelerAlgorithmProvider")
def id(self):
return "model"
def icon(self):
return QgsApplication.getThemeIcon("/processingModel.svg")
def svgIconPath(self):
return QgsApplication.iconPath("processingModel.svg")
def supportsNonFileBasedOutput(self):
return True
def loadAlgorithms(self):
with QgsRuntimeProfiler.profile("Load model algorithms"):
if self.isLoading:
return
self.isLoading = True
self.algs = []
folders = ModelerUtils.modelsFolders()
for f in folders:
self.loadFromFolder(f)
for a in self.algs:
self.addAlgorithm(a)
self.isLoading = False
def loadFromFolder(self, folder):
if not os.path.exists(folder):
return
for path, subdirs, files in os.walk(folder):
for descriptionFile in files:
if descriptionFile.endswith("model3"):
fullpath = os.path.join(path, descriptionFile)
alg = QgsProcessingModelAlgorithm()
if alg.fromFile(fullpath):
if alg.name():
alg.setSourceFilePath(fullpath)
self.algs.append(alg)
else:
QgsMessageLog.logMessage(
self.tr(
"Could not load model {0}", "ModelerAlgorithmProvider"
).format(descriptionFile),
self.tr("Processing"),
Qgis.MessageLevel.Critical,
)
|
