blee/beatsaber-playlist-tool
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Minor comment cleanup and type hint.

Browse Source
This commit is contained in:
Brian Lee 2024-07-09 16:04:17 -07:00
parent 2d396737e4
commit 693b15faee
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/saberlist/scoresaber.py
View File

@ -48,7 +48,7 @@ def leaderboards_to_playlist(leaderboards: list, playlist_title: str, playlist_a
logging.info(f"Playlist written to {playlist_title}.bplist") logging.info(f"Playlist written to {playlist_title}.bplist")
return playlist_json return playlist_json
def scores_to_playlist(scores, playlist_title, playlist_author = "SaberList Tool"): def scores_to_playlist(scores: list, playlist_title, playlist_author = "SaberList Tool"):
playlist = { playlist = {
"playlistTitle": playlist_title, "playlistTitle": playlist_title,
"playlistAuthor": playlist_author, "playlistAuthor": playlist_author,
@ -69,7 +69,6 @@ def scores_to_playlist(scores, playlist_title, playlist_author = "SaberList Tool
} }
playlist["songs"].append(song_entry) playlist["songs"].append(song_entry)
playlist_json = json.dumps(playlist, indent=4) playlist_json = json.dumps(playlist, indent=4)
with open(f"{playlist_title}.bplist", 'w') as file: with open(f"{playlist_title}.bplist", 'w') as file:
file.write(playlist_json) file.write(playlist_json)
@ -80,7 +79,7 @@ def scores_to_playlist(scores, playlist_title, playlist_author = "SaberList Tool
async def async_replay_ranked(): async def async_replay_ranked():
scoresaber = ExtendedScoreSaberAPI() scoresaber = ExtendedScoreSaberAPI()
try: try:
await scoresaber.start() # Initialize the API client await scoresaber.start()
score_sort = ScoreSort.TOP score_sort = ScoreSort.TOP
scores = [] scores = []
@ -99,7 +98,7 @@ async def async_replay_ranked():
filtered_sorted_scores = filter_and_sort_scores_by_stars(scores, min_stars, max_stars) filtered_sorted_scores = filter_and_sort_scores_by_stars(scores, min_stars, max_stars)
scores_to_playlist(filtered_sorted_scores, playlist_title) scores_to_playlist(filtered_sorted_scores, playlist_title)
finally: finally:
await scoresaber._http_client.close() # Ensure the session is closed await scoresaber._http_client.close()
def replay_ranked(): def replay_ranked():
try: try:
@ -173,8 +172,7 @@ class ExtendedScoreSaberAPI(ScoreSaberAPI):
if max_page == -1: if max_page == -1:
max_page = ceil(recent_scores.metadata.total / 100) + 1 max_page = ceil(recent_scores.metadata.total / 100) + 1
logging.info(f"Got {len(recent_scores.player_scores)} scores") logging.info(f"Got {len(recent_scores.player_scores)} scores from page {page}")
yield recent_scores.player_scores yield recent_scores.player_scores
page += 1 page += 1