get_youtube_music

get_youtube_music.py

import youtube_dl
import pandas as pd
def get_youtube_music(row):
    path = 'data/music/{}/{}.mp3'.format(row['style'],str(row['uuid']))
    exists = os.path.isfile(path)
    if exists:
        print('Already exist {}'.format(row['uuid']))
        return
    if row['videoID_youtube'] != "missing":
        videoID = row['videoID_youtube']
        ydl_opts = {
            'outtmpl': path,
            'format': 'bestaudio/best',
            'postprocessors': [{
                'key': 'FFmpegExtractAudio',
                'preferredcodec': 'mp3'
            }]
        }
        with youtube_dl.YoutubeDL(ydl_opts) as ydl:
            ydl.download(['https://www.youtube.com/watch?v={}'.format(videoID)])
            
df = pd.read_csv(output_file,sep=";", header=0)            
df.progress_apply(get_youtube_music, axis=1)