Youtubify/runner.py at main · evanyweber/Youtubify · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
from spotify_client import SpotifyClient
from youtube_client import YoutubeClient

def run(playlist_id, sp_user_client):
    # for server-side processing
    sp_client = SpotifyClient()
    yt_client = YoutubeClient()

    yt_playlist = yt_client.getPlaylistInfo(playlist_id)
    playlist_title = yt_playlist.title
    videos = yt_client.getVideosFromPlaylist(playlist_id)
    tracks = [] # list of track ids for Spotify
    trackCnt = 0 # current number of processed tracks

    # create spotify playlist
    sp_playlist = sp_user_client.user_playlist_create (
        user=sp_user_client.me()['id'],
        name=playlist_title,
        public=False,
        collaborative=False,
        description=f'A collection of songs from the YouTube playlist {playlist_title}.'
    )

    for video in videos:
        title = video.title
        artist = video.artist
        track = video.track
        trackId = -1
        if artist and track:
            trackId = sp_client.getTrackId('{} {}'.format(artist, track))
        else:
            trackId = sp_client.getTrackId(title)
        #print(title)
        if trackId != -1:
            #print(trackId)
            tracks.append(trackId)
            trackCnt += 1
        # add tracks to created playlist, 100 tracks per request (max possible per request)
        if trackCnt == 100:
            sp_user_client.user_playlist_add_tracks(sp_user_client.me()['id'], sp_playlist['id'], tracks)
            tracks = []
            trackCnt = 0

    # add leftover tracks
    if trackCnt > 0:
        sp_user_client.user_playlist_add_tracks(sp_user_client.me()['id'], sp_playlist['id'], tracks)