Skip to content

Commit

Permalink
support ytm likes
Browse files Browse the repository at this point in the history
  • Loading branch information
platers committed May 18, 2022
1 parent 1822057 commit 32dc8b0
Showing 1 changed file with 16 additions and 1 deletion.
17 changes: 16 additions & 1 deletion unitunes/services/ytm.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
from abc import ABC, abstractmethod
from pathlib import Path
import time
from typing import Any, List, Optional
from tqdm import tqdm
from ytmusicapi import YTMusic
from unitunes.playlist import PlaylistMetadata
from youtube_title_parse import get_artist_title
Expand Down Expand Up @@ -96,10 +98,23 @@ def edit_description(self, playlist_id: str, description: str) -> None:

def add_tracks(self, playlist_id: str, track_ids: List[str]) -> None:
"""Add tracks to a playlist."""
self.ytm.add_playlist_items(playlist_id, track_ids)
if playlist_id == "LM":
for track_id in tqdm(track_ids, desc="Rating songs"):
self.ytm.rate_song(track_id, "LIKE")
time.sleep(0.8)

else:
self.ytm.add_playlist_items(playlist_id, track_ids)

def remove_tracks(self, playlist_id: str, track_ids: List[str]) -> None:
"""Remove tracks from a playlist."""

if playlist_id == "LM":
for track_id in tqdm(track_ids, desc="Unrating songs"):
self.ytm.rate_song(track_id, "INDIFFERENT")
time.sleep(0.8)
return

playlist = self.get_playlist(playlist_id)
playlist_items = playlist["tracks"]
videos_to_remove = [
Expand Down

0 comments on commit 32dc8b0

Please sign in to comment.