Y-Tracker/components/mainWindow.py

from PyQt6.QtWidgets import (
    QWidget,
    QPushButton,
    QVBoxLayout,
    QHBoxLayout,
)

from PyQt6.QtCore import Qt
from components.customButton import CustomButton

from data.collectibles import collectibles

class MainWindow(QWidget):
    collectibles = {}
    
    def __init__(self: QWidget, *args, **kwargs):
        super().__init__(*args, **kwargs)
        
        self.initUI()
        
    def initUI(self):
        # Set the window title
        self.setWindowTitle('Y-Tracker')
        self.collectibles = collectibles
        
        # Create & Edit Layout
        layout = QVBoxLayout()
        self.setLayout(layout)
        
        vLayout = QVBoxLayout()
        hLayout = QHBoxLayout()
        
        layout.addLayout(hLayout)
        layout.addLayout(vLayout)
        
        self.layout_dict = {}
        self.label_dict = {}
        self.button_dict = {}
        self.modifiers = Qt.KeyboardModifier.NoModifier
        
        # Build the Inputs
        for type in self.collectibles:
            
            if self.collectibles[type]['meta']['orient'] == 'horizontal':
                self.layout_dict[type] = QHBoxLayout()
                vLayout.addLayout(self.layout_dict[type])
                
            else:
                self.layout_dict[type] = QVBoxLayout()
                hLayout.addLayout(self.layout_dict[type])
            
            self.button_dict[type] = {}
            
            for item in self.collectibles[type]:
                
                if item != 'meta':
                    self.button_dict[type][item] = CustomButton()
                    self.button_dict[type][item].setObjectName(f'{type}-{item}')
                    
                    if self.collectibles[type]['meta']['type'] == 'bool':
                        self.button_dict[type][item].setText(self.collectibles[type][item]['name'])
                        
                        if self.collectibles[type][item]['value'] == False:
                            self.button_dict[type][item].setStyleSheet("""
                                QPushButton#""" + self.button_dict[type][item].objectName() + """ {
                                    color: #88FFFFFF
                                }
                                """)
                            
                        else:
                            self.button_dict[type][item].setStyleSheet("""
                                QPushButton#""" + self.button_dict[type][item].objectName() + """ {
                                    color: #FFFFFFFF
                                }
                                """)
                            
                        self.button_dict[type][item].clicked.connect(lambda state, x=[type, item]: self.boolButtonSelect(x[0], x[1]))
                        
                    elif self.collectibles[type]['meta']['type'] == 'int':
                        self.button_dict[type][item].setText(f'{self.collectibles[type][item]["name"]}: {self.collectibles[type][item]["value"]}')
                        self.button_dict[type][item].clicked.connect(lambda state, x=[type, item]: self.intButtonSelect(x[0], x[1]))
                        
                    elif self.collectibles[type]['meta']['type'] == 'list':
                        self.button_dict[type][item].setText(self.collectibles[type]['meta']['list'][self.collectibles[type][item]['value']])
                        self.button_dict[type][item].clicked.connect(lambda state, x=[type, item]: self.listButtonSelect(x[0], x[1]))
                        
                    self.layout_dict[type].addWidget(self.button_dict[type][item])
                    
        # show the window
        self.show()
                
    # Add Slots
    def boolButtonSelect(self, type, item):
        if self.collectibles[type][item]['value']:
            self.collectibles[type][item]['value'] = False
            self.button_dict[type][item].setStyleSheet("""
                QPushButton#""" + self.button_dict[type][item].objectName() + """ {
                    color: #88FFFFFF
                }
                """)
        else:
            self.collectibles[type][item]['value'] = True
            self.button_dict[type][item].setStyleSheet("""
                QPushButton#""" + self.button_dict[type][item].objectName() + """ {
                    color: #FFFFFFFF
                }
                """)
    
    def intButtonSelect(self, type, item):
        if self.modifiers == Qt.KeyboardModifier.ShiftModifier:
            self.collectibles[type][item]['value'] = self.collectibles[type][item]['value'] - 1
        else:
            self.collectibles[type][item]['value'] = self.collectibles[type][item]['value'] + 1
        
        self.button_dict[type][item].setText(f'{self.collectibles[type][item]["name"]}: {self.collectibles[type][item]["value"]}')

    def listButtonSelect(self, type, item):
        if self.modifiers == Qt.KeyboardModifier.ShiftModifier:
            self.collectibles[type][item]['value'] = self.collectibles[type][item]['value'] - 1
        else:
            self.collectibles[type][item]['value'] = self.collectibles[type][item]['value'] + 1
        
        try:
            self.button_dict[type][item].setText(self.collectibles[type]['meta']['list'][self.collectibles[type][item]['value']])
        except:
            if self.collectibles[type][item]['value'] < 0:
                self.collectibles[type][item]['value'] = len(self.collectibles[type]['meta']['list']) - 1
            else:
                self.collectibles[type][item]['value'] = 0
            
            self.button_dict[type][item].setText(self.collectibles[type]['meta']['list'][self.collectibles[type][item]['value']])
            
    def mousePressEvent(self, e):
        self.modifiers = e.modifiers()