# -*- coding: utf-8 -*-
import sys
from PyQt4.QtCore import QAbstractTableModel, QModelIndex, Qt, QVariant
from PyQt4.QtGui import QMainWindow, QTableView, QApplication
class SimpleTableModel(QAbstractTableModel):
def __init__(self, table, parent=None):
QAbstractTableModel.__init__(self, parent)
self.table = table
def rowCount(self, parent=QModelIndex()):
return len(self.table)
def columnCount(self, parent=QModelIndex()):
return max(len(row) for row in self.table)
def flags(self, index):
return Qt.ItemIsEnabled | Qt.ItemIsSelectable
def data(self, index, role):
if role != Qt.DisplayRole or not index.isValid():
return QVariant()
try:
return QVariant(
self.table[index.row()][index.column()])
except IndexError:
return QVariant(0)
class AutoSizeTableView(QTableView):
def __init__(self, parent=None):
QTableView.__init__(self, parent)
self.auto_resize_columns = False
def resizeEvent(self, evt):
if self.auto_resize_columns:
size = self.viewport().size()
width = size.width() / self.model().columnCount()
for col in xrange(self.model().columnCount()):
self.setColumnWidth(col, width)
QTableView.resizeEvent(self, evt)
class MainWindow(QMainWindow):
def __init__(self, parent=None):
QMainWindow.__init__(self, parent)
self.setWindowTitle('Table test')
view = AutoSizeTableView(self)
view.auto_resize_columns = True
model = SimpleTableModel([['foo', 'bar'], ['spam', 'eggs']], view)
view.setModel(model)
self.setCentralWidget(view)
def main():
app = QApplication(sys.argv)
window = MainWindow()
window.show()
app.exec_()
if __name__ == '__main__':
main()