def createTreeItem(self, parent, text, uuid = ""):
"""
Creates tree items
parent: parent item
text: item text
uuid: complex uuid
"""
count = parent.childCount()
children = []
#making a list of item names
for i in range(count):
child = parent.child(i)
children.append(child.text(0))
#checking if the text is already in the tree widget
if text not in children:
#case not it should be created
item = QTreeWidgetItem(parent)
item.setExpanded(True)
item.setText(0,text)
#adding the complex uuid to the tree widget
if uuid != "":
item.setText(1, str(uuid))
else:
#case already exists the correspondind item should be returned
for i in range(count):
child = parent.child(i)
if child.text(0) == text:
item = child
return item
def addRecentAlgorithms(self, updating=True):
showRecent = ProcessingConfig.getSetting(
ProcessingConfig.SHOW_RECENT_ALGORITHMS)
if showRecent:
recent = QgsGui.instance().processingRecentAlgorithmLog(
).recentAlgorithmIds()
if len(recent) != 0:
found = False
if updating:
recentItem = self.algorithmTree.topLevelItem(0)
if recentItem.text(0) == self.tr('Recently used'):
treeWidget = recentItem.treeWidget()
treeWidget.takeTopLevelItem(
treeWidget.indexOfTopLevelItem(recentItem))
recentItem = QTreeWidgetItem()
recentItem.setText(0, self.tr('Recently used'))
for algorithm_id in recent:
alg = QgsApplication.processingRegistry(
).createAlgorithmById(algorithm_id)
if alg is not None:
algItem = TreeAlgorithmItem(alg)
recentItem.addChild(algItem)
found = True
if found:
self.algorithmTree.insertTopLevelItem(0, recentItem)
recentItem.setExpanded(True)
self.algorithmTree.setWordWrap(True)
def addRecentAlgorithms(self, updating):
showRecent = ProcessingConfig.getSetting(
ProcessingConfig.SHOW_RECENT_ALGORITHMS)
if showRecent:
recent = ProcessingLog.getRecentAlgorithms()
if len(recent) != 0:
found = False
if updating:
recentItem = self.algorithmTree.topLevelItem(0)
if recentItem.text(0) == self.tr('Recently used'):
treeWidget = recentItem.treeWidget()
treeWidget.takeTopLevelItem(
treeWidget.indexOfTopLevelItem(recentItem))
recentItem = QTreeWidgetItem()
recentItem.setText(0, self.tr('Recently used'))
for algname in recent:
alg = QgsApplication.processingRegistry().createAlgorithmById(algname)
if alg is not None:
algItem = TreeAlgorithmItem(alg)
recentItem.addChild(algItem)
found = True
if found:
self.algorithmTree.insertTopLevelItem(0, recentItem)
recentItem.setExpanded(True)
self.algorithmTree.setWordWrap(True)
def computeDiffs(self):
self.featuresTree.clear()
self.changes = self.localChanges(self.layer)
layerItem = QTreeWidgetItem()
layerItem.setText(0, self.layer.name())
layerItem.setIcon(0, layerIcon)
self.featuresTree.addTopLevelItem(layerItem)
addedItem = QTreeWidgetItem()
addedItem.setText(0, "Added")
addedItem.setIcon(0, addedIcon)
removedItem = QTreeWidgetItem()
removedItem.setText(0, "Removed")
removedItem.setIcon(0, removedIcon)
modifiedItem = QTreeWidgetItem()
modifiedItem.setText(0, "Modified")
modifiedItem.setIcon(0, modifiedIcon)
layerSubItems = {LOCAL_FEATURE_ADDED: addedItem,
LOCAL_FEATURE_REMOVED: removedItem,
LOCAL_FEATURE_MODIFIED: modifiedItem}
for c in list(self.changes.values()):
item = QTreeWidgetItem()
item.setText(0, c.fid)
item.setIcon(0, featureIcon)
layerSubItems[c.changetype].addChild(item)
for i in [LOCAL_FEATURE_ADDED, LOCAL_FEATURE_REMOVED, LOCAL_FEATURE_MODIFIED]:
layerItem.addChild(layerSubItems[i])
layerSubItems[i].setText(0, "%s [%i features]" % (layerSubItems[i].text(0), layerSubItems[i].childCount()))
self.attributesTable.clear()
self.attributesTable.verticalHeader().hide()
self.attributesTable.horizontalHeader().hide()
layerItem.setExpanded(True)
def buildParentItem(self, dcollection, title, rootresource):
'''
Builds tree widget items from a dictionary.
'''
rtItem = QTreeWidgetItem()
rtItem.setText(0, title)
rtItem.setIcon(0, QIcon(rootresource))
topLevelItems = []
for k, v in dcollection.items():
parentItem = QTreeWidgetItem()
if isinstance(v, dict):
parentItem.setText(0, k)
for kc, vc in v.items():
child = QTreeWidgetItem()
child.setText(0, self._combine(kc, vc))
parentItem.addChild(child)
else:
parentItem.setText(0, self._combine(k, v))
topLevelItems.append(parentItem)
rtItem.addChildren(topLevelItems)
rtItem.setExpanded(True)
return rtItem
def fillTree(self):
self.tree.clear()
entries = ProcessingLog.getLogEntries()
names = {}
icons = {}
groupItem = QTreeWidgetItem()
groupItem.setText(0, 'ALGORITHM')
groupItem.setIcon(0, self.groupIcon)
for entry in entries:
icon = self.keyIcon
name = ''
match = re.search('processing.run\\("(.*?)"', entry.text)
if match.group:
algorithm_id = match.group(1)
if algorithm_id not in names:
algorithm = QgsApplication.processingRegistry(
).algorithmById(algorithm_id)
if algorithm:
names[algorithm_id] = algorithm.displayName()
icons[
algorithm_id] = QgsApplication.processingRegistry(
).algorithmById(algorithm_id).icon()
else:
names[algorithm_id] = ''
icons[algorithm_id] = self.keyIcon
name = names[algorithm_id]
icon = icons[algorithm_id]
item = TreeLogEntryItem(entry, True, name)
item.setIcon(0, icon)
groupItem.insertChild(0, item)
self.tree.addTopLevelItem(groupItem)
groupItem.setExpanded(True)
def updateUiManyToOnePolymorphic(self):
layerFeature = dict()
for relation in self._polymorphicRelation.generateRelations():
layer = relation.referencingLayer()
request = relation.getRelatedFeaturesRequest(self.feature())
finalLayer = relation.referencedLayer()
for feature in layer.getFeatures(request):
if finalLayer in layerFeature:
layerFeature[finalLayer].append(feature)
else:
layerFeature[finalLayer] = [feature]
for layer in layerFeature:
treeWidgetItemLayer = QTreeWidgetItem(self.mFeaturesTreeWidget,
[layer.name()])
treeWidgetItemLayer.setData(0, TreeWidgetItemRole.Type,
TreeWidgetItemType.Layer)
treeWidgetItemLayer.setData(0, TreeWidgetItemRole.Layer, layer)
treeWidgetItemLayer.setIcon(0, QgsIconUtils.iconForLayer(layer))
for feature in layerFeature[layer]:
treeWidgetItem = QTreeWidgetItem(treeWidgetItemLayer, [
QgsVectorLayerUtils.getFeatureDisplayString(
layerFeature, feature)
])
treeWidgetItem.setData(0, TreeWidgetItemRole.Type,
TreeWidgetItemType.Feature)
treeWidgetItem.setData(0, TreeWidgetItemRole.Layer, layer)
treeWidgetItem.setData(0, TreeWidgetItemRole.Feature, feature)
treeWidgetItemLayer.setExpanded(True)
def addRecentAlgorithms(self, updating):
showRecent = ProcessingConfig.getSetting(
ProcessingConfig.SHOW_RECENT_ALGORITHMS)
if showRecent:
recent = ProcessingLog.getRecentAlgorithms()
if len(recent) != 0:
found = False
if updating:
recentItem = self.algorithmTree.topLevelItem(0)
treeWidget = recentItem.treeWidget()
treeWidget.takeTopLevelItem(
treeWidget.indexOfTopLevelItem(recentItem))
recentItem = QTreeWidgetItem()
recentItem.setText(0, self.tr('Recently used algorithms'))
for algname in recent:
alg = Processing.getAlgorithm(algname)
if alg is not None:
algItem = TreeAlgorithmItem(alg)
recentItem.addChild(algItem)
found = True
if found:
self.algorithmTree.insertTopLevelItem(0, recentItem)
recentItem.setExpanded(True)
self.algorithmTree.setWordWrap(True)
def fill_item(self, item, value):
"""
Helper method used by load_dict_into_widget
"""
item.setExpanded(True)
if type(value) is dict:
for key, val in sorted(value.items()):
child = QTreeWidgetItem()
child.setText(0, str(key))
item.addChild(child)
self.fill_item(child, val)
elif type(value) is list:
for val in value:
child = QTreeWidgetItem()
item.addChild(child)
if type(val) is dict:
child.setText(0, '[dict]')
self.fill_item(child, val)
elif type(val) is list:
child.setText(0, '[list]')
self.fill_item(child, val)
else:
child.setText(0, str(val))
child.setExpanded(True)
else:
child = QTreeWidgetItem()
child.setText(0, str(value))
item.addChild(child)
def populate_classified_values(
self, unassigned_values, assigned_values, default_classes):
"""Populate lstUniqueValues and treeClasses.from the parameters.
:param unassigned_values: List of values that haven't been assigned
to a class. It will be put in self.lstUniqueValues.
:type unassigned_values: list
:param assigned_values: Dictionary with class as the key and list of
value as the value of the dictionary. It will be put in
self.treeClasses.
:type assigned_values: dict
:param default_classes: Default classes from unit.
:type default_classes: list
"""
# Populate the unique values list
self.lstUniqueValues.clear()
self.lstUniqueValues.setSelectionMode(
QAbstractItemView.ExtendedSelection)
for value in unassigned_values:
value_as_string = value is not None and str(value) or 'NULL'
list_item = QListWidgetItem(self.lstUniqueValues)
list_item.setFlags(
Qt.ItemIsEnabled
| Qt.ItemIsSelectable
| Qt.ItemIsDragEnabled)
list_item.setData(Qt.UserRole, value)
list_item.setText(value_as_string)
self.lstUniqueValues.addItem(list_item)
# Populate assigned values tree
self.treeClasses.clear()
self.treeClasses.invisibleRootItem().setFlags(Qt.ItemIsEnabled)
for default_class in default_classes:
# Create branch for class
tree_branch = QTreeWidgetItem(self.treeClasses)
tree_branch.setFlags(Qt.ItemIsDropEnabled | Qt.ItemIsEnabled)
tree_branch.setExpanded(True)
tree_branch.setFont(0, bold_font)
if 'name' in default_class:
default_class_name = default_class['name']
else:
default_class_name = default_class['key']
tree_branch.setText(0, default_class_name)
tree_branch.setData(0, Qt.UserRole, default_class['key'])
if 'description' in default_class:
tree_branch.setToolTip(0, default_class['description'])
# Assign known values
for value in assigned_values[default_class['key']]:
string_value = value is not None and str(value) or 'NULL'
tree_leaf = QTreeWidgetItem(tree_branch)
tree_leaf.setFlags(
Qt.ItemIsEnabled
| Qt.ItemIsSelectable
| Qt.ItemIsDragEnabled)
tree_leaf.setData(0, Qt.UserRole, value)
tree_leaf.setText(0, string_value)
def collect_overlays_summary(root, layer_data):
sublayers = layer_data.get('layers')
if sublayers:
item = QTreeWidgetItem(root)
item.setText(0, layer_data['name'])
for sublayer_data in sublayers:
collect_overlays_summary(item, sublayer_data)
item.setExpanded(True)
else:
if 'visibility_scale_max' in layer_data:
scale_visibility = 'Maximum (inclusive): {0}, Minimum (exclusive): {1}'.format(
layer_data['visibility_scale_max'],
layer_data['visibility_scale_min'])
else:
scale_visibility = ''
if layer_data.get('hidden'):
create_formatted_tree(root,
{layer_data['name']: "Hidden: True"})
return
create_formatted_tree(
root, {
'{0}': [
"Visible: {1}", "Queryable: {2}", "Extent: {3}",
"CRS: {4}", "Geometry type: {5}",
"Scale based visibility: {6}", "Labels: {7}",
"Provider type: {8}",
"Attributes: {9}", "Attribution:",
["Title: {10}", "URL: {11}"], "Metadata:",
[
"Title: {12}", "Abstract: {13}",
"Keyword list: {14}"
]
]
}, [
layer_data['name'],
layer_data['visible'],
layer_data['queryable'],
layer_data['extent'],
layer_data['projection'],
layer_data.get('geom_type', ''),
scale_visibility,
layer_data.get('labels', False),
layer_data['provider_type'],
", ".join([
attribute.get('title', attribute['name'])
for attribute in layer_data.get('attributes', [])
]),
opt_value(layer_data, 'attribution.title'),
opt_value(layer_data, 'attribution.url'),
layer_data['metadata']['title'],
layer_data['metadata']['abstract'],
layer_data['metadata']['keyword_list'],
])
def fillAlgorithmTreeUsingProviders(self):
self.algorithmTree.clear()
text = str(self.searchBox.text())
search_strings = text.split(' ')
allAlgs = algList.algs
for provider_id in list(allAlgs.keys()):
name = 'ACTIVATE_' + provider_id.upper().replace(' ', '_')
if not ProcessingConfig.getSetting(name):
continue
groups = {}
algs = list(allAlgs[provider_id].values())
# Add algorithms
for alg in algs:
if not alg.showInModeler:
continue
if alg.commandLineName() == self.alg.commandLineName():
continue
item_text = [alg.name.lower()]
item_text.extend(alg.tags.split(','))
show = not search_strings or all(
any(part in t for t in item_text)
for part in search_strings)
if show:
if alg.group in groups:
groupItem = groups[alg.group]
else:
groupItem = QTreeWidgetItem()
name = alg.i18n_group or alg.group
groupItem.setText(0, name)
groupItem.setToolTip(0, name)
groups[alg.group] = groupItem
algItem = TreeAlgorithmItem(alg)
groupItem.addChild(algItem)
if len(groups) > 0:
providerItem = QTreeWidgetItem()
provider = QgsApplication.processingRegistry().providerById(
provider_id)
providerItem.setText(0, provider.name())
providerItem.setToolTip(0, provider.name())
providerItem.setIcon(0, provider.icon())
for groupItem in list(groups.values()):
providerItem.addChild(groupItem)
self.algorithmTree.addTopLevelItem(providerItem)
providerItem.setExpanded(text != '')
for groupItem in list(groups.values()):
if text != '':
groupItem.setExpanded(True)
self.algorithmTree.sortItems(0, Qt.AscendingOrder)
def fillTree(self):
self.tree.clear()
entries = ProcessingLog.getLogEntries()
groupItem = QTreeWidgetItem()
groupItem.setText(0, 'ALGORITHM')
groupItem.setIcon(0, self.groupIcon)
for entry in entries:
item = TreeLogEntryItem(entry, True)
item.setIcon(0, self.keyIcon)
groupItem.insertChild(0, item)
self.tree.addTopLevelItem(groupItem)
groupItem.setExpanded(True)
def fillInputsTree(self):
icon = QIcon(os.path.join(pluginPath, 'images', 'input.png'))
parametersItem = QTreeWidgetItem()
parametersItem.setText(0, self.tr('Parameters'))
for paramType in ModelerParameterDefinitionDialog.paramTypes:
paramItem = QTreeWidgetItem()
paramItem.setText(0, paramType)
paramItem.setIcon(0, icon)
paramItem.setFlags(Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsDragEnabled)
parametersItem.addChild(paramItem)
self.inputsTree.addTopLevelItem(parametersItem)
parametersItem.setExpanded(True)
def fillTree(self):
self.tree.clear()
entries = ProcessingLog.getLogEntries()
groupItem = QTreeWidgetItem()
groupItem.setText(0, 'ALGORITHM')
groupItem.setIcon(0, self.groupIcon)
for entry in entries:
item = TreeLogEntryItem(entry, True)
item.setIcon(0, self.keyIcon)
groupItem.insertChild(0, item)
self.tree.addTopLevelItem(groupItem)
groupItem.setExpanded(True)
def fillInputsTree(self):
icon = QIcon(os.path.join(pluginPath, 'images', 'input.svg'))
parametersItem = QTreeWidgetItem()
parametersItem.setText(0, self.tr('Parameters'))
for paramType in ModelerParameterDefinitionDialog.paramTypes:
paramItem = QTreeWidgetItem()
paramItem.setText(0, paramType)
paramItem.setIcon(0, icon)
paramItem.setFlags(Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsDragEnabled)
parametersItem.addChild(paramItem)
self.inputsTree.addTopLevelItem(parametersItem)
parametersItem.setExpanded(True)
def fillAlgorithmTreeUsingProviders(self):
self.algorithmTree.clear()
text = str(self.searchBox.text())
search_strings = text.split(' ')
allAlgs = algList.algs
for provider_id in list(allAlgs.keys()):
name = 'ACTIVATE_' + provider_id.upper().replace(' ', '_')
if not ProcessingConfig.getSetting(name):
continue
groups = {}
algs = list(allAlgs[provider_id].values())
# Add algorithms
for alg in algs:
if not alg.showInModeler:
continue
if alg.commandLineName() == self.alg.commandLineName():
continue
item_text = [alg.name.lower()]
item_text.extend(alg.tags.split(','))
show = not search_strings or all(
any(part in t for t in item_text)
for part in search_strings)
if show:
if alg.group in groups:
groupItem = groups[alg.group]
else:
groupItem = QTreeWidgetItem()
name = alg.i18n_group or alg.group
groupItem.setText(0, name)
groupItem.setToolTip(0, name)
groups[alg.group] = groupItem
algItem = TreeAlgorithmItem(alg)
groupItem.addChild(algItem)
if len(groups) > 0:
providerItem = QTreeWidgetItem()
provider = QgsApplication.processingRegistry().providerById(provider_id)
providerItem.setText(0, provider.name())
providerItem.setToolTip(0, provider.name())
providerItem.setIcon(0, provider.icon())
for groupItem in list(groups.values()):
providerItem.addChild(groupItem)
self.algorithmTree.addTopLevelItem(providerItem)
providerItem.setExpanded(text != '')
for groupItem in list(groups.values()):
if text != '':
groupItem.setExpanded(True)
self.algorithmTree.sortItems(0, Qt.AscendingOrder)
def fillAlgorithmTreeUsingProviders(self):
self.algorithmTree.clear()
text = str(self.searchBox.text())
search_strings = text.split(' ')
for provider in QgsApplication.processingRegistry().providers():
if not provider.isActive():
continue
groups = {}
# Add algorithms
for alg in provider.algorithms():
if alg.flags() & QgsProcessingAlgorithm.FlagHideFromModeler:
continue
if alg.id() == self.alg.id():
continue
item_text = [alg.displayName().lower()]
item_text.extend(alg.tags())
show = not search_strings or all(
any(part in t for t in item_text)
for part in search_strings)
if show:
if alg.group() in groups:
groupItem = groups[alg.group()]
else:
groupItem = QTreeWidgetItem()
name = alg.group()
groupItem.setText(0, name)
groupItem.setToolTip(0, name)
groups[alg.group()] = groupItem
algItem = TreeAlgorithmItem(alg)
groupItem.addChild(algItem)
if len(groups) > 0:
providerItem = QTreeWidgetItem()
providerItem.setText(0, provider.name())
providerItem.setToolTip(0, provider.name())
providerItem.setIcon(0, provider.icon())
for groupItem in list(groups.values()):
providerItem.addChild(groupItem)
self.algorithmTree.addTopLevelItem(providerItem)
providerItem.setExpanded(text != '')
for groupItem in list(groups.values()):
if text != '':
groupItem.setExpanded(True)
self.algorithmTree.sortItems(0, Qt.AscendingOrder)
def fillAlgorithmTreeUsingProviders(self):
self.algorithmTree.clear()
text = str(self.searchBox.text())
search_strings = text.split(' ')
for provider in QgsApplication.processingRegistry().providers():
if not provider.isActive():
continue
groups = {}
# Add algorithms
for alg in provider.algorithms():
if alg.flags() & QgsProcessingAlgorithm.FlagHideFromModeler:
continue
if alg.id() == self.alg.id():
continue
item_text = [alg.displayName().lower()]
item_text.extend(alg.tags())
show = not search_strings or all(
any(part in t for t in item_text)
for part in search_strings)
if show:
if alg.group() in groups:
groupItem = groups[alg.group()]
else:
groupItem = QTreeWidgetItem()
name = alg.group()
groupItem.setText(0, name)
groupItem.setToolTip(0, name)
groups[alg.group()] = groupItem
algItem = TreeAlgorithmItem(alg)
groupItem.addChild(algItem)
if len(groups) > 0:
providerItem = QTreeWidgetItem()
providerItem.setText(0, provider.name())
providerItem.setToolTip(0, provider.name())
providerItem.setIcon(0, provider.icon())
for groupItem in list(groups.values()):
providerItem.addChild(groupItem)
self.algorithmTree.addTopLevelItem(providerItem)
providerItem.setExpanded(text != '')
for groupItem in list(groups.values()):
if text != '':
groupItem.setExpanded(True)
self.algorithmTree.sortItems(0, Qt.AscendingOrder)
def fillInputsTree(self):
icon = QIcon(os.path.join(pluginPath, 'images', 'input.svg'))
parametersItem = QTreeWidgetItem()
parametersItem.setText(0, self.tr('Parameters'))
sortedParams = sorted(QgsApplication.instance().processingRegistry().parameterTypes(), key=lambda pt: pt.name())
for param in sortedParams:
if param.flags() & QgsProcessingParameterType.ExposeToModeler:
paramItem = QTreeWidgetItem()
paramItem.setText(0, param.name())
paramItem.setData(0, Qt.UserRole, param.id())
paramItem.setIcon(0, icon)
paramItem.setFlags(Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsDragEnabled)
paramItem.setToolTip(0, param.description())
parametersItem.addChild(paramItem)
self.inputsTree.addTopLevelItem(parametersItem)
parametersItem.setExpanded(True)
def fillInputsTree(self):
icon = QIcon(os.path.join(pluginPath, 'images', 'input.svg'))
parametersItem = QTreeWidgetItem()
parametersItem.setText(0, self.tr('Parameters'))
sortedParams = sorted(QgsApplication.instance().processingRegistry().parameterTypes(), key=lambda pt: pt.name())
for param in sortedParams:
if param.flags() & QgsProcessingParameterType.ExposeToModeler:
paramItem = QTreeWidgetItem()
paramItem.setText(0, param.name())
paramItem.setData(0, Qt.UserRole, param.id())
paramItem.setIcon(0, icon)
paramItem.setFlags(Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsDragEnabled)
paramItem.setToolTip(0, param.description())
parametersItem.addChild(paramItem)
self.inputsTree.addTopLevelItem(parametersItem)
parametersItem.setExpanded(True)
def fillAlgorithmTreeUsingProviders(self):
self.algorithmTree.clear()
text = unicode(self.searchBox.text())
allAlgs = ModelerUtils.allAlgs
for providerName in allAlgs.keys():
name = 'ACTIVATE_' + providerName.upper().replace(' ', '_')
if not ProcessingConfig.getSetting(name):
continue
groups = {}
provider = allAlgs[providerName]
algs = provider.values()
# Add algorithms
for alg in algs:
if not alg.showInModeler or alg.allowOnlyOpenedLayers:
continue
if alg.commandLineName() == self.alg.commandLineName():
continue
if text == '' or text.lower() in alg.name.lower():
if alg.group in groups:
groupItem = groups[alg.group]
else:
groupItem = QTreeWidgetItem()
name = alg.i18n_group or alg.group
groupItem.setText(0, name)
groupItem.setToolTip(0, name)
groups[alg.group] = groupItem
algItem = TreeAlgorithmItem(alg)
groupItem.addChild(algItem)
if len(groups) > 0:
providerItem = QTreeWidgetItem()
providerItem.setText(0,
ModelerUtils.providers[providerName].getDescription())
providerItem.setToolTip(0,
ModelerUtils.providers[providerName].getDescription())
providerItem.setIcon(0,
ModelerUtils.providers[providerName].getIcon())
for groupItem in groups.values():
providerItem.addChild(groupItem)
self.algorithmTree.addTopLevelItem(providerItem)
providerItem.setExpanded(text != '')
for groupItem in groups.values():
if text != '':
groupItem.setExpanded(True)
self.algorithmTree.sortItems(0, Qt.AscendingOrder)
def fillAlgorithmTreeUsingProviders(self):
self.algorithmTree.clear()
text = unicode(self.searchBox.text())
allAlgs = ModelerUtils.allAlgs
for providerName in allAlgs.keys():
name = 'ACTIVATE_' + providerName.upper().replace(' ', '_')
if not ProcessingConfig.getSetting(name):
continue
groups = {}
provider = allAlgs[providerName]
algs = provider.values()
# Add algorithms
for alg in algs:
if not alg.showInModeler or alg.allowOnlyOpenedLayers:
continue
if alg.commandLineName() == self.alg.commandLineName():
continue
if text == '' or text.lower() in alg.name.lower():
if alg.group in groups:
groupItem = groups[alg.group]
else:
groupItem = QTreeWidgetItem()
name = alg.i18n_group or alg.group
groupItem.setText(0, name)
groupItem.setToolTip(0, name)
groups[alg.group] = groupItem
algItem = TreeAlgorithmItem(alg)
groupItem.addChild(algItem)
if len(groups) > 0:
providerItem = QTreeWidgetItem()
providerItem.setText(
0, ModelerUtils.providers[providerName].getDescription())
providerItem.setToolTip(
0, ModelerUtils.providers[providerName].getDescription())
providerItem.setIcon(
0, ModelerUtils.providers[providerName].getIcon())
for groupItem in groups.values():
providerItem.addChild(groupItem)
self.algorithmTree.addTopLevelItem(providerItem)
providerItem.setExpanded(text != '')
for groupItem in groups.values():
if text != '':
groupItem.setExpanded(True)
self.algorithmTree.sortItems(0, Qt.AscendingOrder)
def computeDiffs(self):
self.commit1 = self.commit1Panel.getRef()
self.commit2 = self.commit2Panel.getRef()
self.featuresTree.clear()
changes = execute(lambda: self.repo.diff(self.commit1.commitid, self.commit2.commitid))
layerItems = {}
layerSubItems = {}
self.changes = {}
for c in changes:
self.changes[c.path] = c
layername = c.path.split("/")[0]
featureid = c.path.split("/")[-1]
if layername not in layerItems:
item = QTreeWidgetItem()
item.setText(0, layername)
item.setIcon(0, layerIcon)
layerItems[layername] = item
addedItem = QTreeWidgetItem()
addedItem.setText(0, "Added")
addedItem.setIcon(0, addedIcon)
removedItem = QTreeWidgetItem()
removedItem.setText(0, "Removed")
removedItem.setIcon(0, removedIcon)
modifiedItem = QTreeWidgetItem()
modifiedItem.setText(0, "Modified")
modifiedItem.setIcon(0, modifiedIcon)
layerSubItems[layername] = {FEATURE_ADDED: addedItem,
FEATURE_REMOVED: removedItem,
FEATURE_MODIFIED:modifiedItem}
item = FeatureItem(layername, featureid)
layerSubItems[layername][c.changetype].addChild(item)
for layername, item in layerItems.iteritems():
for i in [FEATURE_ADDED, FEATURE_REMOVED, FEATURE_MODIFIED]:
subItem = layerSubItems[layername][i]
item.addChild(subItem)
subItem.setText(0, "%s [%i features]" %
(subItem.text(0),
subItem.childCount()))
self.featuresTree.addTopLevelItem(item)
self.attributesTable.clear()
self.attributesTable.verticalHeader().hide()
self.attributesTable.horizontalHeader().hide()
for item in layerItems.values():
item.setExpanded(True)
def fillInputsTree(self):
from processing.core.Processing import Processing
icon = QIcon(os.path.join(pluginPath, 'images', 'input.svg'))
parametersItem = QTreeWidgetItem()
parametersItem.setText(0, self.tr('Parameters'))
sortedParams = sorted(Processing.registeredParameters().items())
for param in sortedParams:
if param[1]['exposeToModeller']:
paramItem = QTreeWidgetItem()
paramItem.setText(0, param[1]['name'])
paramItem.setData(0, Qt.UserRole, param[0])
paramItem.setIcon(0, icon)
paramItem.setFlags(Qt.ItemIsEnabled | Qt.ItemIsSelectable | Qt.ItemIsDragEnabled)
paramItem.setToolTip(0, param[1]['description'])
parametersItem.addChild(paramItem)
self.inputsTree.addTopLevelItem(parametersItem)
parametersItem.setExpanded(True)
def fillInputsTree(self):
from processing.core.Processing import Processing
icon = QIcon(os.path.join(pluginPath, 'images', 'input.svg'))
parametersItem = QTreeWidgetItem()
parametersItem.setText(0, self.tr('Parameters'))
sortedParams = sorted(Processing.registeredParameters().items())
for param in sortedParams:
if param[1]['exposeToModeller']:
paramItem = QTreeWidgetItem()
paramItem.setText(0, param[1]['name'])
paramItem.setData(0, Qt.UserRole, param[0])
paramItem.setIcon(0, icon)
paramItem.setFlags(Qt.ItemIsEnabled | Qt.ItemIsSelectable
| Qt.ItemIsDragEnabled)
paramItem.setToolTip(0, param[1]['description'])
parametersItem.addChild(paramItem)
self.inputsTree.addTopLevelItem(parametersItem)
parametersItem.setExpanded(True)
def updateUiManyToManyPolymorphic(self):
layer = self.relation().referencingLayer()
request = self.relation().getRelatedFeaturesRequest(self.feature())
layerFeature = dict()
for linkFeature in layer.getFeatures(request):
for relation in self._polymorphicRelation.generateRelations():
referencedFeatureRequest = relation.getReferencedFeatureRequest(
linkFeature)
filterExpression = referencedFeatureRequest.filterExpression()
nmRequest = QgsFeatureRequest()
nmRequest.setFilterExpression(filterExpression.expression())
finalLayer = relation.referencedLayer()
for finalFeature in finalLayer.getFeatures(nmRequest):
features = finalFeature, linkFeature
if finalLayer in layerFeature:
layerFeature[finalLayer].append(features)
else:
layerFeature[finalLayer] = [features]
for layer in layerFeature:
treeWidgetItemLayer = QTreeWidgetItem(self.mFeaturesTreeWidget,
[layer.name()])
treeWidgetItemLayer.setData(0, TreeWidgetItemRole.Type,
TreeWidgetItemType.Layer)
treeWidgetItemLayer.setData(0, TreeWidgetItemRole.Layer, layer)
treeWidgetItemLayer.setIcon(0, QgsIconUtils.iconForLayer(layer))
for feature, linkFeature in layerFeature[layer]:
treeWidgetItem = QTreeWidgetItem(treeWidgetItemLayer, [
QgsVectorLayerUtils.getFeatureDisplayString(
layer, feature)
])
treeWidgetItem.setData(0, TreeWidgetItemRole.Type,
TreeWidgetItemType.Feature)
treeWidgetItem.setData(0, TreeWidgetItemRole.Layer, layer)
treeWidgetItem.setData(0, TreeWidgetItemRole.Feature, feature)
treeWidgetItem.setData(0, TreeWidgetItemRole.LinkFeature,
linkFeature)
treeWidgetItemLayer.setExpanded(True)
def loadEarthCoverage(self):
"""
Loads a previously saved earth coverage configuration
"""
try:
self.clearTree()
if self.earthCoverageDict == dict():
self.getEarthCoverageDict()
rootItem = self.earthCoverageTreeWidget.invisibleRootItem()
#database item
for key in list(self.earthCoverageDict.keys()):
item = QTreeWidgetItem(rootItem)
item.setText(0, key)
item.setExpanded(True)
for cl in self.earthCoverageDict[key]:
covItem = QTreeWidgetItem(item)
covItem.setText(1, cl)
covItem.setExpanded(True)
except Exception as e:
QgsMessageLog.logMessage(
self.tr('Earth Coverage not loaded! Check log for details.') +
':'.join(e.args), "DSGTools Plugin", Qgis.Critical)
def _populate_reporting_tab(self):
"""Populate trees about layers."""
self.tree.clear()
self.add_layer.setEnabled(False)
self.remove_layer.setEnabled(False)
self.move_up.setEnabled(False)
self.move_down.setEnabled(False)
self.tree.setColumnCount(1)
self.tree.setRootIsDecorated(False)
self.tree.setHeaderHidden(True)
analysis_branch = QTreeWidgetItem(self.tree.invisibleRootItem(),
[FROM_ANALYSIS['name']])
analysis_branch.setFont(0, bold_font)
analysis_branch.setExpanded(True)
analysis_branch.setFlags(Qt.ItemIsEnabled)
if self._multi_exposure_if:
expected = self._multi_exposure_if.output_layers_expected()
for group, layers in list(expected.items()):
group_branch = QTreeWidgetItem(analysis_branch, [group])
group_branch.setFont(0, bold_font)
group_branch.setExpanded(True)
group_branch.setFlags(Qt.ItemIsEnabled)
for layer in layers:
layer = definition(layer)
if layer.get('allowed_geometries', None):
item = QTreeWidgetItem(group_branch,
[layer.get('name')])
item.setData(0, LAYER_ORIGIN_ROLE,
FROM_ANALYSIS['key'])
item.setData(0, LAYER_PARENT_ANALYSIS_ROLE, group)
item.setData(0, LAYER_PURPOSE_KEY_OR_ID_ROLE,
layer['key'])
item.setFlags(Qt.ItemIsEnabled | Qt.ItemIsSelectable)
canvas_branch = QTreeWidgetItem(self.tree.invisibleRootItem(),
[FROM_CANVAS['name']])
canvas_branch.setFont(0, bold_font)
canvas_branch.setExpanded(True)
canvas_branch.setFlags(Qt.ItemIsEnabled)
# List layers from the canvas
loaded_layers = list(QgsProject.instance().mapLayers().values())
canvas_layers = self.iface.mapCanvas().layers()
flag = setting('visibleLayersOnlyFlag', expected_type=bool)
for loaded_layer in loaded_layers:
if flag and loaded_layer not in canvas_layers:
continue
title = loaded_layer.name()
item = QTreeWidgetItem(canvas_branch, [title])
item.setData(0, LAYER_ORIGIN_ROLE, FROM_CANVAS['key'])
item.setData(0, LAYER_PURPOSE_KEY_OR_ID_ROLE, loaded_layer.id())
item.setFlags(Qt.ItemIsEnabled | Qt.ItemIsSelectable)
self.tree.resizeColumnToContents(0)
def populateTree(self):
for layer in self.layers:
item = QTreeWidgetItem()
item.setText(0, layer)
subitem = QTreeWidgetItem()
subitem.setText(0, "Publish symbology")
subitem.setIcon(0, SYMBOLOGY_ICON)
item.addChild(subitem)
subitem = QTreeWidgetItem()
subitem.setText(0, "Publish data")
subitem.setIcon(0, DATA_ICON)
item.addChild(subitem)
subitem = QTreeWidgetItem()
subitem.setText(0, "Publish metadata")
subitem.setIcon(0, METADATA_ICON)
item.addChild(subitem)
self.treeWidget.addTopLevelItem(item)
item.setExpanded(False)
item = QTreeWidgetItem()
item.setText(0, "Create layer groups")
item.setIcon(0, GROUPS_ICON)
self.treeWidget.addTopLevelItem(item)
QCoreApplication.processEvents()
def populateTree(self):
for lyr_name in (lyr_utils.getLayerById(id_).name() for id_ in self.layer_ids):
item = QTreeWidgetItem()
item.setText(0, lyr_name)
subitem = QTreeWidgetItem()
subitem.setText(0, "Publish symbology")
subitem.setIcon(0, SYMBOLOGY_ICON)
item.addChild(subitem)
subitem = QTreeWidgetItem()
subitem.setText(0, "Publish data")
subitem.setIcon(0, DATA_ICON)
item.addChild(subitem)
subitem = QTreeWidgetItem()
subitem.setText(0, "Publish metadata")
subitem.setIcon(0, METADATA_ICON)
item.addChild(subitem)
self.treeWidget.addTopLevelItem(item)
item.setExpanded(False)
item = QTreeWidgetItem()
item.setText(0, "Create layer groups")
item.setIcon(0, GROUPS_ICON)
self.treeWidget.addTopLevelItem(item)
QCoreApplication.processEvents()
def display(self):
'''
Initialize top-level items
'''
if len(self.items) == 0:
return
self.tree.clear()
# If there is only one item then set it as the root item
if len(self.items) == 1:
rootItem = self.items[0]
# Set root font
rtFont = rootItem.font(0)
rtFont.setBold(True)
rootItem.setFont(0, rtFont)
# Add the tree item to the tree widget
self.tree.addTopLevelItem(rootItem)
rootItem.setExpanded(True)
else:
rootItem = QTreeWidgetItem(self.tree)
rootItem.setText(0, self.title)
rootItem.setIcon(0, QIcon(self.rootResource))
# Set root font
rtFont = rootItem.font(0)
rtFont.setBold(True)
rootItem.setFont(0, rtFont)
rootItem.addChildren(self.items)
rootItem.setExpanded(True)
# Force the horizontal scrollbar to show
self.tree.header().setSectionResizeMode(QHeaderView.ResizeToContents)
def __init__(self):
super(TestSelector, self).__init__()
self.setupUi(self)
self.tests = None
self.bar = QgsMessageBar()
self.bar.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Fixed)
self.layout().insertWidget(1, self.bar)
allTests = defaultdict(list)
for test in tests.tests:
allTests[test.group].append(test)
for group, groupTests in allTests.items():
groupItem = QTreeWidgetItem()
groupItem.setText(0, group)
groupItem.setFlags(groupItem.flags() | Qt.ItemIsTristate)
unitItem = QTreeWidgetItem()
unitItem.setText(0, 'Fully Automated Tests')
unitItem.setFlags(unitItem.flags() | Qt.ItemIsTristate)
manualItem = QTreeWidgetItem()
manualItem.setText(0, 'Manual and Semi-Automated Tests')
manualItem.setFlags(manualItem.flags() | Qt.ItemIsTristate)
unitTestsByCategories = defaultdict(list)
manualTestsByCategories = defaultdict(list)
for test in groupTests:
if isinstance(test, UnitTestWrapper):
unitTestsByCategories[test.category].append(test)
else:
manualTestsByCategories[test.category].append(test)
for testsList, parentItem in [(unitTestsByCategories, unitItem),
(manualTestsByCategories, manualItem)
]:
for cat, catTests in testsList.items():
categoryItem = QTreeWidgetItem()
categoryItem.setText(0, cat)
categoryItem.setFlags(manualItem.flags()
| Qt.ItemIsTristate)
for test in catTests:
testItem = QTreeWidgetItem()
testItem.setFlags(testItem.flags()
| Qt.ItemIsUserCheckable)
testItem.setCheckState(0, Qt.Unchecked)
testItem.test = test
testItem.setText(0, test.name)
categoryItem.addChild(testItem)
parentItem.addChild(categoryItem)
if manualItem.childCount():
groupItem.addChild(manualItem)
if unitItem.childCount():
groupItem.addChild(unitItem)
self.testsTree.addTopLevelItem(groupItem)
groupItem.setExpanded(True)
self.testsTree.itemChanged.connect(self.toggleRunButton)
self.buttonBox.button(
QDialogButtonBox.Ok).setText('Run Selected Tests')
self.buttonBox.button(QDialogButtonBox.Ok).setEnabled(False)
self.buttonBox.accepted.connect(self.okPressed)
self.buttonBox.rejected.connect(self.cancelPressed)
self.selectAllLabel.linkActivated.connect(
lambda: self.checkTests(lambda t: Qt.Checked))
self.unselectAllLabel.linkActivated.connect(
lambda: self.checkTests(lambda t: Qt.Unchecked))
def _onlyManual(t):
if isinstance(t, UnitTestWrapper):
return Qt.Unchecked
else:
return Qt.Checked
self.onlyManualLabel.linkActivated.connect(
lambda: self.checkTests(_onlyManual))
def _onlyUnit(t):
if isinstance(t, UnitTestWrapper):
return Qt.Checked
else:
return Qt.Unchecked
self.onlyUnitLabel.linkActivated.connect(
lambda: self.checkTests(_onlyUnit))
filepath = os.path.expanduser('~/.testerplugin/failed.txt')
if os.path.exists(filepath):
with open(filepath) as f:
failed = json.load(f)
else:
failed = []
def _onlyLastFailures(t):
if t.group in failed and t.name in failed[t.group]:
return Qt.Checked
else:
return Qt.Unchecked
self.onlyLastFailuresLabel.linkActivated.connect(
lambda: self.checkTests(_onlyLastFailures))
self.exportButton.clicked.connect(self.export)
def __init__(self):
super(TestSelector, self).__init__()
self.setupUi(self)
self.tests = None
self.bar = QgsMessageBar()
self.bar.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Fixed)
self.layout().insertWidget(1, self.bar)
allTests = defaultdict(list)
for test in tests.tests:
allTests[test.group].append(test)
for group, groupTests in iteritems(allTests):
groupItem = QTreeWidgetItem()
groupItem.setText(0, group)
groupItem.setFlags(groupItem.flags() | Qt.ItemIsTristate)
unitItem = QTreeWidgetItem()
unitItem.setText(0, "Fully automated tests")
unitItem.setFlags(unitItem.flags() | Qt.ItemIsTristate)
manualItem = QTreeWidgetItem()
manualItem.setText(0, "Manual and semi-automated tests")
manualItem.setFlags(manualItem.flags() | Qt.ItemIsTristate)
for test in groupTests:
testItem = QTreeWidgetItem()
testItem.setFlags(testItem.flags() | Qt.ItemIsUserCheckable)
testItem.setCheckState(0, Qt.Unchecked)
testItem.test = test
testItem.setText(0, test.name)
if isinstance(test, UnitTestWrapper):
unitItem.addChild(testItem)
else:
manualItem.addChild(testItem)
if manualItem.childCount():
groupItem.addChild(manualItem)
if unitItem.childCount():
groupItem.addChild(unitItem)
self.testsTree.addTopLevelItem(groupItem)
groupItem.setExpanded(True)
self.buttonBox.button(QDialogButtonBox.Ok).setText("Run selected tests")
self.buttonBox.accepted.connect(self.okPressed)
self.buttonBox.rejected.connect(self.cancelPressed)
self.selectAllLabel.linkActivated.connect(lambda: self.checkTests(lambda t: Qt.Checked))
self.unselectAllLabel.linkActivated.connect(lambda: self.checkTests(lambda t: Qt.Unchecked))
def _onlyManual(t):
if isinstance(t, UnitTestWrapper):
return Qt.Unchecked
else:
return Qt.Checked
self.onlyManualLabel.linkActivated.connect(lambda: self.checkTests(_onlyManual))
def _onlyUnit(t):
if isinstance(t, UnitTestWrapper):
return Qt.Checked
else:
return Qt.Unchecked
self.onlyUnitLabel.linkActivated.connect(lambda: self.checkTests(_onlyUnit))
self.exportButton.clicked.connect(self.export)
class MapStoryExplorer(BASE, WIDGET):
def __init__(self):
super(MapStoryExplorer, self).__init__(None)
self.story = None
self.currentLayerItem = None
self.setupUi(self)
self.setAllowedAreas(Qt.RightDockWidgetArea | Qt.LeftDockWidgetArea)
self.layersTree.itemClicked.connect(self.treeItemClicked)
self.layerDescription.setOpenLinks(False)
self.layerDescription.anchorClicked.connect(self.layerDescriptionLinkClicked)
self.layerDescription.setFocusPolicy(Qt.NoFocus)
self.storyDescription.setOpenLinks(False)
self.storyDescription.anchorClicked.connect(self.storyDescriptionLinkClicked)
self.storyDescription.setFocusPolicy(Qt.NoFocus)
with open(resourceFile("layerdescription.css")) as f:
sheet = "".join(f.readlines())
self.layerDescription.document().setDefaultStyleSheet(sheet)
self.storyDescription.document().setDefaultStyleSheet(sheet)
if qtVersion < 5:
self.layersTree.header().setResizeMode(0, QHeaderView.Stretch)
self.layersTree.header().setResizeMode(1, QHeaderView.ResizeToContents)
self.updateCurrentStory(None)
def storyDescriptionLinkClicked(self, url):
url = url.toString()
if url == "search":
dlg = SearchDialog()
dlg.exec_()
if dlg.mapstory is not None:
story = Story.storyFromNumberId(dlg.mapstory)
if story is None:
QMessageBox.warning(iface.mainWindow(), "MapStory", "Cannot get MapStory data.\nCheck that the provided ID is correct.")
else:
self.updateCurrentStory(story)
elif url == "download":
outDir = QFileDialog.getExistingDirectory(self,
self.tr("Select output directory"),
"."
)
if not outDir:
return
QDir().mkpath(outDir)
settings = QSettings()
systemEncoding = settings.value('/UI/encoding', 'System')
startProgressBar(len(self.story.storyLayers()), "Download layers for off-line use:")
for i, layer in enumerate(self.story.storyLayers()):
filename = os.path.join(outDir, layer.name() + ".shp")
uri = "%s?srsname=%s&typename=geonode:%s&version=1.0.0&request=GetFeature&service=WFS" % (layer.wfsUrl(), layer.crs(), layer.name())
qgslayer = QgsVectorLayer(uri, layer.name(), "WFS")
writer = QgsVectorFileWriter(filename, systemEncoding,
qgslayer.pendingFields(),
qgslayer.dataProvider().geometryType(),
qgslayer.crs())
for feat in qgslayer.getFeatures():
writer.addFeature(feat)
del writer
fieldname = self._getTimeField(qgslayer)
if fieldname is not None:
filename = os.path.join(outDir, layer.name() + ".timefield")
with open(filename, "w") as f:
f.write(fieldname)
setProgress(i+1)
closeProgressBar()
iface.messageBar().pushMessage("MapStory", "Layers have been correctly saved as QGIS project.", level=QgsMessageBar.INFO, duration=3)
def _getTimeField(self, layer):
fields = layer.pendingFields()
for f in fields:
if f.typeName() == "xsd:dateTime":
return f.name()
def layerDescriptionLinkClicked(self, url):
url = url.toString()
service, url, name, crs = url.split("|")
if service == "wms":
uri = "url=%s&styles=&layers=%s&format=image/png&crs=%s" % (url,name, crs)
qgslayer = execute(lambda: QgsRasterLayer(uri, name, "wms"))
if not qgslayer.isValid():
raise Exception ("Layer at %s is not a valid layer" % uri)
QgsMapLayerRegistry.instance().addMapLayers([qgslayer])
timeValues = self.story.storyLayerFromName(name).wmsTimeValues()
addWmsAnimation(qgslayer, timeValues)
elif service == "wfs":
def f():
crs = iface.mapCanvas().mapRenderer().destinationCrs()
uri = "%s?srsname=%s&typename=geonode:%s&version=1.0.0&request=GetFeature&service=WFS" % (url, crs.authid(), name)
qgslayer = QgsVectorLayer(uri, name, "WFS")
if not qgslayer.isValid():
raise Exception ("Layer at %s is not a valid layer" % uri)
fieldname = self._getTimeField(qgslayer)
if fieldname is None:
QgsMapLayerRegistry.instance().addMapLayers([qgslayer])
else:
memlayer = QgsVectorLayer("%s?crs=%s" % (GEOM_TYPE_MAP[qgslayer.wkbType()], crs.authid()), name, "memory")
memlayer.startEditing()
for field in qgslayer.pendingFields():
memlayer.addAttribute(field)
for feat in qgslayer.getFeatures():
memlayer.addFeatures([feat])
memlayer.commitChanges()
QgsMapLayerRegistry.instance().addMapLayers([memlayer])
memlayer.setSelectedFeatures([])
addWfsAnimation(memlayer, fieldname)
execute(f)
def updateCurrentStory(self, story):
self.layersTree.clear()
self.currentLayerItem = None
self.story = story
if story is None:
self.storyDescription.setText("No MapStory selected. <a href='search'>[Click to open a MapStory]</a>")
return
self.storyDescription.setText(story.description())
self.layersItem = QTreeWidgetItem()
self.layersItem.setText(0, "Layers")
self.layersItem.setIcon(0, layersIcon)
for layer in story.storyLayers():
item = LayerItem(layer)
self.layersItem.addChild(item)
self.layersTree.addTopLevelItem(self.layersItem)
self.layersItem.setExpanded(True)
def treeItemClicked(self, item, i):
if self.currentLayerItem == item:
return
self.currentLayerItem = item
if isinstance(item, LayerItem):
self.updateCurrentLayer()
def updateCurrentLayer(self):
self.layerDescription.setText(self.currentLayerItem.layer.description())
def populate_classified_values(
unassigned_values, assigned_values, default_classes,
list_unique_values, tree_mapping_widget):
"""Populate lstUniqueValues and treeClasses.from the parameters.
:param unassigned_values: List of values that haven't been assigned
to a class. It will be put in list_unique_values.
:type unassigned_values: list
:param assigned_values: Dictionary with class as the key and list of
value as the value of the dictionary. It will be put in
tree_mapping_widget.
:type assigned_values: dict
:param default_classes: Default classes from unit.
:type default_classes: list
:param list_unique_values: List Widget for unique values
:type list_unique_values: QListWidget
:param tree_mapping_widget: Tree Widget for classifying.
:type tree_mapping_widget: QTreeWidget
"""
# Populate the unique values list
list_unique_values.clear()
list_unique_values.setSelectionMode(
QAbstractItemView.ExtendedSelection)
for value in unassigned_values:
value_as_string = value is not None and str(value) or 'NULL'
list_item = QListWidgetItem(list_unique_values)
list_item.setFlags(
Qt.ItemIsEnabled
| Qt.ItemIsSelectable
| Qt.ItemIsDragEnabled)
list_item.setData(Qt.UserRole, value)
list_item.setText(value_as_string)
list_unique_values.addItem(list_item)
# Populate assigned values tree
tree_mapping_widget.clear()
bold_font = QFont()
bold_font.setItalic(True)
bold_font.setBold(True)
bold_font.setWeight(75)
tree_mapping_widget.invisibleRootItem().setFlags(
Qt.ItemIsEnabled)
for default_class in default_classes:
# Create branch for class
tree_branch = QTreeWidgetItem(tree_mapping_widget)
tree_branch.setFlags(
Qt.ItemIsDropEnabled | Qt.ItemIsEnabled)
tree_branch.setExpanded(True)
tree_branch.setFont(0, bold_font)
if 'name' in default_class:
default_class_name = default_class['name']
else:
default_class_name = default_class['key']
tree_branch.setText(0, default_class_name)
tree_branch.setData(0, Qt.UserRole, default_class['key'])
if 'description' in default_class:
tree_branch.setToolTip(0, default_class['description'])
# Assign known values
for value in assigned_values[default_class['key']]:
string_value = value is not None and str(value) or 'NULL'
tree_leaf = QTreeWidgetItem(tree_branch)
tree_leaf.setFlags(
Qt.ItemIsEnabled
| Qt.ItemIsSelectable
| Qt.ItemIsDragEnabled)
tree_leaf.setData(0, Qt.UserRole, value)
tree_leaf.setText(0, string_value)
def __init__(self):
super(TestSelector, self).__init__()
self.setupUi(self)
self.tests = None
self.bar = QgsMessageBar()
self.bar.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Fixed)
self.layout().insertWidget(1, self.bar)
allTests = defaultdict(list)
for test in tests.tests:
allTests[test.group].append(test)
for group, groupTests in iteritems(allTests):
groupItem = QTreeWidgetItem()
groupItem.setText(0, group)
groupItem.setFlags(groupItem.flags() | Qt.ItemIsTristate);
unitItem = QTreeWidgetItem()
unitItem.setText(0, "Fully automated tests")
unitItem.setFlags(unitItem.flags() | Qt.ItemIsTristate);
manualItem = QTreeWidgetItem()
manualItem.setText(0, "Manual and semi-automated tests")
manualItem.setFlags(manualItem.flags() | Qt.ItemIsTristate);
unitTestsByCategories = defaultdict(list)
manualTestsByCategories = defaultdict(list)
for test in groupTests:
if isinstance(test, UnitTestWrapper):
unitTestsByCategories[test.category].append(test)
else:
manualTestsByCategories[test.category].append(test)
for testsList, parentItem in [(unitTestsByCategories, unitItem), (manualTestsByCategories, manualItem)]:
for cat, catTests in iteritems(testsList):
categoryItem = QTreeWidgetItem()
categoryItem.setText(0, cat)
categoryItem.setFlags(manualItem.flags() | Qt.ItemIsTristate);
for test in catTests:
testItem = QTreeWidgetItem()
testItem.setFlags(testItem.flags() | Qt.ItemIsUserCheckable);
testItem.setCheckState(0, Qt.Unchecked);
testItem.test = test
testItem.setText(0, test.name)
categoryItem.addChild(testItem)
parentItem.addChild(categoryItem)
if manualItem.childCount():
groupItem.addChild(manualItem)
if unitItem.childCount():
groupItem.addChild(unitItem)
self.testsTree.addTopLevelItem(groupItem)
groupItem.setExpanded(True)
self.buttonBox.button(QDialogButtonBox.Ok).setText("Run selected tests")
self.buttonBox.accepted.connect(self.okPressed)
self.buttonBox.rejected.connect(self.cancelPressed)
self.selectAllLabel.linkActivated.connect(lambda: self.checkTests(lambda t: Qt.Checked))
self.unselectAllLabel.linkActivated.connect(lambda: self.checkTests(lambda t: Qt.Unchecked))
def _onlyManual(t):
if isinstance(t, UnitTestWrapper):
return Qt.Unchecked
else:
return Qt.Checked
self.onlyManualLabel.linkActivated.connect(lambda: self.checkTests(_onlyManual))
def _onlyUnit(t):
if isinstance(t, UnitTestWrapper):
return Qt.Checked
else:
return Qt.Unchecked
self.onlyUnitLabel.linkActivated.connect(lambda: self.checkTests(_onlyUnit))
filepath = os.path.expanduser("~/.testerplugin/failed.txt")
if os.path.exists(filepath):
with open(filepath) as f:
failed = json.load(f)
else:
failed = []
def _onlyLastFailures(t):
if t.group in failed and t.name in failed[t.group]:
return Qt.Checked
else:
return Qt.Unchecked
self.onlyLastFailuresLabel.linkActivated.connect(lambda: self.checkTests(_onlyLastFailures))
self.exportButton.clicked.connect(self.export)
def __init__(self):
super(TestSelector, self).__init__()
self.setupUi(self)
self.tests = None
self.bar = QgsMessageBar()
self.bar.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.Fixed)
self.layout().insertWidget(1, self.bar)
allTests = defaultdict(list)
for test in tests.tests:
allTests[test.group].append(test)
for group, groupTests in iteritems(allTests):
groupItem = QTreeWidgetItem()
groupItem.setText(0, group)
groupItem.setFlags(groupItem.flags() | Qt.ItemIsTristate)
unitItem = QTreeWidgetItem()
unitItem.setText(0, "Fully automated tests")
unitItem.setFlags(unitItem.flags() | Qt.ItemIsTristate)
manualItem = QTreeWidgetItem()
manualItem.setText(0, "Manual and semi-automated tests")
manualItem.setFlags(manualItem.flags() | Qt.ItemIsTristate)
unitTestsByCategories = defaultdict(list)
manualTestsByCategories = defaultdict(list)
for test in groupTests:
if isinstance(test, UnitTestWrapper):
unitTestsByCategories[test.category].append(test)
else:
manualTestsByCategories[test.category].append(test)
for testsList, parentItem in [(unitTestsByCategories, unitItem),
(manualTestsByCategories, manualItem)
]:
for cat, catTests in iteritems(testsList):
categoryItem = QTreeWidgetItem()
categoryItem.setText(0, cat)
categoryItem.setFlags(manualItem.flags()
| Qt.ItemIsTristate)
for test in catTests:
testItem = QTreeWidgetItem()
testItem.setFlags(testItem.flags()
| Qt.ItemIsUserCheckable)
testItem.setCheckState(0, Qt.Unchecked)
testItem.test = test
testItem.setText(0, test.name)
categoryItem.addChild(testItem)
parentItem.addChild(categoryItem)
if manualItem.childCount():
groupItem.addChild(manualItem)
if unitItem.childCount():
groupItem.addChild(unitItem)
self.testsTree.addTopLevelItem(groupItem)
groupItem.setExpanded(True)
self.buttonBox.button(
QDialogButtonBox.Ok).setText("Run selected tests")
self.buttonBox.accepted.connect(self.okPressed)
self.buttonBox.rejected.connect(self.cancelPressed)
self.selectAllLabel.linkActivated.connect(
lambda: self.checkTests(lambda t: Qt.Checked))
self.unselectAllLabel.linkActivated.connect(
lambda: self.checkTests(lambda t: Qt.Unchecked))
def _onlyManual(t):
if isinstance(t, UnitTestWrapper):
return Qt.Unchecked
else:
return Qt.Checked
self.onlyManualLabel.linkActivated.connect(
lambda: self.checkTests(_onlyManual))
def _onlyUnit(t):
if isinstance(t, UnitTestWrapper):
return Qt.Checked
else:
return Qt.Unchecked
self.onlyUnitLabel.linkActivated.connect(
lambda: self.checkTests(_onlyUnit))
self.exportButton.clicked.connect(self.export)
