SoftwareManager/core/ui/gui.py

from PySide6 import QtWidgets
from PySide6.QtCore import Qt
from PySide6.QtWidgets import QLineEdit, QPushButton, QWidget, QVBoxLayout, QListWidget, QToolBar, QDialogButtonBox, QRadioButton, QGroupBox, QDialog, QVBoxLayout, QLabel, QHBoxLayout
from PySide6.QtGui import QIcon, QAction
import darkdetect
import requests
from core.scraping.uztracker_scraper import scrape_uztracker
from core.scraping.uztracker_scraper import get_item_index


def state_debug(setting):
    global debug
    if setting is True:
        debug = True
    else:
        debug = False

def choose_tracker():
    global rutracker
    choice = input("use rutracker? (y/n): ")
    print(choice)
    if choice.lower() == 'y':
        print("Using rutracker...")
        rutracker = True
    else:
        print("Using uztracker...")
        rutracker = False


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)
        
        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.setMovable(False)

    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()
<<<<<<< HEAD
        if item is not None:
            if debug:
                print(f"Downloading {self.softwareList.currentItem().text()}")
            get_item_index(self.softwareList.currentItem().text(), self.postnames, self.postlinks, debug)
=======
        if rutracker == True:
            if item is not None:
                if debug:
                    print(f"Downloading {self.softwareList.currentItem().text()}")
                self.get_item_index(self.softwareList.currentItem().text(), self.postnames_rutracker, self.postlinks_rutracker, debug)
            else:
                if debug:
                    print("No item selected for download.")
>>>>>>> af99e99 (added rutracker as option. check full message)
        else:
            if item is not None:
                if debug:
                    print(f"Downloading {self.softwareList.currentItem().text()}")
                self.get_item_index(self.softwareList.currentItem().text(), self.postnames_uztracker, self.postlinks_uztracker, 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):
<<<<<<< HEAD
        search_text = self.searchbar.text()
        if debug:
            print("User searched for:", search_text)
        
        response = scrape_uztracker(search_text, debug)
        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}\"")

        



=======
        if rutracker == True:
            global postnames
            global postlinks
            search_text = self.searchbar.text()
            search = requests.get(f"https://pizzasucht.net/search/{search_text}")
            self.softwareList.clear()
            data = search.json()
            self.postnames_rutracker = data["titles"]
            for self.postnames_rutracker in data["titles"]:
                self.softwareList.addItem(self.postnames_rutracker)
            self.postnames_rutracker = data["titles"]
            self.postlinks_rutracker = data["links"]
        else:
            search_text = self.searchbar.text()
            if debug:
                print("User searched for:", search_text)
            
            response = scrape_uztracker(search_text, debug)
            if response:
                self.postnames_uztracker, self.postlinks_uztracker = response
                self.softwareList.clear()
                if self.postnames_uztracker:
                    self.softwareList.addItems(self.postnames_uztracker)
            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 rutracker == True:
                selected = listlinks[position]
            else:
                selected = "https://uztracker.net/" + listlinks[position].lstrip("./")
            if debug:
                print("Selected URL: ", selected)
            selected_magnet = get_magnet_link(selected, debug)
            start_client()
            add_magnet(selected_magnet)
>>>>>>> af99e99 (added rutracker as option. check full message)