def populateMovieListWidget(self): allMovies = Movie.getAllMovies(self.currentPage * self.getPerPage(), self.getPerPage()) self.trwMovieList.clear() for movie in allMovies: item = QTreeWidgetItem(self.trwMovieList) item.setText(0, movie.title) item.setText(1, str(movie.length)) item.setText(2, str(movie.mediums)) item.setText(3, str(movie.rated)) item.setText(4, movie.getCategoriesAsCommaSeparatedString()) item.setText(5, movie.getActorsAsCommaSeparatedString()) item.setText(6, "%.2f €" % movie.cost) item.setData(0, roles.UserRoles.MOVIE_ID.value, movie.id) self.trwMovieList.addTopLevelItem(item) for i in range(0, self.trwMovieList.columnCount()): item.setTextAlignment( i, self.trwMovieList.headerItem().textAlignment(i)) item.setTextAlignment(item.columnCount() - 1, Qt.AlignRight) allMovies.clear()
def __init__(self, parent: QWidget, mode: roles.WindowSelectionRoles = -1): super(SelectionWindow, self).__init__(parent) self.setupUi(self) # set the icons for the buttonbox self.buttonBox.button(QDialogButtonBox.Ok).setIcon(QIcon(":/icons/add.png")) self.buttonBox.button(QDialogButtonBox.Ok).setIconSize(QSize(16, 16)) self.buttonBox.button(QDialogButtonBox.Cancel).setIcon(QIcon(":/icons/cancel.png")) self.buttonBox.button(QDialogButtonBox.Cancel).setIconSize(QSize(16, 16)) self.treeWidget.header().setSectionResizeMode(QHeaderView.Stretch) if mode == roles.WindowSelectionRoles.ACTOR.value: self.treeWidget.headerItem().setText(0, "Schauspieler") self.setWindowTitle("Schauspieler hinzufügen") for actor in Actor.getAllActors(): item = QTreeWidgetItem(self.treeWidget) item.setText(0, "{}, {}".format(actor.name, actor.prename)) item.setData(0, roles.UserRoles.ACTOR_ID.value, actor.id) self.treeWidget.addTopLevelItem(item) elif mode == roles.WindowSelectionRoles.MOVIE.value: self.treeWidget.headerItem().setText(0, "Filme") self.setWindowTitle("Film hinzufügen") for movie in Movie.getAllMovies(): item = QTreeWidgetItem(self.treeWidget) item.setText(0, movie.title) item.setData(0, roles.UserRoles.MOVIE_ID.value, movie.id) self.treeWidget.addTopLevelItem(item) elif mode == roles.WindowSelectionRoles.CATEGORY.value: self.treeWidget.headerItem().setText(0, "Kategorie") self.setWindowTitle("Kategorie hinzufügen") for category in Category.getAllCategories(): item = QTreeWidgetItem(self.treeWidget) item.setText(0, category.name) item.setData(0, roles.UserRoles.CATEGORY_ID.value, category.id) self.treeWidget.addTopLevelItem(item) else: QMessageBox.critical(self, "Internal Error", "Selection Window was created with unknown mode", QMessageBox.Ok)