Merge pull request #70 from troykelly/68-dont-exhaust-genre-when-only…

…-no-similar-tracks Update main.py
troykelly · Jun 3, 2024 · 371b924 · 371b924
2 parents 75fbafa + d6c5075
commit 371b924
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/radioplaylist/main.py b/src/radioplaylist/main.py
@@ -408,9 +408,6 @@ def refresh_genres(genres: List[str]) -> List[str]:
                 '',
             )                        
             for candidate in candidate_tracks:
-                if random.random() < 0.3:
-                    break
-
                 similar_track = self.playlist_manager.get_track_by_title_and_artist(
                     candidate['title'], candidate['artist']
                 )
@@ -427,6 +424,9 @@ def refresh_genres(genres: List[str]) -> List[str]:
                         '',
                     )                    
                     add_track_to_playlist(similar_track, playlist)
+
+                if random.random() < 0.3:
+                    break
 
         return playlist