main.py

"""
====================================================================================================
    MAIN UI CREATED BY   : WANDERSON M.PIMENTA  (https://github.com/Wanderson-Magalhaes)
    EDITED/ENHANCED BY   : OFFICIALAHMED        (https://github.com/OfficialAhmed)
====================================================================================================
"""

# BUILT-INS
from json import load as jsonLoad
from json import dump as jsonDump
from os import environ, path, makedirs
from sys import argv, exit
from platform import system

# QT / PYSIDE FRAMEWORK
from PySide6.QtWidgets import *
from PySide6.QtCore import *
from PySide6.QtGui import *

# CUSTOME WIDGETS STYLING
from frontend.modules.home import *
from frontend.modules.settings import *


# FIX Problem for High DPI and Scale above 100%
environ["QT_FONT_DPI"] = "96"


class MainWindow(QMainWindow):

    widgets = None

    def __init__(self):
        QMainWindow.__init__(self)

        # SET AS GLOBAL WIDGETS
        self.ui = Ui()
        self.ui.setupUi(self)

        # global widgets
        self.widgets = self.ui
        shared_views.set_widgets(self.widgets, self.ui)

        # USE CUSTOM TITLE BAR | USE AS "False" FOR MAC OR LINUX
        self.ui.EENABLE_CUSTOM_TITLE_BAR = True if system() == "Windows" else False

        self.setWindowTitle(
            "File Engine"
        )

        self.widgets.titleRightInfo.setText(
            "File Management & Automation Tool"
        )
    
        # SET UI DEFINITIONS
        UiSettings.uiDefinitions(self)

        # LEFT MENUS
        self.widgets.home_page_btn.clicked.connect(
            lambda: shared_views.change(
                self.widgets.home_page_btn, "home_page", self.widgets.home_widgets)
        )
        self.widgets.delete_page_btn.clicked.connect(
            lambda: shared_views.change(
                self.widgets.delete_page_btn, "delete_page", self.widgets.delete_widgets)
        )
        self.widgets.rename_page_btn.clicked.connect(
            lambda: shared_views.change(
                self.widgets.rename_page_btn, "rename_page", self.widgets.rename_widgets)
        )
        self.widgets.search_page_btn.clicked.connect(
            lambda: shared_views.change(
                self.widgets.search_page_btn, "search_page", self.widgets.search_widgets)
        )
        self.widgets.move_page_btn.clicked.connect(
            lambda: shared_views.change(
                self.widgets.move_page_btn, "move_page", self.widgets.move_widgets)
        )

        # TOGGLE MENU
        self.widgets.toggleButton.clicked.connect(
            lambda: UiSettings.toggleMenu(self, True)
        )

        # EXTRA LEFT MENU
        self.widgets.toggleLeftBox.clicked.connect(
            lambda: UiSettings.toggleLeftBox(self, True)
        )

        self.widgets.extraCloseColumnBtn.clicked.connect(
            lambda: UiSettings.toggleLeftBox(self, True)
        )

        # EXTRA RIGHT MENU
        self.widgets.moreBtn.clicked.connect(
            lambda: UiSettings.toggleRightBox(self, True)
        )

        self.show()

        default_settings = self.get_settings()

        # APPLY THEME - DARK IS THE DEFAULT
        if default_settings.get("is_light_theme"):
            UiSettings.theme(self, True)

        # SET HOME PAGE AND SELECT MENU
        self.widgets.stackedWidget.setCurrentWidget(
            self.widgets.home_widgets
        )
        self.widgets.home_page_btn.setStyleSheet(
            UiSettings.selectMenu(
                self.widgets.home_page_btn.styleSheet()
            )
        )

        # SIGNALS TO CHANGE PAGE FROM SEARCH PAGE
        self.widgets.search_widgets.findChild(QPushButton, "deleteOptionBtn").clicked.connect(
            lambda: shared_views.change_indirect("delete_page")
        )
        self.widgets.search_widgets.findChild(QPushButton, "renameOptionBtn").clicked.connect(
            lambda: shared_views.change_indirect("rename_page")
        )
        self.widgets.search_widgets.findChild(QPushButton, "moveOptionBtn").clicked.connect(
            lambda: shared_views.change_indirect("move_page")
        )

    def resizeEvent(self, event):
        """
            RESIZE EVENTS
        """

        # Update Size Grips
        # UiSettings.resize_grips(self)
        if UiStyle.ENABLE_CUSTOM_TITLE_BAR:
            self.left_grip.setGeometry(
                0, 10, 10, self.height()
            )
            self.right_grip.setGeometry(
                self.width() - 10, 10, 10, self.height()
            )
            self.top_grip.setGeometry(
                0, 0, self.width(), 10
            )
            self.bottom_grip.setGeometry(
                0, self.height() - 10, self.width(), 10
            )

    def mousePressEvent(self, event):
        """
            MOUSE CLICK EVENTS
        """

        # GET CURRENT POSITION AS QPointF
        qpointf = event.globalPosition()

        # CONVERT QPointF TO QPoint
        self.dragPos = QPoint(
            int(qpointf.x()),
            int(qpointf.y())
        )

    def get_settings(self) -> dict:

        dir = "data/settings/"
        file = "default.json"

        if not path.exists(dir):
            makedirs(dir)

        # MAKE SURE FILE EXISTS BEFORE MOVING ON
        if path.exists(dir + file):
            return jsonLoad(open(dir + file))

        # DEFAULT SETTINGS ON CREATE
        data = {
            "is_light_theme": False,
        }

        jsonDump(data, open(dir + file, "w+"))

        return data


if __name__ == "__main__":
    from backend.views import shared_views

    app = QApplication(argv)
    app.setWindowIcon(QIcon("icon.ico"))
    window = MainWindow()
    exit(app.exec())