mirror of
https://github.com/Lost-MSth/Arcaea-server.git
synced 2025-12-14 08:06:23 +08:00
d65cc3bcbe
- Add support for missions - PTT mechanism: Change first play protection to new best protection - Adapt to the new difficulty ETR - Uncap DORO*C - Incomplete support for "pick_ticket" - Fix requirements: cryptography >= 35.0.0 Note: This is an intermediate test version, only for Arcaea 5.4.0c. Next version will adapt to 5.4.0.
118 lines
4.2 KiB
Python
118 lines
4.2 KiB
Python
from .error import NoData
|
|
from .config_manager import Config
|
|
|
|
|
|
class Chart:
|
|
# defnum: chart const * 10
|
|
|
|
def __init__(self, c=None, song_id: str = None, difficulty: int = None) -> None:
|
|
self.c = c
|
|
self.song_id: str = None
|
|
self.difficulty: int = None
|
|
self.set_chart(song_id, difficulty)
|
|
self.defnum: int = None
|
|
self.song_name: str = None
|
|
|
|
def to_dict(self) -> dict:
|
|
return {
|
|
'difficulty': self.difficulty,
|
|
'chart_const': self.chart_const
|
|
}
|
|
|
|
@property
|
|
def chart_const(self) -> float:
|
|
return self.defnum / 10 if self.defnum else -1
|
|
|
|
@property
|
|
def song_id_difficulty(self) -> str:
|
|
return '%s%d' % (self.song_id, self.difficulty)
|
|
|
|
def set_chart(self, song_id: str = None, difficulty: int = None) -> None:
|
|
self.song_id = song_id
|
|
self.difficulty = int(difficulty) if difficulty is not None else None
|
|
|
|
def select(self) -> None:
|
|
self.c.execute(
|
|
'''select rating_pst, rating_prs, rating_ftr, rating_byn, rating_etr from chart where song_id=:a''', {'a': self.song_id})
|
|
x = self.c.fetchone()
|
|
if x is None:
|
|
if Config.ALLOW_SCORE_WITH_NO_SONG:
|
|
self.defnum = -10
|
|
else:
|
|
raise NoData(f'The song `{self.song_id}` does not exist.', 120)
|
|
else:
|
|
self.defnum = x[self.difficulty]
|
|
|
|
|
|
class Song:
|
|
def __init__(self, c=None, song_id: str = None) -> None:
|
|
self.c = c
|
|
self.song_id: str = song_id
|
|
self.name: str = None
|
|
self.charts: dict = None
|
|
|
|
def to_dict(self) -> dict:
|
|
return {
|
|
'song_id': self.song_id,
|
|
'name': self.name,
|
|
'charts': [chart.to_dict() for chart in self.charts]
|
|
}
|
|
|
|
def from_list(self, x: list) -> 'Song':
|
|
if self.song_id is None:
|
|
self.song_id = x[0]
|
|
self.name = x[1]
|
|
self.charts = [Chart(self.c, self.song_id, 0), Chart(self.c, self.song_id, 1), Chart(
|
|
self.c, self.song_id, 2), Chart(self.c, self.song_id, 3), Chart(self.c, self.song_id, 4)]
|
|
self.charts[0].defnum = x[2]
|
|
self.charts[1].defnum = x[3]
|
|
self.charts[2].defnum = x[4]
|
|
self.charts[3].defnum = x[5]
|
|
self.charts[4].defnum = x[6]
|
|
return self
|
|
|
|
def from_dict(self, d: dict) -> 'Song':
|
|
self.song_id = d['song_id']
|
|
self.name = d.get('name', '')
|
|
self.charts = [Chart(self.c, self.song_id, 0), Chart(self.c, self.song_id, 1), Chart(
|
|
self.c, self.song_id, 2), Chart(self.c, self.song_id, 3)]
|
|
for i in range(4):
|
|
self.charts[i].defnum = -10
|
|
for chart in d['charts']:
|
|
self.charts[chart['difficulty']].defnum = round(
|
|
chart['chart_const'] * 10)
|
|
return self
|
|
|
|
def delete(self) -> None:
|
|
self.c.execute(
|
|
'''delete from chart where song_id=?''', (self.song_id,))
|
|
|
|
def update(self) -> None:
|
|
'''全部更新'''
|
|
self.c.execute(
|
|
'''update chart set name=?, rating_pst=?, rating_prs=?, rating_ftr=?, rating_byn=?, rating_etr=? where song_id=?''', (self.name, self.charts[0].defnum, self.charts[1].defnum, self.charts[2].defnum, self.charts[3].defnum, self.charts[4].defnum, self.song_id))
|
|
|
|
def insert(self) -> None:
|
|
self.c.execute(
|
|
'''insert into chart values (?,?,?,?,?,?,?)''', (self.song_id, self.name, self.charts[0].defnum, self.charts[1].defnum, self.charts[2].defnum, self.charts[3].defnum, self.charts[4].defnum))
|
|
|
|
def select_exists(self, song_id: str = None) -> bool:
|
|
if song_id is not None:
|
|
self.song_id = song_id
|
|
|
|
self.c.execute(
|
|
'''select exists(select * from chart where song_id=?)''', (self.song_id,))
|
|
return bool(self.c.fetchone()[0])
|
|
|
|
def select(self, song_id: str = None) -> 'Song':
|
|
if song_id is not None:
|
|
self.song_id = song_id
|
|
|
|
self.c.execute('''select * from chart where song_id=:a''', {
|
|
'a': self.song_id})
|
|
x = self.c.fetchone()
|
|
if x is None:
|
|
raise NoData(f'The song `{self.song_id}` does not exist.')
|
|
|
|
return self.from_list(x)
|