Ejemplo n.º 1
0
    def __init__(self, *args, parent=None):
        super(MainWindow, self).__init__()

        self.dbPhotos = args[0]
        ''' :type DbPhotos'''
        self.dbTags = args[2]
        self.dbThumbs = args[3]

        loadUi('data/mainwindow.ui', self)

        leftToolbarButtons = [SideBarButton('Albums', 1, 'image'),
                SideBarButton('Tags', 2, 'image')]
        rightToolbarButtons = [SideBarButton('Properties', 1, 'image'),
                SideBarButton('Metadata', 2, 'image')]
        self.leftSideToolbar = GSideToolbar(leftToolbarButtons)
        self.leftSideToolbar.selectionChanged.connect(self.onLeftSideToolChange)
        self.rightSideToolbar = GSideToolbar(rightToolbarButtons)

        self.centralLayout.insertWidget(-1, self.rightSideToolbar)
        self.centralLayout.insertWidget(0, self.leftSideToolbar)

        self.thumbView = ThumbView()
        self.thumbView.needThumb.connect(self.onThumbViewNeedThumb, Qt.QueuedConnection)
        self.splitterThumbs.addWidget(self.thumbView)

        self.stackLeftActivateAlbums()
        self.albumsTreeData = self.dbPhotos.getAlbumTree()

        self.treeAlbums.headerItem().setText(0, 'Albums')
        self.fillTreeWidget(self.treeAlbums, self.albumsTreeData)

        self.treeAlbums.itemActivated.connect(self.onTreeAlbumsItemActivated)

        self.thumbReader = ThumbReader(self.dbThumbs)
        self.thumbReader.thumbReady.connect(self.thumbView.updateThumb)
Ejemplo n.º 2
0
class MainWindow(QMainWindow):
    def __init__(self, *args, parent=None):
        super(MainWindow, self).__init__()

        self.dbPhotos = args[0]
        ''' :type DbPhotos'''
        self.dbTags = args[2]
        self.dbThumbs = args[3]

        loadUi('data/mainwindow.ui', self)

        leftToolbarButtons = [SideBarButton('Albums', 1, 'image'),
                SideBarButton('Tags', 2, 'image')]
        rightToolbarButtons = [SideBarButton('Properties', 1, 'image'),
                SideBarButton('Metadata', 2, 'image')]
        self.leftSideToolbar = GSideToolbar(leftToolbarButtons)
        self.leftSideToolbar.selectionChanged.connect(self.onLeftSideToolChange)
        self.rightSideToolbar = GSideToolbar(rightToolbarButtons)

        self.centralLayout.insertWidget(-1, self.rightSideToolbar)
        self.centralLayout.insertWidget(0, self.leftSideToolbar)

        self.thumbView = ThumbView()
        self.thumbView.needThumb.connect(self.onThumbViewNeedThumb, Qt.QueuedConnection)
        self.splitterThumbs.addWidget(self.thumbView)

        self.stackLeftActivateAlbums()
        self.albumsTreeData = self.dbPhotos.getAlbumTree()

        self.treeAlbums.headerItem().setText(0, 'Albums')
        self.fillTreeWidget(self.treeAlbums, self.albumsTreeData)

        self.treeAlbums.itemActivated.connect(self.onTreeAlbumsItemActivated)

        self.thumbReader = ThumbReader(self.dbThumbs)
        self.thumbReader.thumbReady.connect(self.thumbView.updateThumb)

    @pyqtSlot(int, str)
    def onThumbViewNeedThumb(self, thumbNumber : int, path : str):
        self.thumbReader.add(thumbNumber, path)

    def onThumbReaderThumbReady(self, path : str, pic : QPixmap):
        print('MW: thumb ready: ', path)

    def onTreeAlbumsItemActivated(self, item : QTreeWidgetItem, column : int):
        def getTreePath(item : QTreeWidgetItem, root : str):
            path = []
            while item is not None:
                path.append(item.text(0))
                item = item.parent()

            path.reverse()
            for f in path:
                root = os.path.join(root, f)
            return root

        root = self.albumsTreeData.parent.name
        treePath = getTreePath(item, root)
        print('Album item activated', treePath)

        photos = self.dbPhotos.getPhotosByPath(treePath)
        self.thumbView.setItems(photos)

    def setAlbumsTree(self):
        pass

    def stackLeftActivateAlbums(self):
        self.stackLeft.setCurrentWidget(self.treeAlbums)

    def stackLeftActivateTags(self):
        self.stackLeft.setCurrentWidget(self.treeTags)

    def fillTreeWidget(self, widget : QTreeWidget, tree : Tree):
        def helper(t : Tree, parent):
            item = QTreeWidgetItem()
            item.setText(0, t.name)
            item.setIcon(0, QIcon('data/gfx/folder.png'))
            if type(parent) == QTreeWidget:
                parent.insertTopLevelItem(0, item)
            else:
                parent.addChild(item)
            for ch in t.children:
                helper(ch, item)

        widget.setColumnCount(1)
        helper(tree, widget)

    def onLeftSideToolChange(self, buttonId):
        if buttonId == 'Albums':
            print('Albums!!')
            items = [i for i in range(100)]
            self.thumbView.setItems(items)
        else:
            print('Button clicked: ', buttonId)

    @staticmethod
    def start(*args):
        app = QApplication(sys.argv)

        w = MainWindow(*args)
        w.show()

        sys.exit(app.exec_())