def layerTreeInsertionPoint(tree_view: QgsLayerTreeView) -> tuple:
"""
Direct copy of the code QgisApp::layerTreeInsertionPoint for QGIS < 3.10
"""
insert_group = tree_view.layerTreeModel().rootGroup()
current = tree_view.currentIndex()
index = 0
if current.isValid():
index = current.row()
current_node = tree_view.currentNode()
if current_node:
# if the insertion point is actually a group, insert new layers into the group
if QgsLayerTree.isGroup(current_node):
# if the group is embedded go to the first non-embedded group, at worst the top level item
insert_group = firstGroupWithoutCustomProperty(
current_node, "embedded")
return InsertionPoint(insert_group, 0)
# otherwise just set the insertion point in front of the current node
parent_node = current_node.parent()
if QgsLayerTree.isGroup(parent_node):
# if the group is embedded go to the first non-embedded group, at worst the top level item
insert_group = firstGroupWithoutCustomProperty(
parent_node, "embedded")
if parent_node != insert_group:
index = 0
return InsertionPoint(insert_group, index)
def testProxyModelCurrentIndex(self):
"""Test a crash spotted out while developing the proxy model"""
view = QgsLayerTreeView()
view.setModel(self.model)
if USE_MODEL_TESTER:
proxy_tester = QAbstractItemModelTester(view.model())
tree_tester = QAbstractItemModelTester(view.layerTreeModel())
view.setCurrentLayer(self.layer3)
self.layer3.setFlags(self.layer.Private)
def testNode2IndexMethods(self):
"""Test node2index and node2sourceIndex"""
view = QgsLayerTreeView()
view.setModel(self.model)
if USE_MODEL_TESTER:
proxy_tester = QAbstractItemModelTester(view.model())
tree_tester = QAbstractItemModelTester(view.layerTreeModel())
tree_model = view.layerTreeModel()
proxy_model = view.proxyModel()
proxy_index = proxy_model.index(1, 0)
node2 = view.index2node(proxy_index)
self.assertEqual(node2.name(), 'layer2')
proxy_layer2_index = view.node2index(node2)
self.assertEqual(proxy_layer2_index, view.node2index(node2))
source_index = tree_model.index(1, 0)
tree_layer2_index = view.node2sourceIndex(node2)
self.assertEqual(tree_layer2_index, view.node2sourceIndex(node2))
def testSetModel(self):
view = QgsLayerTreeView()
# should not work
string_list_model = QStringListModel()
view.setModel(string_list_model)
self.assertFalse(view.model())
# should work
view.setModel(self.model)
if USE_MODEL_TESTER:
proxy_tester = QAbstractItemModelTester(view.model())
self.assertEqual(view.layerTreeModel(), self.model)
def testProxyModel(self):
"""Test proxy model filtering and private layers"""
view = QgsLayerTreeView()
view.setModel(self.model)
if USE_MODEL_TESTER:
proxy_tester = QAbstractItemModelTester(view.model())
tree_model = view.layerTreeModel()
proxy_model = view.proxyModel()
self.assertEqual(tree_model.rowCount(), 3)
self.assertEqual(proxy_model.rowCount(), 3)
items = []
for r in range(tree_model.rowCount()):
items.append(tree_model.data(tree_model.index(r, 0)))
self.assertEqual(items, ['layer1', 'layer2', 'layer3'])
proxy_items = []
for r in range(proxy_model.rowCount()):
proxy_items.append(proxy_model.data(proxy_model.index(r, 0)))
self.assertEqual(proxy_items, ['layer1', 'layer2', 'layer3'])
self.layer3.setFlags(self.layer.Private)
self.assertEqual(tree_model.rowCount(), 3)
self.assertEqual(proxy_model.rowCount(), 2)
proxy_items = []
for r in range(proxy_model.rowCount()):
proxy_items.append(proxy_model.data(proxy_model.index(r, 0)))
self.assertEqual(proxy_items, ['layer1', 'layer2'])
view.setShowPrivateLayers(True)
self.assertEqual(proxy_model.rowCount(), 3)
proxy_items = []
for r in range(proxy_model.rowCount()):
proxy_items.append(proxy_model.data(proxy_model.index(r, 0)))
self.assertEqual(proxy_items, ['layer1', 'layer2', 'layer3'])
view.setShowPrivateLayers(False)
self.assertEqual(proxy_model.rowCount(), 2)
proxy_items = []
for r in range(proxy_model.rowCount()):
proxy_items.append(proxy_model.data(proxy_model.index(r, 0)))
self.assertEqual(proxy_items, ['layer1', 'layer2'])
# Test filters
proxy_model.setFilterText('layer2')
self.assertEqual(proxy_model.rowCount(), 1)
proxy_items = []
for r in range(proxy_model.rowCount()):
proxy_items.append(proxy_model.data(proxy_model.index(r, 0)))
self.assertEqual(proxy_items, ['layer2'])
