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

impr(db): better recommend_charts

Browse Source
This commit is contained in:
283375 2023-09-23 17:51:32 +08:00
parent 3867b273c7
commit 3349620b3a
Signed by: 283375
SSH Key Fingerprint: SHA256:UcX0qg6ZOSDOeieKPGokA5h7soykG61nz2uxuQgVLSk
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/arcaea_offline/database.py
View File

@ -283,7 +283,7 @@ class Database(metaclass=Singleton):
session.delete(score) session.delete(score)
session.commit() session.commit()
def recommend_charts(self, play_result: float): def recommend_charts(self, play_result: float, bounds: float = 0.1):
base_constant = math.ceil(play_result * 10) base_constant = math.ceil(play_result * 10)
results = [] results = []
@ -306,6 +306,8 @@ class Database(metaclass=Singleton):
(ScoreBest.song_id == chart.song_id) (ScoreBest.song_id == chart.song_id)
& (ScoreBest.rating_class == chart.rating_class) & (ScoreBest.rating_class == chart.rating_class)
& (ScoreBest.score >= min_score) & (ScoreBest.score >= min_score)
& (play_result - bounds < ScoreBest.potential)
& (ScoreBest.potential < play_result + bounds)
) )
if session.scalar(score_best_stmt): if session.scalar(score_best_stmt):
chart_id = f"{chart.song_id},{chart.rating_class}" chart_id = f"{chart.song_id},{chart.rating_class}"