use LikeStatus enum instead of strings

Author: apastel

tests/conftest.py

@@ -11,6 +11,7 @@
 from ytmusic_deleter import cli
 from ytmusic_deleter import common
 from ytmusicapi import YTMusic
+from ytmusicapi.models.content.enums import LikeStatus
 
 
 def get_resource(file: str) -> str:
@@ -321,7 +322,7 @@ def cleanup_library(yt_browser: YTMusic):
 
 @pytest.fixture(name="add_library_album")
 def fixture_add_library_album(yt_browser: YTMusic, sample_album_as_playlist):
-    response = yt_browser.rate_playlist(sample_album_as_playlist, common.LIKE)
+    response = yt_browser.rate_playlist(sample_album_as_playlist, LikeStatus.LIKE)
     assert "actions" in response
 
     # Wait for album to finish processing
@@ -362,7 +363,7 @@ def fixture_add_library_song(yt_browser: YTMusic, sample_album_browse_id):
 
 @pytest.fixture(name="add_podcast")
 def fixture_add_podcast(yt_browser: YTMusic, sample_podcast):
-    response = yt_browser.rate_playlist(sample_podcast, common.LIKE)
+    response = yt_browser.rate_playlist(sample_podcast, LikeStatus.LIKE)
     assert "actions" in response
 
     # Wait for podcast to be in library
@@ -380,7 +381,7 @@ def fixture_add_podcast(yt_browser: YTMusic, sample_podcast):
 
 @pytest.fixture(name="like_song")
 def fixture_like_song(yt_browser: YTMusic, sample_video):
-    response = yt_browser.rate_song(sample_video, common.LIKE)
+    response = yt_browser.rate_song(sample_video, LikeStatus.LIKE)
     assert "actions" in response
 
     # Wait for song to finish processing
@@ -397,7 +398,7 @@ def fixture_like_song(yt_browser: YTMusic, sample_video):
         time.sleep(2)
 
     # Remove song from library to clean up
-    yt_browser.rate_playlist("OLAK5uy_lZ90LvUqQdKrByCbk99v54d8XpUOmFavo", common.INDIFFERENT)
+    yt_browser.rate_playlist("OLAK5uy_lZ90LvUqQdKrByCbk99v54d8XpUOmFavo", LikeStatus.INDIFFERENT)
 
 
 @pytest.fixture(name="like_songs")
@@ -408,9 +409,9 @@ def fixture_like_songs(yt_browser: YTMusic, medium_song_list):
         if any(common.string_exists_in_dict(existing_like, song) for existing_like in existing_likes):
             print(f"Song {song!r} was already in likes...")
             continue
-        response = yt_browser.rate_song(song, common.LIKE)
+        response = yt_browser.rate_song(song, LikeStatus.LIKE)
         while num_retries > 0 and (not common.string_exists_in_dict(response, "consistencyTokenJar")):
-            response = yt_browser.rate_song(song, common.LIKE)
+            response = yt_browser.rate_song(song, LikeStatus.LIKE)
             num_retries -= 1
         if num_retries == 0:
             pytest.fail(f"Ran out of tries to add song {song!r} to likes.")
@@ -423,13 +424,13 @@ def fixture_like_songs(yt_browser: YTMusic, medium_song_list):
 @pytest.fixture(name="like_many_songs")
 def fixture_like_many_songs(yt_browser: YTMusic, long_song_list):
     for song in long_song_list:
-        response = yt_browser.rate_song(song, common.LIKE)
+        response = yt_browser.rate_song(song, LikeStatus.LIKE)
         num_retries = 300
         while num_retries > 0 and (
             not common.string_exists_in_dict(response, "Removed from liked music")
             or not common.string_exists_in_dict(response, "consistencyTokenJar")
         ):
-            response = yt_browser.rate_song(song, common.LIKE)
+            response = yt_browser.rate_song(song, LikeStatus.LIKE)
             num_retries -= 1
 
 

ytmusic_deleter/cli.py

@@ -17,6 +17,7 @@
 from ytmusic_deleter.progress import manager
 from ytmusic_deleter.uploads import maybe_delete_uploaded_albums
 from ytmusicapi import YTMusic
+from ytmusicapi.models.content.enums import LikeStatus
 
 
 @click.group()
@@ -169,7 +170,7 @@ def remove_library_podcasts():
         if not id:
             logging.debug(f"\tCan't delete podcast {title!r} because it doesn't have an ID.")
             continue
-        response = yt_auth.rate_playlist(id, common.INDIFFERENT)
+        response = yt_auth.rate_playlist(id, LikeStatus.INDIFFERENT)
         if "actions" in response:
             logging.info(f"\tRemoved {title!r} from your library.")
             podcasts_removed += 1
@@ -197,14 +198,15 @@ def remove_library_items(library_items):
         response = None
         if id:
             logging.debug(f"Removing album using id: {id}")
-            response = yt_auth.rate_playlist(id, common.INDIFFERENT)
+            response = yt_auth.rate_playlist(id, LikeStatus.INDIFFERENT)
         elif item.get("feedbackTokens") and isinstance(item.get("feedbackTokens"), dict) and item.get("album"):
             logging.debug("This is a song, removing item by removing containing album.")
             album_browse_id = item["album"]["id"]
             audio_playlist_id = common.get_album_audio_playlist_id(album_browse_id)
             response = None
             if audio_playlist_id:
-                response = yt_auth.rate_playlist(audio_playlist_id, common.INDIFFERENT)
+                response = yt_auth.rate_playlist(audio_playlist_id, LikeStatus.INDIFFERENT)
+                yt_auth.edit_song_library_status
         else:
             logging.error(
                 f"""
@@ -254,14 +256,14 @@ def unlike_all(ctx: click.Context):
         title = track["title"]
         logging.info(f"Processing track: {artist} - {title!r}")
         try:
-            response = yt_auth.rate_song(track["videoId"], common.INDIFFERENT)
+            response = yt_auth.rate_song(track["videoId"], LikeStatus.INDIFFERENT)
             num_retries = 300
             while num_retries > 0 and (
                 not common.string_exists_in_dict(response, "Removed from liked music")
                 or not common.string_exists_in_dict(response, "consistencyTokenJar")
             ):
                 logging.info("\tRetrying track...")
-                response = yt_auth.rate_song(track["videoId"], common.INDIFFERENT)
+                response = yt_auth.rate_song(track["videoId"], LikeStatus.INDIFFERENT)
                 num_retries -= 1
             if num_retries == 0:
                 logging.error("\tRan out of retries to remove track from Likes. Try running 'Unlike All' again.")
@@ -511,7 +513,7 @@ def remove_duplicates(ctx: click.Context, playlist_title, exact):
         return
     if playlist_id == "LM":
         for song in items_to_remove:
-            yt_auth.rate_song(song["videoId"], common.INDIFFERENT)
+            yt_auth.rate_song(song["videoId"], LikeStatus.INDIFFERENT)
     else:
         for chunk in common.chunked(items_to_remove, 50):
             yt_auth.remove_playlist_items(playlist_id, chunk)

ytmusic_deleter/common.py

@@ -4,8 +4,6 @@
 import click
 from ytmusicapi import YTMusic
 
-LIKE = "LIKE"
-INDIFFERENT = "INDIFFERENT"
 BROWSER_FILENAME = "browser.json"
 OAUTH_FILENAME = "oauth.json"
 UNKNOWN_ARTIST = "Unknown Artist"

ytmusic_deleter/uploads.py

@@ -10,6 +10,7 @@
 from ytmusic_deleter.progress import manager
 from ytmusic_deleter.progress import update_progress
 from ytmusicapi import YTMusic
+from ytmusicapi.models.content.enums import LikeStatus
 
 
 def maybe_delete_uploaded_albums() -> tuple[int, int]:
@@ -126,7 +127,7 @@ def scorer(query, choice):
 
     audio_playlist_id = common.get_album_audio_playlist_id(match["browseId"], yt_auth)
     if audio_playlist_id:
-        success = yt_auth.rate_playlist(audio_playlist_id, common.LIKE)
+        success = yt_auth.rate_playlist(audio_playlist_id, LikeStatus.LIKE)
         if success:
             logging.info("\tAdded album to library.")
             return match