def __init__(self):
QMainWindow.__init__(self)
self.setupUi(self)
#Initialize db
createdb.init_db()
model = QSqlRelationalTableModel(self.bookTable)
model.setEditStrategy(QSqlTableModel.OnManualSubmit)
model.setTable("books")
# Remember the indexes of the columns:
author_idx = model.fieldIndex("author")
genre_idx = model.fieldIndex("genre")
# Set the relations to the other database tables:
model.setRelation(author_idx, QSqlRelation("authors", "id", "name"))
model.setRelation(genre_idx, QSqlRelation("genres", "id", "name"))
# Set the localized header captions:
model.setHeaderData(author_idx, Qt.Horizontal, self.tr("Author Name"))
model.setHeaderData(genre_idx, Qt.Horizontal, self.tr("Genre"))
model.setHeaderData(model.fieldIndex("title"), Qt.Horizontal, self.tr("Title"))
model.setHeaderData(model.fieldIndex("year"), Qt.Horizontal, self.tr("Year"))
model.setHeaderData(model.fieldIndex("rating"), Qt.Horizontal, self.tr("Rating"))
if not model.select():
print(model.lastError())
# Set the model and hide the ID column:
self.bookTable.setModel(model)
self.bookTable.setItemDelegate(BookDelegate(self.bookTable))
self.bookTable.setColumnHidden(model.fieldIndex("id"), True)
self.bookTable.setSelectionMode(QAbstractItemView.SingleSelection)
# Initialize the Author combo box:
self.authorEdit.setModel(model.relationModel(author_idx))
self.authorEdit.setModelColumn(model.relationModel(author_idx).fieldIndex("name"))
self.genreEdit.setModel(model.relationModel(genre_idx))
self.genreEdit.setModelColumn(model.relationModel(genre_idx).fieldIndex("name"))
# Lock and prohibit resizing of the width of the rating column:
self.bookTable.horizontalHeader().setSectionResizeMode(model.fieldIndex("rating"),
QHeaderView.ResizeToContents)
mapper = QDataWidgetMapper(self)
mapper.setModel(model)
mapper.setItemDelegate(BookDelegate(self))
mapper.addMapping(self.titleEdit, model.fieldIndex("title"))
mapper.addMapping(self.yearEdit, model.fieldIndex("year"))
mapper.addMapping(self.authorEdit, author_idx)
mapper.addMapping(self.genreEdit, genre_idx)
mapper.addMapping(self.ratingEdit, model.fieldIndex("rating"))
selection_model = self.bookTable.selectionModel()
selection_model.currentRowChanged.connect(mapper.setCurrentModelIndex)
self.bookTable.setCurrentIndex(model.index(0, 0))
self.create_menubar()
def UseSqlTable(db, table_name, columns, relations):
if relations:
model = QSqlRelationalTableModel(db=db)
else:
model = QSqlTableModel(db=db)
model.setTable(table_name)
model.setEditStrategy(QSqlTableModel.OnManualSubmit)
if relations:
model.setJoinMode(QSqlRelationalTableModel.LeftJoin
) # to work correctly with NULL values in fields
for relation in relations:
model.setRelation(
model.fieldIndex(relation[rel_idx.KEY_FIELD]),
QSqlRelation(relation[rel_idx.LOOKUP_TABLE],
relation[rel_idx.FOREIGN_KEY],
relation[rel_idx.LOOKUP_FIELD]))
for column in columns:
if column[hcol_idx.DISPLAY_NAME]:
model.setHeaderData(
model.fieldIndex(column[hcol_idx.DB_NAME]), Qt.Horizontal,
g_tr('TableViewConfig', column[hcol_idx.DISPLAY_NAME]))
if column[hcol_idx.SORT_ORDER] is not None:
model.setSort(model.fieldIndex(column[hcol_idx.DB_NAME]),
column[hcol_idx.SORT_ORDER])
return model