283375/arcaea-offline-pyside-ui
1
0
Fork 0
mirror of https://github.com/283375/arcaea-offline-pyside-ui.git synced 2025-04-20 17:50:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: SongIdSelectorMode

Browse Source
This commit is contained in:
283375 2023-09-17 00:40:26 +08:00
parent b5aefb5f28
commit 716e872f08
Signed by: 283375
SSH Key Fingerprint: SHA256:UcX0qg6ZOSDOeieKPGokA5h7soykG61nz2uxuQgVLSk
2 changed files with 40 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
ui/implements/components/chartSelector.py
View File

@ -9,6 +9,7 @@ from PySide6.QtWidgets import QWidget
from ui.designer.components.chartSelector_ui import Ui_ChartSelector
from ui.extends.shared.database import databaseUpdateSignals
from ui.extends.shared.language import LanguageChangeEventFilter
from ui.implements.components.songIdSelector import SongIdSelectorMode
logger = logging.getLogger(__name__)
@ -36,6 +37,9 @@ class ChartSelector(Ui_ChartSelector, QWidget):
)
databaseUpdateSignals.songDataUpdated.connect(self.updateDatabase)
def setSongIdSelectorMode(self, mode: SongIdSelectorMode):
self.songIdSelector.setMode(mode)
def value(self):
songId = self.songIdSelector.songId()
ratingClass = self.ratingClassSelector.value()
@ -79,8 +83,11 @@ class ChartSelector(Ui_ChartSelector, QWidget):
ratingClasses = []
songId = self.songIdSelector.songId()
if songId:
charts = self.db.get_charts_by_song_id(songId)
ratingClasses = [chart.rating_class for chart in charts]
if self.songIdSelector.mode == SongIdSelectorMode.Chart:
items = self.db.get_charts_by_song_id(songId)
else:
items = self.db.get_difficulties_by_song_id(songId)
ratingClasses = [item.rating_class for item in items]
self.ratingClassSelector.setButtonsEnabled(ratingClasses)
@Slot()

41
ui/implements/components/songIdSelector.py
View File

@ -1,5 +1,6 @@
import logging
import re
from enum import IntEnum
from typing import Literal
from arcaea_offline.database import Database
@ -16,6 +17,11 @@ from ui.extends.shared.language import LanguageChangeEventFilter
logger = logging.getLogger(__name__)
class SongIdSelectorMode(IntEnum):
SongId = 0
Chart = 1
class SongIdSelector(Ui_SongIdSelector, QWidget):
valueChanged = Signal()
@ -40,6 +46,8 @@ class SongIdSelector(Ui_SongIdSelector, QWidget):
lambda: self.quickSwitchSelection("next", "package")
)
self.mode = SongIdSelectorMode.SongId
databaseUpdateSignals.songDataUpdated.connect(self.updateDatabase)
self.fillPackComboBox()
@ -62,6 +70,9 @@ class SongIdSelector(Ui_SongIdSelector, QWidget):
self.packComboBox.currentIndexChanged.connect(self.valueChanged)
self.songIdComboBox.currentIndexChanged.connect(self.valueChanged)
def setMode(self, mode: SongIdSelectorMode):
self.mode = mode
def quickSwitchSelection(
self,
direction: Literal["previous", "next"],
@ -135,20 +146,30 @@ class SongIdSelector(Ui_SongIdSelector, QWidget):
self.songIdComboBox.clear()
packId = self.packComboBox.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)
if self.mode == SongIdSelectorMode.SongId:
items = self.db.get_songs_by_pack_id(packId)
elif self.mode == SongIdSelectorMode.Chart:
items = self.db.get_charts_by_pack_id(packId)
else:
raise ValueError("Unknown SongIdSelectorMode.")
insertedSongIds = []
for item in items:
if self.mode == SongIdSelectorMode.SongId:
itemId = item.id
elif self.mode == SongIdSelectorMode.Chart:
itemId = item.song_id
else:
continue
if itemId not in insertedSongIds:
self.songIdComboBox.addItem(f"{item.title} ({itemId})", itemId)
insertedSongIds.append(itemId)
row = self.songIdComboBox.count() - 1
self.songIdComboBox.setItemData(
row, chart.title, DescriptionDelegate.MainTextRole
row, item.title, DescriptionDelegate.MainTextRole
)
self.songIdComboBox.setItemData(
row, chart.song_id, DescriptionDelegate.DescriptionTextRole
row, itemId, DescriptionDelegate.DescriptionTextRole
)
self.songIdComboBox.setCurrentIndex(-1)