mirror of
https://github.com/283375/arcaea-offline-pyside-ui.git
synced 2025-04-22 02:30:18 +00:00
121 lines
4.2 KiB
Python
121 lines
4.2 KiB
Python
from typing import Type
|
|
|
|
from PySide6.QtCore import Signal
|
|
from PySide6.QtGui import QColor
|
|
from PySide6.QtWidgets import QHBoxLayout, QSizePolicy, QVBoxLayout, QWidget
|
|
|
|
from ui.implements.components.ratingClassRadioButton import RatingClassRadioButton
|
|
|
|
|
|
class RatingClassSelector(QWidget):
|
|
valueChanged = Signal()
|
|
selected = Signal(int)
|
|
|
|
def __init__(self, parent=None):
|
|
super().__init__(parent)
|
|
|
|
self.preferredLayout = QHBoxLayout(self)
|
|
self.preferredLayout.setSpacing(0)
|
|
|
|
self.pstButton = RatingClassRadioButton(self)
|
|
self.pstButton.setObjectName("pstButton")
|
|
self.pstButton.setText("PAST")
|
|
self.pstButton.setAutoExclusive(False)
|
|
self.preferredLayout.addWidget(self.pstButton)
|
|
|
|
self.prsButton = RatingClassRadioButton(self)
|
|
self.prsButton.setObjectName("prsButton")
|
|
self.prsButton.setText("PRESENT")
|
|
self.prsButton.setAutoExclusive(False)
|
|
self.preferredLayout.addWidget(self.prsButton)
|
|
|
|
self.ftrButton = RatingClassRadioButton(self)
|
|
self.ftrButton.setObjectName("ftrButton")
|
|
self.ftrButton.setText("FUTURE")
|
|
self.ftrButton.setAutoExclusive(False)
|
|
self.preferredLayout.addWidget(self.ftrButton)
|
|
|
|
self.bydButton = RatingClassRadioButton(self)
|
|
self.bydButton.setObjectName("bydButton")
|
|
self.bydButton.setText("BEYOND")
|
|
self.bydButton.setAutoExclusive(False)
|
|
self.preferredLayout.addWidget(self.bydButton)
|
|
|
|
self.buttons = [self.pstButton, self.prsButton, self.ftrButton, self.bydButton]
|
|
self.pstButton.setColors(QColor("#399bb2"), QColor("#f0f8fa"))
|
|
self.prsButton.setColors(QColor("#809955"), QColor("#f7f9f4"))
|
|
self.ftrButton.setColors(QColor("#702d60"), QColor("#f7ebf4"))
|
|
self.bydButton.setColors(QColor("#710f25"), QColor("#f9ced8"))
|
|
|
|
self.pstButton.clicked.connect(self.select)
|
|
self.prsButton.clicked.connect(self.select)
|
|
self.ftrButton.clicked.connect(self.select)
|
|
self.bydButton.clicked.connect(self.select)
|
|
self.reset()
|
|
self.setButtonsEnabled([])
|
|
|
|
def setLayout(self, layoutType: Type[QVBoxLayout] | Type[QHBoxLayout]):
|
|
while self.preferredLayout.takeAt(0):
|
|
...
|
|
self.preferredLayout.destroyed.connect(lambda: self.__setLayout(layoutType))
|
|
self.preferredLayout.deleteLater()
|
|
|
|
def __setLayout(self, layoutType: Type[QVBoxLayout] | Type[QHBoxLayout]):
|
|
self.preferredLayout = layoutType(self)
|
|
self.preferredLayout.setSpacing(0)
|
|
for button in self.buttons:
|
|
if layoutType == QVBoxLayout:
|
|
sizePolicy = QSizePolicy(
|
|
QSizePolicy.Policy.Preferred, QSizePolicy.Policy.Expanding
|
|
)
|
|
else:
|
|
sizePolicy = QSizePolicy(
|
|
QSizePolicy.Policy.Expanding, QSizePolicy.Policy.Preferred
|
|
)
|
|
button.setSizePolicy(sizePolicy)
|
|
self.preferredLayout.addWidget(button)
|
|
self.update()
|
|
|
|
def value(self):
|
|
for i, button in enumerate(self.buttons):
|
|
if button.isChecked():
|
|
return i
|
|
return None
|
|
|
|
def reset(self):
|
|
for button in self.buttons:
|
|
button.setChecked(False)
|
|
self.valueChanged.emit()
|
|
|
|
def setButtonsEnabled(self, ratingClasses: list[int]):
|
|
for i, button in enumerate(self.buttons):
|
|
if i in ratingClasses:
|
|
button.setEnabled(True)
|
|
else:
|
|
button.setChecked(False)
|
|
button.setEnabled(False)
|
|
self.valueChanged.emit()
|
|
|
|
def select(self, ratingClass: int | bool | None = None):
|
|
if not (
|
|
type(ratingClass) == int
|
|
or type(ratingClass) == bool
|
|
or ratingClass is not None
|
|
):
|
|
return
|
|
|
|
if ratingClass is None or isinstance(ratingClass, bool):
|
|
button = self.sender()
|
|
elif ratingClass in range(4):
|
|
button = self.buttons[ratingClass]
|
|
else:
|
|
return
|
|
|
|
if not button.isEnabled():
|
|
return
|
|
|
|
self.reset()
|
|
button.setChecked(True)
|
|
self.valueChanged.emit()
|
|
self.selected.emit(self.buttons.index(button))
|