mirror of
https://github.com/283375/arcaea-offline-pyside-ui.git
synced 2025-04-21 10:10:17 +00:00
263 lines
10 KiB
Python
263 lines
10 KiB
Python
import re
|
|
from typing import Literal
|
|
|
|
from arcaea_offline.database import Database
|
|
from arcaea_offline.models import Chart, Pack
|
|
from arcaea_offline.utils.rating import rating_class_to_text
|
|
from PySide6.QtCore import QModelIndex, Qt, Signal, Slot
|
|
from PySide6.QtGui import QColor, QShowEvent
|
|
from PySide6.QtWidgets import QCompleter, QWidget
|
|
|
|
from ui.designer.components.chartSelector_ui import Ui_ChartSelector
|
|
from ui.extends.components.chartSelector import FuzzySearchCompleterModel
|
|
from ui.extends.shared.delegates.descriptionDelegate import DescriptionDelegate
|
|
from ui.implements.components.ratingClassRadioButton import RatingClassRadioButton
|
|
|
|
|
|
class ChartSelector(Ui_ChartSelector, QWidget):
|
|
valueChanged = Signal()
|
|
|
|
def __init__(self, parent=None):
|
|
super().__init__(parent)
|
|
self.db = Database()
|
|
self.setupUi(self)
|
|
|
|
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.__RATING_CLASS_BUTTONS = [
|
|
self.pstButton,
|
|
self.prsButton,
|
|
self.ftrButton,
|
|
self.bydButton,
|
|
]
|
|
self.pstButton.clicked.connect(self.selectRatingClass)
|
|
self.prsButton.clicked.connect(self.selectRatingClass)
|
|
self.ftrButton.clicked.connect(self.selectRatingClass)
|
|
self.bydButton.clicked.connect(self.selectRatingClass)
|
|
self.deselectAllRatingClassButtons()
|
|
self.updateRatingClassButtonsEnabled([])
|
|
|
|
self.previousPackageButton.clicked.connect(
|
|
lambda: self.quickSwitchSelection("previous", "package")
|
|
)
|
|
self.previousSongIdButton.clicked.connect(
|
|
lambda: self.quickSwitchSelection("previous", "songId")
|
|
)
|
|
self.nextSongIdButton.clicked.connect(
|
|
lambda: self.quickSwitchSelection("next", "songId")
|
|
)
|
|
self.nextPackageButton.clicked.connect(
|
|
lambda: self.quickSwitchSelection("next", "package")
|
|
)
|
|
|
|
self.valueChanged.connect(self.updateResultLabel)
|
|
|
|
self.fillPackageComboBox()
|
|
self.packageComboBox.setCurrentIndex(-1)
|
|
self.songIdComboBox.setCurrentIndex(-1)
|
|
|
|
self.fuzzySearchCompleterModel = FuzzySearchCompleterModel()
|
|
self.fuzzySearchCompleter = QCompleter(self.fuzzySearchCompleterModel)
|
|
self.fuzzySearchCompleter.popup().setItemDelegate(
|
|
DescriptionDelegate(self.fuzzySearchCompleter.popup())
|
|
)
|
|
self.fuzzySearchCompleter.activated[QModelIndex].connect(
|
|
self.fuzzySearchCompleterSetSelection
|
|
)
|
|
self.fuzzySearchLineEdit.setCompleter(self.fuzzySearchCompleter)
|
|
|
|
self.packageComboBox.setItemDelegate(DescriptionDelegate(self.packageComboBox))
|
|
self.songIdComboBox.setItemDelegate(DescriptionDelegate(self.songIdComboBox))
|
|
|
|
self.pstButton.toggled.connect(self.valueChanged)
|
|
self.prsButton.toggled.connect(self.valueChanged)
|
|
self.ftrButton.toggled.connect(self.valueChanged)
|
|
self.bydButton.toggled.connect(self.valueChanged)
|
|
self.packageComboBox.currentIndexChanged.connect(self.valueChanged)
|
|
self.songIdComboBox.currentIndexChanged.connect(self.valueChanged)
|
|
|
|
def quickSwitchSelection(
|
|
self,
|
|
direction: Literal["previous", "next"],
|
|
model: Literal["package", "songId"],
|
|
):
|
|
minIndex = 0
|
|
if model == "package":
|
|
maxIndex = self.packageComboBox.count() - 1
|
|
currentIndex = self.packageComboBox.currentIndex() + (
|
|
1 if direction == "next" else -1
|
|
)
|
|
currentIndex = max(min(maxIndex, currentIndex), minIndex)
|
|
self.packageComboBox.setCurrentIndex(currentIndex)
|
|
elif model == "songId":
|
|
maxIndex = self.songIdComboBox.count() - 1
|
|
currentIndex = self.songIdComboBox.currentIndex() + (
|
|
1 if direction == "next" else -1
|
|
)
|
|
currentIndex = max(min(maxIndex, currentIndex), minIndex)
|
|
self.songIdComboBox.setCurrentIndex(currentIndex)
|
|
else:
|
|
return
|
|
|
|
def value(self):
|
|
packageId = self.packageComboBox.currentData()
|
|
songId = self.songIdComboBox.currentData()
|
|
ratingClass = self.selectedRatingClass()
|
|
|
|
if packageId and songId and isinstance(ratingClass, int):
|
|
return self.db.get_chart(songId, ratingClass)
|
|
return None
|
|
|
|
def showEvent(self, event: QShowEvent):
|
|
# update database results when widget visible
|
|
self.fillPackageComboBox()
|
|
return super().showEvent(event)
|
|
|
|
@Slot()
|
|
def updateResultLabel(self):
|
|
chart = self.value()
|
|
if isinstance(chart, Chart):
|
|
pack = self.db.get_pack_by_id(chart.set)
|
|
texts = [
|
|
[
|
|
pack.name,
|
|
chart.title,
|
|
f"{rating_class_to_text(chart.rating_class)} "
|
|
f"{chart.rating}{'+' if chart.rating_plus else ''}"
|
|
f"({chart.constant / 10})",
|
|
],
|
|
[pack.id, chart.song_id, str(chart.rating_class)],
|
|
]
|
|
texts = [" | ".join(t) for t in texts]
|
|
text = f'{texts[0]}<br><font color="gray">{texts[1]}</font>'
|
|
self.resultLabel.setText(text)
|
|
else:
|
|
self.resultLabel.setText("...")
|
|
|
|
def fillPackageComboBox(self):
|
|
self.packageComboBox.clear()
|
|
packs = self.db.get_packs()
|
|
for pack in packs:
|
|
isAppendPack = re.search(r"_append_.*$", pack.id)
|
|
if isAppendPack:
|
|
basePackId = re.sub(r"_append_.*$", "", pack.id)
|
|
basePackName = self.db.get_pack_by_id(basePackId).name
|
|
packName = f"{basePackName} - {pack.name}"
|
|
else:
|
|
packName = pack.name
|
|
self.packageComboBox.addItem(f"{packName} ({pack.id})", pack.id)
|
|
row = self.packageComboBox.count() - 1
|
|
self.packageComboBox.setItemData(
|
|
row, packName, DescriptionDelegate.MainTextRole
|
|
)
|
|
self.packageComboBox.setItemData(
|
|
row, pack.id, DescriptionDelegate.DescriptionTextRole
|
|
)
|
|
|
|
self.packageComboBox.setCurrentIndex(-1)
|
|
|
|
def fillSongIdComboBox(self):
|
|
self.songIdComboBox.clear()
|
|
packId = self.packageComboBox.currentData()
|
|
if packId:
|
|
charts = self.db.get_charts_by_pack_id(packId)
|
|
inserted_song_ids = []
|
|
for chart in charts:
|
|
if chart.song_id not in inserted_song_ids:
|
|
self.songIdComboBox.addItem(
|
|
f"{chart.title} ({chart.song_id})", chart.song_id
|
|
)
|
|
inserted_song_ids.append(chart.song_id)
|
|
row = self.songIdComboBox.count() - 1
|
|
self.songIdComboBox.setItemData(
|
|
row, chart.title, DescriptionDelegate.MainTextRole
|
|
)
|
|
self.songIdComboBox.setItemData(
|
|
row, chart.song_id, DescriptionDelegate.DescriptionTextRole
|
|
)
|
|
self.songIdComboBox.setCurrentIndex(-1)
|
|
|
|
@Slot()
|
|
def on_packageComboBox_activated(self):
|
|
self.fillSongIdComboBox()
|
|
|
|
@Slot(int)
|
|
def on_songIdComboBox_currentIndexChanged(self, index: int):
|
|
rating_classes = []
|
|
if index > -1:
|
|
charts = self.db.get_charts_by_song_id(self.songIdComboBox.currentData())
|
|
rating_classes = [chart.rating_class for chart in charts]
|
|
self.updateRatingClassButtonsEnabled(rating_classes)
|
|
|
|
@Slot()
|
|
def on_resetButton_clicked(self):
|
|
self.packageComboBox.setCurrentIndex(-1)
|
|
self.songIdComboBox.setCurrentIndex(-1)
|
|
|
|
@Slot(str)
|
|
def on_fuzzySearchLineEdit_textChanged(self, text: str):
|
|
if text:
|
|
self.fuzzySearchCompleterModel.fillDbFuzzySearchResults(self.db, text)
|
|
else:
|
|
self.fuzzySearchCompleterModel.clear()
|
|
|
|
def selectChart(self, chart: Chart):
|
|
packageIdIndex = self.packageComboBox.findData(chart.set)
|
|
if packageIdIndex > -1:
|
|
self.packageComboBox.setCurrentIndex(packageIdIndex)
|
|
else:
|
|
# QMessageBox
|
|
return
|
|
|
|
self.fillSongIdComboBox()
|
|
songIdIndex = self.songIdComboBox.findData(chart.song_id)
|
|
if songIdIndex > -1:
|
|
self.songIdComboBox.setCurrentIndex(songIdIndex)
|
|
else:
|
|
# QMessageBox
|
|
return
|
|
|
|
self.selectRatingClass(chart.rating_class)
|
|
|
|
@Slot(QModelIndex)
|
|
def fuzzySearchCompleterSetSelection(self, index: QModelIndex):
|
|
chart = index.data(Qt.ItemDataRole.UserRole + 10) # type: Chart
|
|
self.selectChart(chart)
|
|
|
|
self.fuzzySearchLineEdit.clear()
|
|
self.fuzzySearchLineEdit.clearFocus()
|
|
|
|
def ratingClassButtons(self):
|
|
return self.__RATING_CLASS_BUTTONS
|
|
|
|
def selectedRatingClass(self):
|
|
for i, button in enumerate(self.__RATING_CLASS_BUTTONS):
|
|
if button.isChecked():
|
|
return i
|
|
|
|
def updateRatingClassButtonsEnabled(self, rating_classes: list[int]):
|
|
for i, button in enumerate(self.__RATING_CLASS_BUTTONS):
|
|
if i in rating_classes:
|
|
button.setEnabled(True)
|
|
else:
|
|
button.setChecked(False)
|
|
button.setEnabled(False)
|
|
|
|
def deselectAllRatingClassButtons(self):
|
|
[button.setChecked(False) for button in self.__RATING_CLASS_BUTTONS]
|
|
|
|
@Slot()
|
|
def selectRatingClass(self, rating_class: int | None = None):
|
|
if type(rating_class) == int and rating_class in range(4):
|
|
self.deselectAllRatingClassButtons()
|
|
button = self.__RATING_CLASS_BUTTONS[rating_class]
|
|
if button.isEnabled():
|
|
button.setChecked(True)
|
|
else:
|
|
button = self.sender()
|
|
if isinstance(button, RatingClassRadioButton) and button.isEnabled():
|
|
self.deselectAllRatingClassButtons()
|
|
button.setChecked(True)
|