Пример #1
0
    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)