def testColumnHidden(self):
""" test hiding columns """
config = QgsAttributeTableConfig()
c1 = QgsAttributeTableConfig.ColumnConfig()
c1.name = 'test'
c1.hidden = False
c2 = QgsAttributeTableConfig.ColumnConfig()
c2.name = 'test2'
c2.hidden = False
config.setColumns([c1, c2])
self.assertFalse(config.columnHidden(0))
self.assertFalse(config.columnHidden(1))
config.setColumnHidden(1, True)
self.assertFalse(config.columnHidden(0))
self.assertTrue(config.columnHidden(1))
self.assertFalse(config.columns()[0].hidden)
self.assertTrue(config.columns()[1].hidden)
config.setColumnHidden(0, True)
self.assertTrue(config.columnHidden(0))
self.assertTrue(config.columnHidden(1))
self.assertTrue(config.columns()[0].hidden)
self.assertTrue(config.columns()[1].hidden)
c2.hidden = True
config.setColumns([c1, c2])
self.assertFalse(config.columnHidden(0))
self.assertTrue(config.columnHidden(1))
def testIsEmpty(self):
""" test isEmpty method """
config = QgsAttributeTableConfig()
self.assertTrue(config.isEmpty())
c = QgsAttributeTableConfig.ColumnConfig()
c.name = 'test'
config.setColumns([c])
self.assertFalse(config.isEmpty())
def testSize(self):
"""
Test QgsAttributeTableConfig.size and __len__
"""
config = QgsAttributeTableConfig()
self.assertEqual(config.size(), 0)
self.assertEqual(len(config), 0)
c1 = QgsAttributeTableConfig.ColumnConfig()
c2 = QgsAttributeTableConfig.ColumnConfig()
config.setColumns([c1, c2])
self.assertEqual(config.size(), 2)
self.assertEqual(len(config), 2)
def testColumnWidth(self):
""" test setting column widths """
config = QgsAttributeTableConfig()
c1 = QgsAttributeTableConfig.ColumnConfig()
c1.name = 'test'
c1.width = -1
c2 = QgsAttributeTableConfig.ColumnConfig()
c2.name = 'test2'
c2.width = 27
config.setColumns([c1, c2])
self.assertEqual(config.columnWidth(0), -1)
self.assertEqual(config.columnWidth(1), 27)
with self.assertRaises(IndexError):
config.columnWidth(-1)
with self.assertRaises(IndexError):
config.columnWidth(2)
config.setColumnWidth(1, -1)
self.assertEqual(config.columnWidth(0), -1)
self.assertEqual(config.columnWidth(1), -1)
self.assertEqual(config.columns()[0].width, -1)
self.assertEqual(config.columns()[1].width, -1)
config.setColumnWidth(0, 34)
self.assertEqual(config.columnWidth(0), 34)
self.assertEqual(config.columnWidth(1), -1)
self.assertEqual(config.columns()[0].width, 34)
self.assertEqual(config.columns()[1].width, -1)
with self.assertRaises(IndexError):
config.setColumnWidth(-1, 11)
with self.assertRaises(IndexError):
config.setColumnWidth(2, 11)
c2.width = 12
config.setColumns([c1, c2])
self.assertEqual(config.columnWidth(0), -1)
self.assertEqual(config.columnWidth(1), 12)
def testSetColumns(self):
""" test setting columns """
config = QgsAttributeTableConfig()
self.assertEqual(config.columns(), [])
c1 = QgsAttributeTableConfig.ColumnConfig()
c1.name = 'test'
c1.hidden = False
c1.width = 9
c2 = QgsAttributeTableConfig.ColumnConfig()
c2.name = 'test2'
c2.hidden = True
c2.width = 11
config.setColumns([c1, c2])
result = config.columns()
self.assertEqual(result[0].name, 'test')
self.assertEqual(result[1].name, 'test2')
self.assertEqual(result[0].hidden, False)
self.assertEqual(result[1].hidden, True)
self.assertEqual(result[0].width, 9)
self.assertEqual(result[1].width, 11)
def layer_has_custom_attribute_table(layer) -> bool:
# Do not use the isEmpty() on the QgsVectorLayer. It's automatically populated if empty with the fields.
config = QgsAttributeTableConfig()
config.update(layer.fields())
return not config.hasSameColumns(layer.attributeTableConfig())
