283375/arcaea-offline-pyside-ui
1
0
Fork 0
mirror of https://github.com/283375/arcaea-offline-pyside-ui.git synced 2025-04-21 10:10: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.designer.components.chartSelector_ui import Ui_ChartSelector
from ui.extends.shared.database import databaseUpdateSignals from ui.extends.shared.database import databaseUpdateSignals
from ui.extends.shared.language import LanguageChangeEventFilter from ui.extends.shared.language import LanguageChangeEventFilter
from ui.implements.components.songIdSelector import SongIdSelectorMode
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -36,6 +37,9 @@ class ChartSelector(Ui_ChartSelector, QWidget):
) )
databaseUpdateSignals.songDataUpdated.connect(self.updateDatabase) databaseUpdateSignals.songDataUpdated.connect(self.updateDatabase)
def setSongIdSelectorMode(self, mode: SongIdSelectorMode):
self.songIdSelector.setMode(mode)
def value(self): def value(self):
songId = self.songIdSelector.songId() songId = self.songIdSelector.songId()
ratingClass = self.ratingClassSelector.value() ratingClass = self.ratingClassSelector.value()
@ -79,8 +83,11 @@ class ChartSelector(Ui_ChartSelector, QWidget):
ratingClasses = [] ratingClasses = []
songId = self.songIdSelector.songId() songId = self.songIdSelector.songId()
if songId: if songId:
charts = self.db.get_charts_by_song_id(songId) if self.songIdSelector.mode == SongIdSelectorMode.Chart:
ratingClasses = [chart.rating_class for chart in charts] 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) self.ratingClassSelector.setButtonsEnabled(ratingClasses)
@Slot() @Slot()

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

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