class TestQgsAttributeTableModel(unittest.TestCase):
@classmethod
def setUpClass(cls):
QgsGui.editorWidgetRegistry().initEditors()
def setUp(self):
self.layer = self.createLayer()
self.cache = QgsVectorLayerCache(self.layer, 100)
self.am = QgsAttributeTableModel(self.cache)
self.am.loadLayer()
def tearDown(self):
del self.am
del self.cache
del self.layer
def createLayer(self):
layer = QgsVectorLayer("Point?field=fldtxt:string&field=fldint:integer",
"addfeat", "memory")
pr = layer.dataProvider()
features = list()
for i in range(10):
f = QgsFeature()
f.setAttributes(["test", i])
f.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(100 * i, 2 ^ i)))
features.append(f)
self.assertTrue(pr.addFeatures(features))
return layer
def testLoad(self):
self.assertEqual(self.am.rowCount(), 10)
self.assertEqual(self.am.columnCount(), 2)
def testRemove(self):
self.layer.startEditing()
self.layer.deleteFeature(5)
self.assertEqual(self.am.rowCount(), 9)
self.layer.selectByIds([1, 3, 6, 7])
self.layer.deleteSelectedFeatures()
self.assertEqual(self.am.rowCount(), 5)
def testAdd(self):
self.layer.startEditing()
f = QgsFeature()
f.setAttributes(["test", 8])
f.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(100, 200)))
self.layer.addFeature(f)
self.assertEqual(self.am.rowCount(), 11)
def testRemoveColumns(self):
self.assertTrue(self.layer.startEditing())
self.assertTrue(self.layer.deleteAttribute(1))
self.assertEqual(self.am.columnCount(), 1)
def testEdit(self):
fid = 2
field_idx = 1
new_value = 333
# get the same feature from model and layer
feature = self.layer.getFeature(fid)
model_index = self.am.idToIndex(fid)
feature_model = self.am.feature(model_index)
# check that feature from layer and model are sync
self.assertEqual(feature.attribute(field_idx), feature_model.attribute(field_idx))
# change attribute value for a feature and commit
self.layer.startEditing()
self.layer.changeAttributeValue(fid, field_idx, new_value)
self.layer.commitChanges()
# check the feature in layer is good
feature = self.layer.getFeature(fid)
self.assertEqual(feature.attribute(field_idx), new_value)
# get the same feature from model and layer
model_index = self.am.idToIndex(fid)
feature_model = self.am.feature(model_index)
# check that index from layer and model are sync
self.assertEqual(feature.attribute(field_idx), feature_model.attribute(field_idx))
class TestQgsAttributeTableModel(unittest.TestCase):
@classmethod
def setUpClass(cls):
QgsGui.editorWidgetRegistry().initEditors()
def setUp(self):
self.layer = self.createLayer()
self.cache = QgsVectorLayerCache(self.layer, 100)
self.am = QgsAttributeTableModel(self.cache)
self.am.loadLayer()
def tearDown(self):
del self.am
del self.cache
del self.layer
def createLayer(self):
layer = QgsVectorLayer("Point?field=fldtxt:string&field=fldint:integer",
"addfeat", "memory")
pr = layer.dataProvider()
features = list()
for i in range(10):
f = QgsFeature()
f.setAttributes(["test", i])
f.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(100 * i, 2 ^ i)))
features.append(f)
self.assertTrue(pr.addFeatures(features))
return layer
def testLoad(self):
self.assertEqual(self.am.rowCount(), 10)
self.assertEqual(self.am.columnCount(), 2)
def testRemove(self):
self.layer.startEditing()
self.layer.deleteFeature(5)
self.assertEqual(self.am.rowCount(), 9)
self.layer.selectByIds([1, 3, 6, 7])
self.layer.deleteSelectedFeatures()
self.assertEqual(self.am.rowCount(), 5)
def testAdd(self):
self.layer.startEditing()
f = QgsFeature()
f.setAttributes(["test", 8])
f.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(100, 200)))
self.layer.addFeature(f)
self.assertEqual(self.am.rowCount(), 11)
def testRemoveColumns(self):
self.assertTrue(self.layer.startEditing())
self.assertTrue(self.layer.deleteAttribute(1))
self.assertEqual(self.am.columnCount(), 1)
def testEdit(self):
fid = 2
field_idx = 1
new_value = 333
# get the same feature from model and layer
feature = self.layer.getFeature(fid)
model_index = self.am.idToIndex(fid)
feature_model = self.am.feature(model_index)
# check that feature from layer and model are sync
self.assertEqual(feature.attribute(field_idx), feature_model.attribute(field_idx))
# change attribute value for a feature and commit
self.layer.startEditing()
self.layer.changeAttributeValue(fid, field_idx, new_value)
self.layer.commitChanges()
# check the feature in layer is good
feature = self.layer.getFeature(fid)
self.assertEqual(feature.attribute(field_idx), new_value)
# get the same feature from model and layer
model_index = self.am.idToIndex(fid)
feature_model = self.am.feature(model_index)
# check that index from layer and model are sync
self.assertEqual(feature.attribute(field_idx), feature_model.attribute(field_idx))
def testStyle(self):
style_threshold = 2
color = QColor(133, 133, 133)
style = QgsConditionalStyle()
style.setRule(f'"fldint" <= {style_threshold}')
style.setTextColor(color)
self.layer.conditionalStyles().setRowStyles([style])
for f in self.layer.getFeatures():
model_index = self.am.idToIndex(f.id())
text_color = self.am.data(model_index, Qt.TextColorRole)
if f['fldint'] <= style_threshold:
self.assertEqual(text_color, color)
else:
self.assertIsNone(text_color)
self.assertTrue(self.layer.startEditing())
feature1 = self.layer.getFeature(2)
feature1['fldint'] = style_threshold + 1
feature2 = self.layer.getFeature(8)
feature2['fldint'] = style_threshold
self.assertTrue(self.layer.updateFeature(feature1))
self.assertTrue(self.layer.updateFeature(feature2))
self.assertTrue(self.layer.commitChanges())
for f in self.layer.getFeatures():
model_index = self.am.idToIndex(f.id())
text_color = self.am.data(model_index, Qt.TextColorRole)
if f['fldint'] <= style_threshold:
self.assertEqual(color, text_color, f'Feature {f.id()} should have color')
else:
self.assertIsNone(text_color, f'Feature {f.id()} should have no color')
self.layer.conditionalStyles().setRowStyles([])
def testTransactionRollback(self):
"""Test issue https://github.com/qgis/QGIS/issues/48171#issuecomment-1132709901"""
d = QTemporaryDir()
path = d.path()
source_fields = QgsFields()
source_fields.append(QgsField('int', QVariant.Int))
vl = QgsMemoryProviderUtils.createMemoryLayer('test', source_fields)
f = QgsFeature()
f.setAttributes([1])
vl.dataProvider().addFeature(f)
tmpfile = os.path.join(path, 'testTransactionRollback.sqlite')
options = {
'driverName': 'SpatiaLite',
'layerName': 'test'
}
err = QgsVectorLayerExporter.exportLayer(vl, tmpfile, "ogr", vl.crs(), False, options)
self.assertEqual(err[0], QgsVectorLayerExporter.NoError,
'unexpected import error {0}'.format(err))
vl = QgsVectorLayer(
'dbname=\'{}\' table="test" () sql='.format(tmpfile), 'test', 'spatialite')
self.assertTrue(vl.isValid())
p = QgsProject.instance()
p.setTransactionMode(Qgis.TransactionMode.AutomaticGroups)
self.assertTrue(p.addMapLayer(vl))
cache = QgsVectorLayerCache(vl, 100)
am = QgsAttributeTableModel(cache)
am.loadLayer()
self.assertEqual(am.rowCount(), 1)
self.assertTrue(vl.startEditing())
vl.beginEditCommand('edit1')
f = QgsFeature()
f.setAttributes([2])
self.assertTrue(vl.addFeature(f))
self.assertEqual(am.rowCount(), 2)
self.assertEqual(len([f for f in vl.getFeatures()]), 2)
vl.endEditCommand()
self.assertTrue(vl.rollBack())
self.assertEqual(len([f for f in vl.getFeatures()]), 1)
self.assertEqual(am.rowCount(), 1)
class TestQgsAttributeTableModel(unittest.TestCase):
@classmethod
def setUpClass(cls):
QgsGui.editorWidgetRegistry().initEditors()
def setUp(self):
self.layer = self.createLayer()
self.cache = QgsVectorLayerCache(self.layer, 100)
self.am = QgsAttributeTableModel(self.cache)
self.am.loadLayer()
def tearDown(self):
del self.am
del self.cache
del self.layer
def createLayer(self):
layer = QgsVectorLayer("Point?field=fldtxt:string&field=fldint:integer",
"addfeat", "memory")
pr = layer.dataProvider()
features = list()
for i in range(10):
f = QgsFeature()
f.setAttributes(["test", i])
f.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(100 * i, 2 ^ i)))
features.append(f)
self.assertTrue(pr.addFeatures(features))
return layer
def testLoad(self):
self.assertEqual(self.am.rowCount(), 10)
self.assertEqual(self.am.columnCount(), 2)
def testRemove(self):
self.layer.startEditing()
self.layer.deleteFeature(5)
self.assertEqual(self.am.rowCount(), 9)
self.layer.selectByIds([1, 3, 6, 7])
self.layer.deleteSelectedFeatures()
self.assertEqual(self.am.rowCount(), 5)
def testAdd(self):
self.layer.startEditing()
f = QgsFeature()
f.setAttributes(["test", 8])
f.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(100, 200)))
self.layer.addFeature(f)
self.assertEqual(self.am.rowCount(), 11)
def testRemoveColumns(self):
self.assertTrue(self.layer.startEditing())
self.assertTrue(self.layer.deleteAttribute(1))
self.assertEqual(self.am.columnCount(), 1)
def testEdit(self):
fid = 2
field_idx = 1
new_value = 333
# get the same feature from model and layer
feature = self.layer.getFeature(fid)
model_index = self.am.idToIndex(fid)
feature_model = self.am.feature(model_index)
# check that feature from layer and model are sync
self.assertEqual(feature.attribute(field_idx), feature_model.attribute(field_idx))
# change attribute value for a feature and commit
self.layer.startEditing()
self.layer.changeAttributeValue(fid, field_idx, new_value)
self.layer.commitChanges()
# check the feature in layer is good
feature = self.layer.getFeature(fid)
self.assertEqual(feature.attribute(field_idx), new_value)
# get the same feature from model and layer
model_index = self.am.idToIndex(fid)
feature_model = self.am.feature(model_index)
# check that index from layer and model are sync
self.assertEqual(feature.attribute(field_idx), feature_model.attribute(field_idx))
class TestQgsAttributeTableModel(unittest.TestCase):
@classmethod
def setUpClass(cls):
QgsGui.editorWidgetRegistry().initEditors()
def setUp(self):
self.layer = self.createLayer()
self.cache = QgsVectorLayerCache(self.layer, 100)
self.am = QgsAttributeTableModel(self.cache)
self.am.loadLayer()
def tearDown(self):
del self.am
del self.cache
del self.layer
def createLayer(self):
layer = QgsVectorLayer("Point?field=fldtxt:string&field=fldint:integer",
"addfeat", "memory")
pr = layer.dataProvider()
features = list()
for i in range(10):
f = QgsFeature()
f.setAttributes(["test", i])
f.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(100 * i, 2 ^ i)))
features.append(f)
self.assertTrue(pr.addFeatures(features))
return layer
def testLoad(self):
self.assertEqual(self.am.rowCount(), 10)
self.assertEqual(self.am.columnCount(), 2)
def testRemove(self):
self.layer.startEditing()
self.layer.deleteFeature(5)
self.assertEqual(self.am.rowCount(), 9)
self.layer.selectByIds([1, 3, 6, 7])
self.layer.deleteSelectedFeatures()
self.assertEqual(self.am.rowCount(), 5)
def testAdd(self):
self.layer.startEditing()
f = QgsFeature()
f.setAttributes(["test", 8])
f.setGeometry(QgsGeometry.fromPointXY(QgsPointXY(100, 200)))
self.layer.addFeature(f)
self.assertEqual(self.am.rowCount(), 11)
def testRemoveColumns(self):
self.assertTrue(self.layer.startEditing())
self.assertTrue(self.layer.deleteAttribute(1))
self.assertEqual(self.am.columnCount(), 1)
def testEdit(self):
fid = 2
field_idx = 1
new_value = 333
# get the same feature from model and layer
feature = self.layer.getFeature(fid)
model_index = self.am.idToIndex(fid)
feature_model = self.am.feature(model_index)
# check that feature from layer and model are sync
self.assertEqual(feature.attribute(field_idx), feature_model.attribute(field_idx))
# change attribute value for a feature and commit
self.layer.startEditing()
self.layer.changeAttributeValue(fid, field_idx, new_value)
self.layer.commitChanges()
# check the feature in layer is good
feature = self.layer.getFeature(fid)
self.assertEqual(feature.attribute(field_idx), new_value)
# get the same feature from model and layer
model_index = self.am.idToIndex(fid)
feature_model = self.am.feature(model_index)
# check that index from layer and model are sync
self.assertEqual(feature.attribute(field_idx), feature_model.attribute(field_idx))
def testStyle(self):
style_threshold = 2
color = QColor(133, 133, 133)
style = QgsConditionalStyle()
style.setRule(f'"fldint" <= {style_threshold}')
style.setTextColor(color)
self.layer.conditionalStyles().setRowStyles([style])
for f in self.layer.getFeatures():
model_index = self.am.idToIndex(f.id())
text_color = self.am.data(model_index, Qt.TextColorRole)
if f['fldint'] <= style_threshold:
self.assertEqual(text_color, color)
else:
self.assertIsNone(text_color)
self.assertTrue(self.layer.startEditing())
feature1 = self.layer.getFeature(2)
feature1['fldint'] = style_threshold + 1
feature2 = self.layer.getFeature(8)
feature2['fldint'] = style_threshold
self.assertTrue(self.layer.updateFeature(feature1))
self.assertTrue(self.layer.updateFeature(feature2))
self.assertTrue(self.layer.commitChanges())
for f in self.layer.getFeatures():
model_index = self.am.idToIndex(f.id())
text_color = self.am.data(model_index, Qt.TextColorRole)
if f['fldint'] <= style_threshold:
self.assertEqual(color, text_color, f'Feature {f.id()} should have color')
else:
self.assertIsNone(text_color, f'Feature {f.id()} should have no color')
self.layer.conditionalStyles().setRowStyles([])
