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")
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 = {
"playlistTitle": playlist_title,
"playlistAuthor": playlist_author,
@ -69,7 +69,6 @@ def scores_to_playlist(scores, playlist_title, playlist_author = "SaberList Tool
}
playlist["songs"].append(song_entry)
playlist_json = json.dumps(playlist, indent=4)
with open(f"{playlist_title}.bplist", 'w') as file:
file.write(playlist_json)
@ -80,7 +79,7 @@ def scores_to_playlist(scores, playlist_title, playlist_author = "SaberList Tool
async def async_replay_ranked():
scoresaber = ExtendedScoreSaberAPI()
try:
await scoresaber.start() # Initialize the API client
await scoresaber.start()
score_sort = ScoreSort.TOP
scores = []
@ -99,7 +98,7 @@ async def async_replay_ranked():
filtered_sorted_scores = filter_and_sort_scores_by_stars(scores, min_stars, max_stars)
scores_to_playlist(filtered_sorted_scores, playlist_title)
finally:
await scoresaber._http_client.close() # Ensure the session is closed
await scoresaber._http_client.close()
def replay_ranked():
try:
@ -173,8 +172,7 @@ class ExtendedScoreSaberAPI(ScoreSaberAPI):
if max_page == -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
page += 1