Implemented Filtering and Image Context Menus in the GUI

akari/gui.py

@@ -1,20 +1,22 @@
 import sys
 from akari.utils import loadDB
 from PyQt5.QtCore import QSize
-from PyQt5.QtGui import QIcon, QPixmap
-from PyQt5.QtWidgets import QApplication, QWidget, QPushButton, QListWidget, QListWidgetItem, QListView, QHBoxLayout, QVBoxLayout
+from PyQt5.QtGui import QIcon, QColor, QFont, QPixmap
+from PyQt5.QtWidgets import QApplication, QWidget, QPushButton, QListWidget, QListWidgetItem, QListView, QHBoxLayout, QVBoxLayout, QAbstractItemView, QMenu
 
 class Main(QWidget):
     def __init__(self):
         super().__init__()
         self.setWindowTitle('akari')
         self.setWindowIcon(QIcon('static/icon.jpg'))
+        self.db = None
 
         self.mainLayout = QHBoxLayout()
 
         self.sidePanel = QVBoxLayout()
         self.tagList = QListWidget()
         self.filerButton = QPushButton("Filter")
+        self.clearFilerButton = QPushButton("Reset")
         self.sidePanel.addWidget(self.tagList)
 
         self.imagePanel = QVBoxLayout()
@@ -32,37 +34,95 @@ def __init__(self):
 
     def initUI(self):
         self.sidePanel.addWidget(self.filerButton, 1)
+        self.sidePanel.addWidget(self.clearFilerButton, 1)
+        self.tagList.setSelectionMode(QAbstractItemView.ExtendedSelection)
 
         self.imageList.setViewMode(QListView.IconMode) 
         self.imageList.setIconSize(QSize(300,300))
+        self.imageList.installEventFilter(self)
 
         self.imagePanel.addLayout(self.imagePanelOptions)
         self.imagePanelOptions.addWidget(self.miscButton1)
         self.imagePanelOptions.addWidget(self.miscButton2)
 
+        self.addEventHandlers()
         self.show()
 
-    def addImages(self, db):
-        for imgPath in db:
+    def updateDB(self, db):
+        self.db = db
+
+    def addEventHandlers(self):
+        self.filerButton.clicked.connect(self.filterButton_pressed)
+        self.clearFilerButton.clicked.connect(self.clearFilterButton_pressed)
+
+    def filterButton_pressed(self):
+        selected_items = self.tagList.selectedItems()
+        selected_tags = []
+        filtered_images = []
+        for i in range(len(selected_items)):
+            selected_tags.append(str(self.tagList.selectedItems()[i].text()).split(" ")[1])
+        for imgPath in self.db:
             if imgPath == 'akari-tags':
                 continue
+            if all(elem in self.db[imgPath] for elem in selected_tags):
+                if imgPath not in filtered_images:
+                    filtered_images.append(imgPath)
+        self.addImages(filtered_images)
+
+    def clearFilterButton_pressed(self):
+        imgList = self.getImgList()
+        self.addImages(imgList)
+        self.addTags()
+
+    def viewMetadata(self):
+        selected_image = self.imageList.selectedItems()[0].data(0)
+        tags = self.db[selected_image]
+        self.tagList.clear()
+        self.tagList.addItems(tags)
+        self.addImages([selected_image])
+
+    def contextMenuEvent(self, event):
+        menu = QMenu(self)
+        if self.imageList.selectedItems() != []:
+            viewMetadata = menu.addAction("View Metadata")
+            action = menu.exec_(self.mapToGlobal(event.pos()))
+            if action == viewMetadata:
+                self.viewMetadata()
+
+    def getImgList(self):
+        imgList = []
+        for imgPath in self.db:
+            if imgPath == 'akari-tags':
+                continue
+            imgList.append(imgPath)
+        return imgList
+
+    def addImages(self, imgList):
+        self.imageList.clear()
+        for imgPath in imgList:
             item = QListWidgetItem()
             icon = QIcon()
             pixmap = QPixmap(imgPath)
             icon.addPixmap(pixmap)
+            item.setFont(QFont("Times",1))
+            item.setForeground(QColor("white"))
+            item.setText(imgPath)
             item.setIcon(icon)
             self.imageList.addItem(item)
 
-    def addTags(self, db):
+    def addTags(self):
+        self.tagList.clear()
         tagList = []
-        for tag in sorted(db['akari-tags'], key=db['akari-tags'].get, reverse=True):
-            tagList.append(str(db['akari-tags'][tag]) + ' ' + tag)
+        for tag in sorted(self.db['akari-tags'], key=self.db['akari-tags'].get, reverse=True):
+            tagList.append(str(self.db['akari-tags'][tag]) + ' ' + tag)
         self.tagList.addItems(tagList)
 
 def init_gui():
     db = loadDB()
     app = QApplication(sys.argv)
     main = Main()
-    main.addTags(db)
-    main.addImages(db)
+    main.updateDB(db)
+    main.addTags()
+    imgList = main.getImgList()
+    main.addImages(imgList)
     sys.exit(app.exec_())