from PySide6 import QtWidgets from PySide6.QtCore import Qt from PySide6.QtWidgets import QLineEdit, QPushButton, QWidget, QVBoxLayout, QListWidget, QToolBar, QDialog, QVBoxLayout, QLabel, QHBoxLayout,QComboBox from PySide6.QtGui import QIcon, QAction import darkdetect import requests from core.scraping.uztracker_scraper import scrape_uztracker from core.scraping.rutracker_scraper import scrape_rutracker from core.scraping.utils import get_magnet_link from core.downloading.download import start_client from core.downloading.download import add_magnet def state_debug(setting): global debug if setting is True: debug = True else: debug = False global tracker tracker = "rutracker" # default tracker class MainWindow(QtWidgets.QMainWindow, QWidget): def __init__(self): global settings_action super().__init__() searchresults = [] self.setWindowTitle("Software Manager") self.setGeometry(100, 100, 800, 600) self.controls = QWidget() self.controlsLayout = QVBoxLayout() # Widgets self.searchbar = QLineEdit() self.searchbar.setPlaceholderText("Search for software...") self.searchbar.setClearButtonEnabled(True) self.searchbar.setMinimumHeight(30) self.searchbar.returnPressed.connect(self.return_pressed) # Triggers scraping functions on enter self.button = QtWidgets.QPushButton("Download") self.softwareList = QListWidget() container = QWidget() containerLayout = QVBoxLayout() containerLayout.addWidget(self.searchbar) containerLayout.addWidget(self.softwareList) self.softwareList.addItems(searchresults) containerLayout.addWidget(self.button) # download button triggers self.button.clicked.connect(lambda: self.download_selected()) self.button.clicked.connect(lambda: self.get_selected_item()) container.setLayout(containerLayout) self.setCentralWidget(container) containerLayout.setAlignment(Qt.AlignmentFlag.AlignBottom) toolbar = QToolBar("Main Toolbar") self.addToolBar(toolbar) toolbar.setLayoutDirection(Qt.RightToLeft) self.tracker_list = QComboBox() self.tracker_list.addItems(["rutracker", "uztracker"]) self.tracker_list.activated.connect(self.set_tracker) if darkdetect.isDark(): settings_action = QAction(QIcon("core/assets/settings_dark.png"), "Settings", self) else: settings_action = QAction(QIcon("core/assets/settings.png"), "Settings", self) settings_action.triggered.connect(self.settings_dialog) toolbar.addAction(settings_action) toolbar.addWidget(self.tracker_list) toolbar.setMovable(False) def set_tracker(self, _): global tracker tracker = self.tracker_list.currentText() def settings_dialog(self): if debug: print("Settings dialog opened") dialog = QDialog(self) dialog.setWindowTitle("Settings") dialog.setFixedSize(400, 300) dialog.setLayout(QVBoxLayout()) label = QLabel("This is a settings dialog.") QtWidgets.QPushButton("Close", clicked=dialog.close) # close button dialog.layout().addWidget(label) dialog.layout().addWidget(QtWidgets.QPushButton("Close", clicked=dialog.close)) layout = QHBoxLayout() save_btn = QPushButton("Save") cancel_btn = QPushButton("Cancel") save_btn.clicked.connect(self.save_settings) cancel_btn.clicked.connect(dialog.reject) layout.addWidget(save_btn) layout.addWidget(cancel_btn) self.setLayout(QVBoxLayout()) dialog.layout().addLayout(layout) dialog.exec() def save_settings(self): pass def download_selected(self): item = self.softwareList.currentItem() if item is not None: if debug: print(f"Downloading {self.softwareList.currentItem().text()}") self.get_item_index(self.softwareList.currentItem().text(), self.postnames, self.postlinks, debug) else: if debug: print("No item selected for download.") # add gui notification for no item selected def get_selected_item(self): item = self.softwareList.currentItem() if item is not None: return item.text() return "" def return_pressed(self): search_text = self.searchbar.text() if debug: print("User searched for:", search_text) if tracker == "uztracker": response = scrape_uztracker(search_text, debug) if tracker == "rutracker": response = scrape_rutracker(search_text) if response: self.postnames, self.postlinks = response self.softwareList.clear() if self.postnames: self.softwareList.addItems(self.postnames) elif debug: print(f"No Results found for \"{search_text}\"") def get_item_index(self, item, list, listlinks, debug): position = list.index(item) if 0 <= position < len(listlinks): # note for myself: py starts counting at 0; check if number is not negative, check if number is not more than list length. if tracker == "uztracker": selected = "https://uztracker.net/" + listlinks[position].lstrip("./") if tracker == "rutracker": selected = listlinks[position] if debug: print("Selected URL: ", selected) selected_magnet = get_magnet_link(selected, debug) start_client() add_magnet(selected_magnet)