diff --git a/core/interface/gui.py b/core/interface/gui.py index 5388447..ee9aec5 100644 --- a/core/interface/gui.py +++ b/core/interface/gui.py @@ -155,10 +155,10 @@ class MainWindow(QtWidgets.QMainWindow, QWidget): # Table Widget for Item List self.qtablewidget = QTableWidget() - self.qtablewidget.setColumnCount(2) + self.qtablewidget.setColumnCount(4) self.qtablewidget.setEditTriggers(QtWidgets.QAbstractItemView.EditTrigger.NoEditTriggers) self.qtablewidget.verticalHeader().setVisible(False) - self.qtablewidget.setHorizontalHeaderLabels(["Post Title", "Author"]) + self.qtablewidget.setHorizontalHeaderLabels(["Post Title", "Author", "Seeders", "Leechers"]) diff --git a/core/interface/utils/searchhelper.py b/core/interface/utils/searchhelper.py index 06d67e5..2b9a6cc 100644 --- a/core/interface/utils/searchhelper.py +++ b/core/interface/utils/searchhelper.py @@ -44,15 +44,19 @@ def return_pressed(self): self.qtablewidget.clear() self.show_empty_results(True) else: - state.post_titles, _, state.post_author = format_data(state.posts) + state.post_titles, _, state.post_author, state.post_seeders, state.post_leechers = format_data(state.posts) self.show_empty_results(False) self.qtablewidget.clear() - self.qtablewidget.setHorizontalHeaderLabels(["Post Title", "Author"]) + self.qtablewidget.setHorizontalHeaderLabels(["Post Title", "Author", "Seeders", "Leechers"]) self.qtablewidget.setRowCount(len(state.post_titles)) for i, author in enumerate(state.post_author): self.qtablewidget.setItem(i, 1, QTableWidgetItem(author)) for i, title in enumerate(state.post_titles): self.qtablewidget.setItem(i, 0, QTableWidgetItem(title)) + for i, seeders in enumerate(state.post_seeders): + self.qtablewidget.setItem(i, 2, QTableWidgetItem(seeders)) + for i, leechers in enumerate(state.post_leechers): + self.qtablewidget.setItem(i, 3, QTableWidgetItem(leechers)) if state.debug == True: consoleLog(f"Response Cached: {cached}") else: diff --git a/core/utils/data/state.py b/core/utils/data/state.py index b5052a1..bf99ba5 100644 --- a/core/utils/data/state.py +++ b/core/utils/data/state.py @@ -11,6 +11,8 @@ class AppState(QObject): self.post_titles: List[str] = [] self.post_urls: List[str] = [] self.post_author: List[str] = [] + self.post_seeders: List[str] = [] + self.post_leechers: List[str] = [] self.version: str = "dev" self._image_path: str = "" self.ignore_updates: bool = False diff --git a/core/utils/network/jsonhandler.py b/core/utils/network/jsonhandler.py index 7055515..882f783 100644 --- a/core/utils/network/jsonhandler.py +++ b/core/utils/network/jsonhandler.py @@ -15,8 +15,10 @@ def format_data(data): post_author = [post["author"] for post in data] post_titles = [post["title"] for post in data] post_links = [post["url"] for post in data] + post_seeders = [post["seeders"] for post in data] + post_leechers = [post["leechers"] for post in data] - return post_titles, post_links, post_author + return post_titles, post_links, post_author, post_seeders, post_leechers