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
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()
class ScoreModel:
def __init__(self, fil: str = None, parent=None):
self.model = QSqlRelationalTableModel(parent)
self.model.setTable('Score')
self.model.setRelation(
9, QSqlRelation('TypeInfo', 'type_id', 'type_name'))
self.model.setHeaderData(9, Qt.Horizontal, 'type')
if fil is not None:
self.model.setFilter(fil)
self.model.select()
print(self.model)
class TmplModel:
def __init__(self, parent=None):
# dirpath = os.path.dirname(os.path.abspath(__file__))
# self.db = QSqlDatabase.addDatabase('QSQLITE')
# self.db.setDatabaseName(os.path.join(dirpath, 'grad.db'))
# self.db.open()
self.model = QSqlRelationalTableModel(parent)
self.model.setTable('Template')
self.model.setRelation(3, QSqlRelation('Teacher', 'tid', 'name'))
self.model.setHeaderData(3, Qt.Horizontal, 'teacher')
self.model.setRelation(
4, QSqlRelation('TypeInfo', 'type_id', 'type_name'))
self.model.setHeaderData(4, Qt.Horizontal, 'type')
self.model.select()
def refresh(self):
self.model.select()